phzhik/gault-millau
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix method

Browse Source
This commit is contained in:
Anatoly 2020-01-30 16:28:32 +03:00
parent 7541909390
commit e6de2cc4f9
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
apps/establishment/models.py
View File

@ -2,7 +2,7 @@
from datetime import datetime from datetime import datetime
from functools import reduce from functools import reduce
from operator import or_ from operator import or_
from typing import List, Union from typing import List
import elasticsearch_dsl import elasticsearch_dsl
from django.conf import settings from django.conf import settings
@ -184,10 +184,9 @@ class EstablishmentQuerySet(models.QuerySet):
"""Return establishments by country code""" """Return establishments by country code"""
return self.filter(address__city__country=country) return self.filter(address__city__country=country)
def by_country_code(self, codes: Union[iter, str]): def by_country_code(self, code: str):
"""Return establishments by country code""" """Return establishments by country code"""
codes = codes if hasattr(codes, '__iter__') else [codes] return self.filter(address__city__country__code=code)
return self.filter(address__city__country__code__in=codes)
def published(self): def published(self):
""" """