66 lines
2.9 KiB
Python
66 lines
2.9 KiB
Python
from django.utils import timezone
|
|
from datetime import timedelta
|
|
|
|
from fellchensammlung.models import AdoptionNotice, Location, RescueOrganization, AdoptionNoticeStatus
|
|
|
|
|
|
def clean_locations(quiet=True):
|
|
# ADOPTION NOTICES
|
|
adoption_notices_without_location = AdoptionNotice.objects.filter(location__isnull=True)
|
|
num_of_all = AdoptionNotice.objects.count()
|
|
num_without_location = adoption_notices_without_location.count()
|
|
if not quiet:
|
|
print(f"From {num_of_all} there are {num_without_location} adoption notices without location "
|
|
f"({num_without_location / num_of_all * 100:.2f}%)")
|
|
for adoption_notice in adoption_notices_without_location:
|
|
if not quiet:
|
|
print(f"Searching {adoption_notice.location_string} in Nominatim")
|
|
location = Location.get_location_from_string(adoption_notice.location_string)
|
|
if location:
|
|
adoption_notice.location = location
|
|
adoption_notice.save()
|
|
|
|
adoption_notices_without_location_new = AdoptionNotice.objects.filter(location__isnull=True)
|
|
num_without_location_new = adoption_notices_without_location_new.count()
|
|
num_new = num_without_location - num_without_location_new
|
|
if not quiet:
|
|
print(f"Added {num_new} new locations")
|
|
|
|
# RESCUE ORGANIZATIONS
|
|
rescue_orgs_without_location = RescueOrganization.objects.filter(location__isnull=True)
|
|
num_of_all = RescueOrganization.objects.count()
|
|
num_without_location = rescue_orgs_without_location.count()
|
|
if not quiet:
|
|
print(f"From {num_of_all} there are {num_without_location} adoption notices without location "
|
|
f"({num_without_location / num_of_all * 100:.2f}%)")
|
|
for rescue_org in rescue_orgs_without_location:
|
|
if not quiet:
|
|
print(f"Searching {rescue_org.location_string} in Nominatim")
|
|
location = Location.get_location_from_string(rescue_org.location_string)
|
|
if location:
|
|
rescue_org.location = location
|
|
rescue_org.save()
|
|
|
|
rescue_orgs_without_location_new = RescueOrganization.objects.filter(location__isnull=True)
|
|
num_without_location_new = rescue_orgs_without_location_new.count()
|
|
num_new = num_without_location - num_without_location_new
|
|
if not quiet:
|
|
print(f"Added {num_new} new locations")
|
|
|
|
|
|
def get_unchecked_adoption_notices(weeks=3):
|
|
now = timezone.now()
|
|
three_weeks_ago = now - timedelta(weeks=weeks)
|
|
|
|
# Query for active adoption notices that were checked in the last three weeks
|
|
unchecked_adoptions = AdoptionNotice.objects.filter(
|
|
last_checked__gte=three_weeks_ago
|
|
)
|
|
active_unchecked_adoptions = [adoption for adoption in unchecked_adoptions if adoption.is_active]
|
|
return active_unchecked_adoptions
|
|
|
|
|
|
def deactivate_unchecked_adoption_notices():
|
|
for adoption_notice in get_unchecked_adoption_notices(weeks=3):
|
|
AdoptionNoticeStatus.objects.get(adoption_notice=adoption_notice).set_unchecked()
|