kopia lustrzana https://github.com/glidernet/ogn-python
94 wiersze
3.8 KiB
Python
94 wiersze
3.8 KiB
Python
from celery.utils.log import get_task_logger
|
|
|
|
from sqlalchemy import Date
|
|
from sqlalchemy import and_, insert, update, exists, between
|
|
from sqlalchemy.sql import func, null
|
|
|
|
from ogn_python.collect.celery import app
|
|
from ogn_python.model import AircraftBeacon, ReceiverCoverage
|
|
from ogn_python.utils import date_to_timestamps
|
|
|
|
logger = get_task_logger(__name__)
|
|
|
|
|
|
@app.task
|
|
def create_receiver_coverage(session=None, date=None):
|
|
"""Create receiver coverages."""
|
|
|
|
logger.info("Compute receiver coverages.")
|
|
|
|
if session is None:
|
|
session = app.session
|
|
|
|
if not date:
|
|
logger.warn("A date is needed for calculating stats. Exiting")
|
|
return None
|
|
else:
|
|
(start, end) = date_to_timestamps(date)
|
|
|
|
# Filter aircraft beacons
|
|
sq = session.query(AircraftBeacon.location_mgrs_short,
|
|
AircraftBeacon.receiver_id,
|
|
AircraftBeacon.signal_quality,
|
|
AircraftBeacon.altitude,
|
|
AircraftBeacon.device_id) \
|
|
.filter(and_(between(AircraftBeacon.timestamp, start, end),
|
|
AircraftBeacon.location_mgrs_short != null(),
|
|
AircraftBeacon.receiver_id != null(),
|
|
AircraftBeacon.device_id != null())) \
|
|
.subquery()
|
|
|
|
# ... and group them by reduced MGRS, receiver and date
|
|
query = session.query(sq.c.location_mgrs_short,
|
|
sq.c.receiver_id,
|
|
func.cast(date, Date).label('date'),
|
|
func.max(sq.c.signal_quality).label('max_signal_quality'),
|
|
func.min(sq.c.altitude).label('min_altitude'),
|
|
func.max(sq.c.altitude).label('max_altitude'),
|
|
func.count(sq.c.altitude).label('aircraft_beacon_count'),
|
|
func.count(func.distinct(sq.c.device_id)).label('device_count')) \
|
|
.group_by(sq.c.location_mgrs_short,
|
|
sq.c.receiver_id) \
|
|
.subquery()
|
|
|
|
# if a receiver coverage entry exist --> update it
|
|
upd = update(ReceiverCoverage) \
|
|
.where(and_(ReceiverCoverage.location_mgrs_short == query.c.location_mgrs_short,
|
|
ReceiverCoverage.receiver_id == query.c.receiver_id,
|
|
ReceiverCoverage.date == date)) \
|
|
.values({"max_signal_quality": query.c.max_signal_quality,
|
|
"min_altitude": query.c.min_altitude,
|
|
"max_altitude": query.c.max_altitude,
|
|
"aircraft_beacon_count": query.c.aircraft_beacon_count,
|
|
"device_count": query.c.device_count})
|
|
|
|
result = session.execute(upd)
|
|
update_counter = result.rowcount
|
|
session.commit()
|
|
logger.debug("Updated receiver coverage entries: {}".format(update_counter))
|
|
|
|
# if a receiver coverage entry doesnt exist --> insert it
|
|
new_coverage_entries = session.query(query) \
|
|
.filter(~exists().where(
|
|
and_(ReceiverCoverage.location_mgrs_short == query.c.location_mgrs_short,
|
|
ReceiverCoverage.receiver_id == query.c.receiver_id,
|
|
ReceiverCoverage.date == date)))
|
|
|
|
ins = insert(ReceiverCoverage).from_select((
|
|
ReceiverCoverage.location_mgrs_short,
|
|
ReceiverCoverage.receiver_id,
|
|
ReceiverCoverage.date,
|
|
ReceiverCoverage.max_signal_quality,
|
|
ReceiverCoverage.min_altitude,
|
|
ReceiverCoverage.max_altitude,
|
|
ReceiverCoverage.aircraft_beacon_count,
|
|
ReceiverCoverage.device_count),
|
|
new_coverage_entries)
|
|
|
|
result = session.execute(ins)
|
|
insert_counter = result.rowcount
|
|
session.commit()
|
|
logger.debug("New receiver coverage entries: {}".format(insert_counter))
|
|
|
|
return "Receiver coverage entries: {} inserted, {} updated".format(insert_counter, update_counter)
|