kopia lustrzana https://github.com/glidernet/ogn-python
95 wiersze
3.1 KiB
Python
95 wiersze
3.1 KiB
Python
import datetime
|
|
|
|
from celery.utils.log import get_task_logger
|
|
|
|
from ogn_python.collect.takeoff_landings import update_entries as takeoff_update_entries
|
|
|
|
from ogn_python.collect.logbook import update_entries as logbook_update_entries
|
|
from ogn_python.collect.logbook import update_max_altitudes as logbook_update_max_altitudes
|
|
|
|
from ogn_python.collect.database import import_ddb as device_infos_import_ddb
|
|
from ogn_python.collect.database import update_country_code as receivers_update_country_code
|
|
|
|
from ogn_python.collect.stats import create_device_stats, update_device_stats_jumps, create_receiver_stats, create_relation_stats, update_qualities, update_receivers, update_devices
|
|
|
|
from ogn_python import db
|
|
from ogn_python import celery
|
|
|
|
|
|
logger = get_task_logger(__name__)
|
|
|
|
|
|
@celery.task(name='update_takeoff_landings')
|
|
def update_takeoff_landings(last_minutes):
|
|
"""Compute takeoffs and landings."""
|
|
|
|
end = datetime.datetime.utcnow()
|
|
start = end - datetime.timedelta(minutes=last_minutes)
|
|
result = takeoff_update_entries(session=db.session, start=start, end=end, logger=logger)
|
|
return result
|
|
|
|
|
|
@celery.task(name='update_logbook_entries')
|
|
def update_logbook_entries():
|
|
"""Add/update logbook entries."""
|
|
|
|
today = datetime.datetime.today()
|
|
result = logbook_update_entries(session=db.session, date=today, logger=logger)
|
|
return result
|
|
|
|
|
|
@celery.task(name='update_logbook_max_altitude')
|
|
def update_logbook_max_altitude():
|
|
"""Add max altitudes in logbook when flight is complete (takeoff and landing)."""
|
|
|
|
result = logbook_update_max_altitudes(session=db.session, logger=logger)
|
|
return result
|
|
|
|
|
|
@celery.task(name='import_ddb')
|
|
def import_ddb():
|
|
"""Import registered devices from the DDB."""
|
|
|
|
result = device_infos_import_ddb(session=db.session, logger=logger)
|
|
return result
|
|
|
|
|
|
@celery.task(name='update_receivers_country_code')
|
|
def update_receivers_country_code():
|
|
"""Update country code in receivers table if None."""
|
|
|
|
result = receivers_update_country_code(session=db.session, logger=logger)
|
|
return result
|
|
|
|
|
|
@celery.task(name='purge_old_data')
|
|
def purge_old_data(max_hours):
|
|
"""Delete AircraftBeacons and ReceiverBeacons older than given 'age'."""
|
|
|
|
from ogn_python.model import AircraftBeacon, ReceiverBeacon
|
|
min_timestamp = datetime.datetime.utcnow() - datetime.timedelta(hours=max_hours)
|
|
db.session.query(AircraftBeacon) \
|
|
.filter(AircraftBeacon.timestamp < min_timestamp) \
|
|
.delete()
|
|
|
|
db.session.query(ReceiverBeacon) \
|
|
.filter(ReceiverBeacon.timestamp < min_timestamp) \
|
|
.delete()
|
|
|
|
db.session.commit()
|
|
|
|
|
|
@celery.task(name='update_stats')
|
|
def update_stats(day_offset):
|
|
"""Create stats and update receivers/devices with stats."""
|
|
|
|
date = datetime.datetime.today() + datetime.timedelta(days=day_offset)
|
|
|
|
create_device_stats(session=db.session, date=date)
|
|
update_device_stats_jumps(session=db.session, date=date)
|
|
create_receiver_stats(session=db.session, date=date)
|
|
create_relation_stats(session=db.session, date=date)
|
|
update_qualities(session=db.session, date=date)
|
|
update_receivers(session=db.session)
|
|
update_devices(session=db.session)
|