moanos/Notfellchen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Add function to convert a search subscription to a search

Browse Source
This commit is contained in:
moanos [he/him] 2024-12-31 13:28:41 +01:00
parent fca5879aeb
commit 280eb83056
1 changed files with 19 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/fellchensammlung/tools/search.py
View File

@ -20,7 +20,7 @@ def notify_search_subscribers(adoption_notice: AdoptionNotice):
text=notification_text) text=notification_text)
class Search(): class Search:
def __init__(self): def __init__(self):
self.sex = None self.sex = None
self.area_search = None self.area_search = None
@ -37,6 +37,8 @@ class Search():
def __eq__(self, other): def __eq__(self, other):
""" """
Custom equals that also supports SearchSubscriptions Custom equals that also supports SearchSubscriptions
Only allowed to be called for located subscriptions
""" """
return self.location.name == other.location.name and self.sex == other.sex and self.max_distance == other.max_distance return self.location.name == other.location.name and self.sex == other.sex and self.max_distance == other.max_distance
@ -60,11 +62,10 @@ class Search():
def get_adoption_notices(self): def get_adoption_notices(self):
adoptions = AdoptionNotice.objects.order_by("-created_at") adoptions = AdoptionNotice.objects.order_by("-created_at")
# Filter for active adoption notices
adoptions = [adoption for adoption in adoptions if adoption.is_active] adoptions = [adoption for adoption in adoptions if adoption.is_active]
if self.sex is not None and self.sex != SexChoicesWithAll.ALL: # Check if adoption notice fits search.
adoptions = [adoption for adoption in adoptions if self.sex in adoption.sexes] adoptions = [adoption for adoption in adoptions if self.adoption_notice_fits_search(adoption)]
if self.area_search and not self.place_not_found:
adoptions = [a for a in adoptions if a.in_distance(self.search_position, self.max_distance)]
return adoptions return adoptions
@ -86,6 +87,19 @@ class Search():
else: else:
self.search_form = AdoptionNoticeSearchForm() self.search_form = AdoptionNoticeSearchForm()
@staticmethod
def search_from_search_subscription(search_subscription: SearchSubscription):
search = Search()
search.sex = search_subscription.sex
search.location = search_subscription.location
search.search_position = (search_subscription.location.latitude, search_subscription.location.longitude)
search.area_search = True
search.max_distance = search_subscription.max_distance
return search
def subscribe(self, user): def subscribe(self, user):
logging.info(f"{user} subscribed to search") logging.info(f"{user} subscribed to search")
self._locate() self._locate()