no longer needed to pass instance to some model methods

This commit is contained in:
Laura Klünder 2024-08-22 14:17:19 +02:00
parent aab4230a3a
commit ca9cfc1e14
6 changed files with 24 additions and 30 deletions

View file

@ -105,7 +105,7 @@ def conditional_geojson(obj, update_cache_key_match):
if update_cache_key_match and not obj._affected_by_changeset:
return obj.get_geojson_key()
result = obj.to_geojson(instance=obj)
result = obj.to_geojson()
result['properties']['changed'] = obj._affected_by_changeset
return result

View file

@ -58,10 +58,10 @@ class GeometryMixin(SerializableMixin):
def get_geojson_key(self):
return (self.__class__.__name__.lower(), self.id)
def to_geojson(self, instance=None) -> dict:
def to_geojson(self) -> dict:
result = {
'type': 'Feature',
'properties': self.get_geojson_properties(instance=instance),
'properties': self.get_geojson_properties(),
'geometry': format_geojson(smart_mapping(self.geometry), rounded=False),
}
original_geometry = getattr(self, 'original_geometry', None)

View file

@ -40,12 +40,12 @@ class LevelGeometryMixin(GeometryMixin):
class Meta:
abstract = True
def get_geojson_properties(self, *args, instance=None, **kwargs) -> dict:
def get_geojson_properties(self, *args, **kwargs) -> dict:
result = super().get_geojson_properties(*args, **kwargs)
result['level'] = self.level_id
if hasattr(self, 'get_color'):
from c3nav.mapdata.render.theme import ColorManager
color = self.get_color(ColorManager.for_theme(None), instance=instance)
color = self.get_color(ColorManager.for_theme(None))
if color:
result['color'] = color
if hasattr(self, 'opacity'):

View file

@ -39,11 +39,11 @@ class SpaceGeometryMixin(GeometryMixin):
except ObjectDoesNotExist:
return None
def get_geojson_properties(self, *args, instance=None, **kwargs) -> dict:
def get_geojson_properties(self, *args, **kwargs) -> dict:
result = super().get_geojson_properties(*args, **kwargs)
if hasattr(self, 'get_color'):
from c3nav.mapdata.render.theme import ColorManager
color = self.get_color(ColorManager.for_theme(None), instance=instance)
color = self.get_color(ColorManager.for_theme(None))
if color:
result['color'] = color
if hasattr(self, 'opacity'):
@ -226,10 +226,10 @@ class Obstacle(SpaceGeometryMixin, models.Model):
default_related_name = 'obstacles'
ordering = ('altitude', 'height')
def get_geojson_properties(self, *args, instance=None, **kwargs) -> dict:
def get_geojson_properties(self, *args, **kwargs) -> dict:
result = super().get_geojson_properties(*args, **kwargs)
from c3nav.mapdata.render.theme import ColorManager
color = self.get_color(ColorManager.for_theme(None), instance=instance)
color = self.get_color(ColorManager.for_theme(None))
if color:
result['color'] = color
return result
@ -242,12 +242,10 @@ class Obstacle(SpaceGeometryMixin, models.Model):
result['color'] = self.get_color(ColorManager.for_theme(None))
return result
def get_color(self, color_manager: 'ThemeColorManager', instance=None):
if instance is None:
instance = self
def get_color(self, color_manager: 'ThemeColorManager'):
return (
color_manager.obstaclegroup_fill_color(instance.group)
if instance.group is not None
color_manager.obstaclegroup_fill_color(self.group)
if self.group is not None
else color_manager.obstacles_default_fill
)
@ -270,10 +268,10 @@ class LineObstacle(SpaceGeometryMixin, models.Model):
default_related_name = 'lineobstacles'
ordering = ('altitude', 'height')
def get_geojson_properties(self, *args, instance=None, **kwargs) -> dict:
def get_geojson_properties(self, *args, **kwargs) -> dict:
result = super().get_geojson_properties(*args, **kwargs)
from c3nav.mapdata.render.theme import ColorManager
color = self.get_color(ColorManager.for_theme(None), instance=instance)
color = self.get_color(ColorManager.for_theme(None))
if color:
result['color'] = color
return result
@ -289,13 +287,11 @@ class LineObstacle(SpaceGeometryMixin, models.Model):
result['buffered_geometry'] = format_geojson(mapping(self.buffered_geometry))
return result
def get_color(self, color_manager: 'ThemeColorManager', instance=None):
if instance is None:
instance = self
def get_color(self, color_manager: 'ThemeColorManager'):
# TODO: should line obstacles use border color?
return (
color_manager.obstaclegroup_fill_color(instance.group)
if instance.group is not None
color_manager.obstaclegroup_fill_color(self.group)
if self.group is not None
else color_manager.obstacles_default_fill
)

View file

@ -73,7 +73,7 @@ class GraphEdge(AccessRestrictionMixin, models.Model):
default_related_name = 'graphedges'
unique_together = (('from_node', 'to_node'), )
def to_geojson(self, instance=None) -> dict:
def to_geojson(self) -> dict:
result = {
'type': 'Feature',
'properties': {

View file

@ -75,10 +75,10 @@ class LocationSlug(SerializableMixin, models.Model):
objects = LocationSlugManager()
def get_child(self, instance=None):
def get_child(self):
for model in get_submodels(Location)+[LocationRedirect]:
with suppress(AttributeError):
return getattr(instance or self, model._meta.default_related_name)
return getattr(self, model._meta.default_related_name)
return None
def get_slug(self):
@ -159,16 +159,14 @@ class Location(LocationSlug, AccessRestrictionMixin, TitledMixin, models.Model):
def grid_square(self):
return None
def get_color(self, color_manager: 'ThemeColorManager', instance=None) -> str | None:
def get_color(self, color_manager: 'ThemeColorManager') -> str | None:
# don't filter in the query here so prefetch_related works
result = self.get_color_sorted(color_manager, instance)
result = self.get_color_sorted(color_manager)
return None if result is None else result[1]
def get_color_sorted(self, color_manager: 'ThemeColorManager', instance=None) -> tuple[tuple, str] | None:
def get_color_sorted(self, color_manager: 'ThemeColorManager') -> tuple[tuple, str] | None:
# don't filter in the query here so prefetch_related works
if instance is None:
instance = self
for group in instance.groups.all():
for group in self.groups.all():
color = color_manager.locationgroup_fill_color(group)
if color and getattr(group.category, 'allow_'+self.__class__._meta.default_related_name):
return (0, group.category.priority, group.hierarchy, group.priority), color