import datetime import json import operator import re from functools import reduce from pathlib import Path from urllib.parse import urljoin from django.conf import settings from django.db import models from django.urls import reverse # from troggle.core.models.survex import SurvexBlock # from troggle.core.models.troggle import DataIssue # circular import. Hmm YEAR_RANGE = (1975, 2050) class Wallet(models.Model): """We do not keep the JSON values in the database, we query them afresh each time, but we will change this when we need to do a Django query on e.g. personame """ fpath = models.CharField(max_length=200) walletname = models.CharField(max_length=200) walletdate = models.DateField(blank=True, null=True) walletyear = models.DateField(blank=True, null=True) class Meta: ordering = ("walletname",) def get_absolute_url(self): return urljoin(settings.URL_ROOT, reverse("singlewallet", kwargs={"path": re.sub("#", "%23", self.walletname)})) def get_json(self): """Read the JSON file for the wallet and do stuff Do it every time it is queried, to be sure the result is fresh""" # jsonfile = Path(self.fpath, 'contents.json') # Get from git repo instead # :drawings: walletjson/2022/2022#01/contents.json # fpath = /mnt/d/EXPO/expofiles/surveyscans/1999/1999#02 fp = Path(self.fpath) wname = fp.name wyear = fp.parent.name wurl = f"/walletedit/{self.walletname}" # .replace('#', ':') if len(wyear) != 4 or len(wname) !=6: # no contents.json for old-style wallets # but this ruined all the tick-list displays.. why?! # return None pass jsonfile = Path(settings.DRAWINGS_DATA, "walletjson") / wyear / wname / "contents.json" if not Path(jsonfile).is_file(): print(f'{jsonfile} is not a file {wyear=} {wname=} ') return None else: with open(jsonfile) as json_f: try: waldata = json.load(json_f) except: message = f"! {str(self.walletname)} Failed to load {jsonfile} JSON file" # print(message) raise if waldata["date"]: datestr = waldata["date"].replace(".", "-") try: thisdate = datetime.date.fromisoformat(datestr) except ValueError: # probably a single digit day number. HACKUS MAXIMUS. # clearly we need to fix this when we first import date strings.. datestr = datestr[:-1] + "0" + datestr[-1] print(f" - {datestr=} ") try: thisdate = datetime.date.fromisoformat(datestr) self.walletdate = thisdate self.save() try: waldata["date"] = thisdate.isoformat() except: message = f"! {str(self.walletname)} Date formatting failure {thisdate}. Failed to load from {jsonfile} JSON file" from troggle.core.models.troggle import DataIssue DataIssue.objects.update_or_create(parser="scans", message=message, url=wurl) except: message = f"! {str(self.walletname)} Date format not ISO {datestr}. Failed to load from {jsonfile} JSON file" from troggle.core.models.troggle import DataIssue DataIssue.objects.update_or_create(parser="scans", message=message, url=wurl) return waldata def year(self): """This gets the year syntactically without opening and reading the JSON""" if len(self.walletname) < 5: return None if self.walletname[4] != "#": return None year = int(self.walletname[0:4]) ymin, ymax = YEAR_RANGE if year < ymin or year > ymax: return None else: self.walletyear = datetime.date(year, 1, 1) self.save() return str(year) # Yes this is horribly, horribly inefficient, esp. for a page that have date, people and cave in it def date(self): """Reads all the JSON data just to get the JSON date.""" if self.walletdate: return self.walletdate if not (jsondata := self.get_json()): # WALRUS return None datestr = jsondata["date"] if not datestr: return None else: datestr = datestr.replace(".", "-") try: samedate = datetime.date.fromisoformat(datestr) self.walletdate = samedate.isoformat() except: try: samedate = datetime.date.fromisoformat(datestr[:10]) self.walletdate = samedate.isoformat() except: samedate = None self.save() return self.walletdate def people(self): if not self.get_json(): return None jsondata = self.get_json() return jsondata["people"] def cave(self): if not self.get_json(): return None jsondata = self.get_json() return jsondata["cave"] def name(self): if not self.get_json(): return None jsondata = self.get_json() return jsondata["name"] def get_fnames(self): '''Filenames without the suffix, i.e. without the ".jpg"''' dirpath = Path(settings.SCANS_ROOT, self.fpath) # does nowt as fpath is a rooted path already files = [] if not self.fpath: files.append(f"Incorrect path to wallet contents: '{self.fpath}'") return files if not dirpath.is_dir(): files.append(f"Incorrect path to wallet contents: '{self.fpath}'") return files else: try: for f in dirpath.iterdir(): if f.is_file(): files.append(Path(f.name).stem) else: files.append(f"-{Path(f.name).stem}-") except FileNotFoundError: files.append("FileNotFoundError") pass return files def fixsurvextick(self, tick): blocks = self.survexblock_set.all() # blocks = SurvexBlock.objects.filter(scanswallet = self) result = tick for b in blocks: if b.survexfile: # if any exist in db, no check for validity or a real file. Refactor. result = "seagreen" # slightly different shade of green return result def get_ticks(self): """Reads all the JSON data and sets the colour of the completion tick for each condition""" ticks = {} waldata = self.get_json() if not waldata: ticks["S"] = "darkgrey" ticks["C"] = "darkgrey" ticks["Q"] = "darkgrey" ticks["N"] = "darkgrey" ticks["P"] = "darkgrey" ticks["E"] = "darkgrey" ticks["T"] = "darkgrey" ticks["W"] = "darkgrey" return ticks ticks = {} # Initially, are there any required survex files present ? # Note that we can't set the survexblock here on the wallet as that info is only available while parsing the survex file survexok = "red" ticks["S"] = "red" if waldata["survex not required"]: survexok = "green" ticks["S"] = "green" else: if waldata["survex file"]: if not type(waldata["survex file"]) == list: # a string also is a sequence type, so do it this way waldata["survex file"] = [waldata["survex file"]] ngood = 0 nbad = 0 ticks["S"] = "purple" for sx in waldata["survex file"]: # this logic appears in several places, inc uploads.py). Refactor. if sx != "": if Path(sx).suffix.lower() != ".svx": sx = sx + ".svx" if (Path(settings.SURVEX_DATA) / sx).is_file(): ngood += 1 else: nbad += 1 if nbad == 0 and ngood >= 1: # all valid ticks["S"] = "green" elif nbad >= 1 and ngood >= 1: # some valid, some invalid ticks["S"] = "orange" elif nbad >= 1 and ngood == 0: # all bad ticks["S"] = "red" elif nbad == 0 and ngood == 0: # list of blank strings ticks["S"] = "red" else: ticks["S"] = "fuchsia" # have fun working out what this means # Cave Description if waldata["description written"]: ticks["C"] = "green" else: ticks["C"] = survexok # QMs if waldata["qms written"]: ticks["Q"] = "green" else: ticks["Q"] = survexok if not self.year(): ticks["Q"] = "darkgrey" else: if int(self.year()) < 2015: ticks["Q"] = "lightgrey" if 'notes not required' not in waldata: waldata['notes not required'] = False # Notes, Plan, Elevation; Tunnel if waldata["electronic survey"]: ticks["N"] = "green" ticks["P"] = "green" ticks["E"] = "green" ticks["T"] = "green" else: files = self.get_fnames() # Notes required notes_scanned = reduce(operator.or_, [f.startswith("note") for f in files], False) notes_scanned = reduce(operator.or_, [f.endswith("notes") for f in files], notes_scanned) notes_required = not (notes_scanned or waldata["notes not required"]) if notes_required: ticks["N"] = "red" else: ticks["N"] = "green" # print(f"{self.walletname} {ticks['N'].upper()} {notes_scanned=} {notes_required=} {waldata['notes not required']=}") # Plan drawing required plan_scanned = reduce(operator.or_, [f.startswith("plan") for f in files], False) plan_scanned = reduce(operator.or_, [f.endswith("plan") for f in files], plan_scanned) plan_drawing_required = not (plan_scanned or waldata["plan drawn"] or waldata["plan not required"]) if plan_drawing_required: ticks["P"] = "red" else: ticks["P"] = "green" # Elev drawing required elev_scanned = reduce(operator.or_, [f.startswith("elev") for f in files], False) elev_scanned = reduce(operator.or_, [f.endswith("elev") for f in files], elev_scanned) elev_scanned = reduce(operator.or_, [f.endswith("elevation") for f in files], elev_scanned) elev_drawing_required = not (elev_scanned or waldata["elev drawn"] or waldata["elev not required"]) if elev_drawing_required: ticks["E"] = "red" else: ticks["E"] = "green" # Tunnel / Therion if elev_drawing_required or plan_drawing_required: ticks["T"] = "red" else: ticks["T"] = "green" # Website if waldata["website updated"]: ticks["W"] = "green" else: ticks["W"] = "red" return ticks def __str__(self): return "[" + str(self.walletname) + " (Wallet)]"