diff options
Diffstat (limited to 'core/models/logbooks.py')
| -rw-r--r-- | core/models/logbooks.py | 181 |
1 files changed, 181 insertions, 0 deletions
diff --git a/core/models/logbooks.py b/core/models/logbooks.py new file mode 100644 index 0000000..da8d970 --- /dev/null +++ b/core/models/logbooks.py @@ -0,0 +1,181 @@ +import datetime +import json +import operator +import os +import re +import string +import subprocess +from collections import defaultdict +from datetime import datetime, timezone +from pathlib import Path +from urllib.parse import urljoin + +from django.conf import settings +from django.contrib.auth.models import User +from django.contrib.contenttypes.models import ContentType +from django.core.files.storage import FileSystemStorage +from django.db import models +from django.db.models import Max, Min +from django.shortcuts import render +from django.template import Context, loader +from django.urls import reverse + +import settings +from troggle.core.models.caves import (Area, Cave, CaveAndEntrance, + Entrance) +from troggle.core.models.survex import SurvexStation +from troggle.core.models.troggle import (DataIssue, Expedition, Person, + TroggleModel) + +'''The model declarations LogBookEntry, PersonTrip, QM +''' + +todo=''' +- Rename PersonTrip as PersonLogEntry or similar +''' + +class LogbookEntry(TroggleModel): + """Single parsed entry from Logbook + """ + date = models.DateField()#MJG wants to turn this into a datetime such that multiple Logbook entries on the same day can be ordered.ld() + expeditionday = models.ForeignKey("ExpeditionDay", null=True,on_delete=models.SET_NULL)#MJG wants to KILL THIS (redundant information) + expedition = models.ForeignKey(Expedition,blank=True, null=True,on_delete=models.SET_NULL) # yes this is double- + title = models.CharField(max_length=200) + cave_slug = models.SlugField(max_length=50, blank=True, null=True) + place = models.CharField(max_length=100,blank=True, null=True,help_text="Only use this if you haven't chosen a cave") + text = models.TextField() + slug = models.SlugField(max_length=50) + time_underground = models.FloatField(null=True,help_text="In decimal hours") + + class Meta: + verbose_name_plural = "Logbook Entries" + # several PersonTrips point in to this object + ordering = ('-date',) + + def cave(self): # Why didn't he just make this a foreign key to Cave ? + c = CaveSlug.objects.get(slug=self.cave_slug, primary=True).cave + return c + + def isLogbookEntry(self): # Function used in templates + return True + + def get_absolute_url(self): + return urljoin(settings.URL_ROOT, reverse('logbookentry',kwargs={'date':self.date,'slug':self.slug})) + + def __str__(self): + return f'{self.date}: {self.title}' + + def get_next_by_id(self): + LogbookEntry.objects.get(id=self.id+1) + + def get_previous_by_id(self): + LogbookEntry.objects.get(id=self.id-1) + + def DayIndex(self): + return list(self.expeditionday.logbookentry_set.all()).index(self) + +class PersonTrip(TroggleModel): + """Single Person going on a trip, which may or may not be written up. + It could account for different T/U for people in same logbook entry. + """ + personexpedition = models.ForeignKey("PersonExpedition",null=True,on_delete=models.CASCADE) + time_underground = models.FloatField(help_text="In decimal hours") + logbook_entry = models.ForeignKey(LogbookEntry,on_delete=models.CASCADE) + is_logbook_entry_author = models.BooleanField(default=False) + + class Meta: + ordering = ('-personexpedition',) + #order_with_respect_to = 'personexpedition' + + def persontrip_next(self): + futurePTs = PersonTrip.objects.filter(personexpedition = self.personexpedition, logbook_entry__date__gt = self.logbook_entry.date).order_by('logbook_entry__date').all() + if len(futurePTs) > 0: + return futurePTs[0] + else: + return None + + def persontrip_prev(self): + pastPTs = PersonTrip.objects.filter(personexpedition = self.personexpedition, logbook_entry__date__lt = self.logbook_entry.date).order_by('-logbook_entry__date').all() + if len(pastPTs) > 0: + return pastPTs[0] + else: + return None + + def place(self): + return self.logbook_entry.cave and self.logbook_entry.cave or self.logbook_entry.place + + def __str__(self): + return f'{self.personexpedition} ({self.logbook_entry.date})' + +class QM(TroggleModel): + """This is based on qm.csv in trunk/expoweb/1623/204 which has the fields: + "Number","Grade","Area","Description","Page reference","Nearest station","Completion description","Comment" + """ + cave = models.ForeignKey(Cave, related_name='QMs',blank=True, null=True,on_delete=models.SET_NULL ) + block = models.ForeignKey('SurvexBlock', null=True,on_delete=models.SET_NULL) # only for QMs from survex files + blockname=models.TextField(blank=True,null=True) # NB truncated copy of survexblock name with last char added + expoyear = models.CharField(max_length=4,blank=True, null=True) # could change to datetime if logbooks similarly chnaged + found_by = models.ForeignKey(LogbookEntry, related_name='QMs_found',blank=True, null=True,on_delete=models.SET_NULL ) + ticked = models.BooleanField(default=False) # for ticked QMs not attached to a logbook entry, should imply completion_description has text + ticked_off_by = models.ForeignKey(LogbookEntry, related_name='QMs_ticked_off',blank=True, null=True,on_delete=models.SET_NULL) # unused, ever?! + number = models.IntegerField(help_text="this is the sequential number in the year, only unique for CSV imports", ) + GRADE_CHOICES=( + ('A', 'A: Large obvious lead'), + ('B', 'B: Average lead'), + ('C', 'C: Tight unpromising lead'), + ('D', 'D: Dig'), + ('X', 'X: Unclimbable aven') + ) # also seen "?" and "V" in imported data - see urls.py + grade = models.CharField(max_length=1, choices=GRADE_CHOICES) + location_description = models.TextField(blank=True) + nearest_station_description = models.CharField(max_length=400,blank=True, null=True) + nearest_station_name = models.CharField(max_length=200,blank=True, null=True) + nearest_station = models.ForeignKey(SurvexStation,blank=True, null=True,on_delete=models.SET_NULL) + area = models.CharField(max_length=100,blank=True, null=True) + completion_description = models.TextField(blank=True,null=True) + comment=models.TextField(blank=True,null=True) + + def __str__(self): + return f'{self.code()}' + + def code(self): + if self.cave: + cavestr = str(self.cave.slug())[5:] + else: + cavestr = "" + if self.expoyear: + expoyearstr = str(self.expoyear) + else: + expoyearstr = str(self.cave.slug())[5:9] + if self.blockname: + blocknamestr = "-" + str(self.blockname) + else: + blocknamestr = "" + return f'{cavestr}-{expoyearstr}-{self.number}{self.grade}{blocknamestr}' + + def get_completion_url(self): + '''assumes html file named is in same folder as cave description file + ''' + cd = None + if self.completion_description: + try: + dir = Path(self.cave.url).parent + cd = dir / self.completion_description + except: + cd = None + return cd + + def newslug(self): + qmslug = f'{str(self.cave)}-{self.expoyear}-{self.blockname}{self.number}{self.grade}' + return qmslug + + def get_absolute_url(self): + # This reverse resolution stuff is pure magic. Just change the regex in urls.py and everything changes to suit. Whacky. + return urljoin(settings.URL_ROOT, reverse('qm',kwargs={'cave_id':self.cave.slug(),'year':self.expoyear, 'blockname':self.blockname,'qm_id':self.number,'grade':self.grade})) + + def get_next_by_id(self): + return QM.objects.get(id=self.id+1) + + def get_previous_by_id(self): + return QM.objects.get(id=self.id-1) + |