diff options
Diffstat (limited to 'core/models/caves.py')
| -rw-r--r-- | core/models/caves.py | 19 |
1 files changed, 15 insertions, 4 deletions
diff --git a/core/models/caves.py b/core/models/caves.py index d0520fb..5bef08c 100644 --- a/core/models/caves.py +++ b/core/models/caves.py @@ -311,12 +311,23 @@ class Entrance(TroggleModel): def __str__(self): return str(self.slug) + def single(self, station): + try: + single = SurvexStation.objects.get(name = station) + return single + except: + stations = SurvexStation.objects.filter(name = station) + print(f" # MULTIPLE stations found with same name '{station}' in Entrance {self}:") + for s in stations: + print(f" # {s.id=} - {s.name} {s.latlong()}") # .id is Django internal field, not one of ours + return stations[0] + def exact_location(self): - return SurvexStation.objects.get(name = self.exact_station) + return self.single(self.exact_station) def other_location(self): - return SurvexStation.objects.get(name = self.other_station) - + return self.single(self.other_station) + def find_location(self): r = {"": "To be entered ", "?": "To be confirmed:", "S": "", "L": "Lost:", "R": "Refindable:"}[self.findability] if self.tag_station: @@ -376,7 +387,7 @@ class Entrance(TroggleModel): return f[1] def tag(self): - return SurvexStation.objects.get(name = self.tag_station) + return self.single(self.tag_station) def needs_surface_work(self): return self.findability != "S" or not self.has_photo or self.marking != "T" |