mirror of
https://github.com/task-17-lct/backend.git
synced 2024-11-24 06:43:45 +03:00
75 lines
2.8 KiB
Python
75 lines
2.8 KiB
Python
from datetime import timedelta
|
||
|
||
from celery import shared_task
|
||
from django.conf import settings
|
||
from django.core.mail import send_mail
|
||
from django.template.loader import render_to_string
|
||
from django.utils.timezone import now
|
||
from django.utils.dateparse import parse_date
|
||
|
||
from passfinder.events.models import City, UserRoute
|
||
from passfinder.events.services import get_position_weather
|
||
|
||
weathers_labels = {
|
||
"light-rain": "небольшой дождь",
|
||
"rain": "дождь",
|
||
"moderate-rain": "дождь",
|
||
"heavy-rain": "сильный дождь",
|
||
"continuous-heavy-rain": "сильный дождь",
|
||
"showers": "ливень",
|
||
"wet-snow": "дождь со снегом",
|
||
"snow-showers": "снегопад",
|
||
"hail": "град",
|
||
"thunderstorm": "гроза",
|
||
"thunderstorm-with-rain": "дождь с грозой",
|
||
"thunderstorm-with-hail": "дождь с грозой",
|
||
}
|
||
|
||
|
||
@shared_task
|
||
def check_temperature():
|
||
cities_temp = {}
|
||
|
||
for route in UserRoute.objects.filter(
|
||
dates__date__gte=now().date(), dates__date__lte=now().date() + timedelta(days=3)
|
||
):
|
||
alerts = []
|
||
for i in range(3):
|
||
date = now().date() + timedelta(days=i)
|
||
points = route.dates.objects.get(date=date).points.all()
|
||
cities = points.values_list("city_id").distinct()
|
||
for city in cities:
|
||
if city:
|
||
city = City.objects.get(oid=city)
|
||
if city in cities_temp:
|
||
weather_conditions = cities_temp[city]
|
||
else:
|
||
weather_conditions = get_position_weather(city.lat, city.lon)
|
||
weather_conditions = dict(
|
||
[(parse_date(x[0]), x[1]) for x in weather_conditions]
|
||
)
|
||
cities_temp[city] = weather_conditions
|
||
|
||
if date in weather_conditions:
|
||
weather = weather_conditions[date]
|
||
if weather in weathers_labels:
|
||
alerts.append(
|
||
f"В городе {city.title} {date.strftime('%d.%m.%Y')} предстоит {weathers_labels[weather]}"
|
||
)
|
||
if alerts:
|
||
context = {
|
||
"user": route.user,
|
||
"route": route,
|
||
"alerts": alerts,
|
||
# TODO: change to frontend link
|
||
"link": f"http://127.0.0.1:8000/path/{route.id}/change",
|
||
}
|
||
|
||
email_plaintext_message = render_to_string("weather_alert.txt", context)
|
||
send_mail(
|
||
"Предупреждение о погоде",
|
||
email_plaintext_message,
|
||
settings.DEFAULT_FROM_EMAIL,
|
||
[route.user.email],
|
||
)
|