import os from urllib.parse import urljoin import re from django.db import models from django.conf import settings from django.urls import reverse class SurvexDirectory(models.Model): path = models.CharField(max_length=200) cave = models.ForeignKey('Cave', blank=True, null=True,on_delete=models.CASCADE) primarysurvexfile = models.ForeignKey('SurvexFile', related_name='primarysurvexfile', blank=True, null=True,on_delete=models.CASCADE) # could also include files in directory but not referenced class Meta: ordering = ('id',) def __str__(self): return str(self.path) + "-" + str(self.primarysurvexfile.path) class SurvexFile(models.Model): path = models.CharField(max_length=200) survexdirectory = models.ForeignKey("SurvexDirectory", blank=True, null=True,on_delete=models.CASCADE) cave = models.ForeignKey('Cave', blank=True, null=True,on_delete=models.CASCADE) class Meta: ordering = ('id',) def exists(self): fname = os.path.join(settings.SURVEX_DATA, self.path + ".svx") return os.path.isfile(fname) def OpenFile(self): fname = os.path.join(settings.SURVEX_DATA, self.path + ".svx") return open(fname) def SetDirectory(self): dirpath = os.path.split(self.path)[0] # pointless search every time we import a survex file if we know there are no duplicates.. # don't use this for initial import. survexdirectorylist = SurvexDirectory.objects.filter(cave=self.cave, path=dirpath) if survexdirectorylist: self.survexdirectory = survexdirectorylist[0] else: survexdirectory = SurvexDirectory(path=dirpath, cave=self.cave, primarysurvexfile=self) survexdirectory.save() self.survexdirectory = survexdirectory self.save() class SurvexStationLookUpManager(models.Manager): def lookup(self, name): blocknames, sep, stationname = name.rpartition(".") return self.get(block = SurvexBlock.objects.lookup(blocknames), name__iexact = stationname) class SurvexStation(models.Model): name = models.CharField(max_length=100) block = models.ForeignKey('SurvexBlock',on_delete=models.CASCADE) # equate = models.ForeignKey('SurvexEquate', blank=True, null=True,on_delete=models.CASCADE) objects = SurvexStationLookUpManager() x = models.FloatField(blank=True, null=True) y = models.FloatField(blank=True, null=True) z = models.FloatField(blank=True, null=True) def path(self): r = self.name b = self.block while True: if b.name: r = b.name + "." + r if b.parent: b = b.parent else: return r # # Single SurvexBlock # class SurvexBlockLookUpManager(models.Manager): def lookup(self, name): if name == "": blocknames = [] else: blocknames = name.split(".") block = SurvexBlock.objects.get(parent=None, survexfile__path=settings.SURVEX_TOPNAME) for blockname in blocknames: block = SurvexBlock.objects.get(parent=block, name__iexact=blockname) return block class SurvexBlock(models.Model): objects = SurvexBlockLookUpManager() name = models.CharField(max_length=100) parent = models.ForeignKey('SurvexBlock', blank=True, null=True,on_delete=models.CASCADE) cave = models.ForeignKey('Cave', blank=True, null=True,on_delete=models.CASCADE) date = models.DateField(blank=True, null=True) expeditionday = models.ForeignKey("ExpeditionDay", null=True,on_delete=models.CASCADE) expedition = models.ForeignKey('Expedition', blank=True, null=True,on_delete=models.CASCADE) survexfile = models.ForeignKey("SurvexFile", blank=True, null=True,on_delete=models.CASCADE) survexpath = models.CharField(max_length=200) # the path for the survex stations scansfolder = models.ForeignKey("ScansFolder", null=True,on_delete=models.CASCADE) legsall = models.IntegerField(null=True) # summary data for this block legssplay = models.IntegerField(null=True) # summary data for this block legssurfc = models.IntegerField(null=True) # summary data for this block totalleglength = models.FloatField(null=True) class Meta: ordering = ('id',) def isSurvexBlock(self): # Function used in templates return True def __str__(self): return self.name and str(self.name) or 'no name' def GetPersonroles(self): res = [ ] for personrole in self.personrole_set.order_by('personexpedition'): if res and res[-1]['person'] == personrole.personexpedition.person: res[-1]['roles'] += ", " + str(personrole.role) else: res.append({'person':personrole.personexpedition.person, 'expeditionyear':personrole.personexpedition.expedition.year, 'roles':str(personrole.role)}) return res def DayIndex(self): return list(self.expeditionday.survexblock_set.all()).index(self) class SurvexTitle(models.Model): survexblock = models.ForeignKey('SurvexBlock',on_delete=models.CASCADE) title = models.CharField(max_length=200) # needed by svxcavesingle.html template cave = models.ForeignKey('Cave', blank=True, null=True,on_delete=models.CASCADE) # # member of a SurvexBlock # ROLE_CHOICES = ( ('insts','Instruments'), ('dog','Other'), ('notes','Notes'), ('pics','Pictures'), ('tape','Tape measure'), ('useless','Useless'), ('helper','Helper'), ('disto','Disto'), ('consultant','Consultant'), ) class SurvexPersonRole(models.Model): survexblock = models.ForeignKey('SurvexBlock',on_delete=models.CASCADE) nrole = models.CharField(choices=ROLE_CHOICES, max_length=200, blank=True, null=True) # increasing levels of precision personname = models.CharField(max_length=100) person = models.ForeignKey('Person', blank=True, null=True,on_delete=models.CASCADE) personexpedition = models.ForeignKey('PersonExpedition', blank=True, null=True,on_delete=models.CASCADE) persontrip = models.ForeignKey('PersonTrip', blank=True, null=True,on_delete=models.CASCADE) expeditionday = models.ForeignKey("ExpeditionDay", null=True,on_delete=models.CASCADE) def __str__(self): return str(self.person) + " - " + str(self.survexblock) + " - " + str(self.nrole) class ScansFolder(models.Model): fpath = models.CharField(max_length=200) walletname = models.CharField(max_length=200) class Meta: ordering = ('walletname',) def get_absolute_url(self): return urljoin(settings.URL_ROOT, reverse('surveyscansfolder', kwargs={"path":re.sub("#", "%23", self.walletname)})) def __str__(self): return str(self.walletname) + " (Survey Scans Folder)" class SingleScan(models.Model): ffile = models.CharField(max_length=200) name = models.CharField(max_length=200) scansfolder = models.ForeignKey("ScansFolder", null=True,on_delete=models.CASCADE) class Meta: ordering = ('name',) def get_absolute_url(self): return urljoin(settings.URL_ROOT, reverse('surveyscansingle', kwargs={"path":re.sub("#", "%23", self.scansfolder.walletname), "file":self.name})) def __str__(self): return "Survey Scan Image: " + str(self.name) + " in " + str(self.scansfolder) class TunnelFile(models.Model): tunnelpath = models.CharField(max_length=200) tunnelname = models.CharField(max_length=200) bfontcolours = models.BooleanField(default=False) manyscansfolders = models.ManyToManyField("ScansFolder") # implicitly links to SVX files scans = models.ManyToManyField("SingleScan") # implicitly links to SVX files tunnelcontains = models.ManyToManyField("TunnelFile") # case when its a frame type filesize = models.IntegerField(default=0) npaths = models.IntegerField(default=0) survexfiles = models.ManyToManyField("SurvexFile") # direct link to SVX files class Meta: ordering = ('tunnelpath',)