update toque calculating

This commit is contained in:
evgeniy-st 2019-11-01 12:57:57 +03:00
parent 04afd4d913
commit 0e04b9d044

View File

@ -309,6 +309,7 @@ class Establishment(ProjectBaseMixin, URLImageMixin, TranslatedFieldsMixin):
public_mark = models.PositiveIntegerField(blank=True, null=True,
default=None,
verbose_name=_('public mark'),)
# todo: set default 0
toque_number = models.PositiveIntegerField(blank=True, null=True,
default=None,
verbose_name=_('toque number'),)
@ -383,7 +384,7 @@ class Establishment(ProjectBaseMixin, URLImageMixin, TranslatedFieldsMixin):
# todo: recalculate toque_number
def recalculate_toque_number(self):
toque_number = None
toque_number = 0
if self.address and self.public_mark:
toque_number = RatingStrategy.objects. \
get_toque_number(country=self.address.city.country,
@ -687,7 +688,7 @@ class RatingStrategyManager(models.Manager):
obj = qs.for_public_mark(public_mark).first()
if obj:
return obj.toque_number
return None
return 0
class RatingStrategyQuerySet(models.QuerySet):