Moved Logbooks to separate model file

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)
+