kopia lustrzana https://github.com/glidernet/ogn-python
87 wiersze
3.6 KiB
Python
87 wiersze
3.6 KiB
Python
from sqlalchemy import Date
|
|
from sqlalchemy import and_, insert, update, exists, between
|
|
from sqlalchemy.sql import func, null
|
|
|
|
from ogn_python.model import AircraftBeacon, ReceiverCoverage
|
|
from ogn_python.utils import date_to_timestamps
|
|
|
|
from ogn_python import app
|
|
|
|
|
|
def update_entries(session, date, logger=None):
|
|
"""Create receiver coverage stats for Melissas ognrange."""
|
|
|
|
if logger is None:
|
|
logger = app.logger
|
|
|
|
logger.info("Compute receiver coverages.")
|
|
|
|
(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()
|
|
|
|
finish_message = "ReceiverCoverage: {} inserted, {} updated".format(insert_counter, update_counter)
|
|
logger.debug(finish_message)
|
|
return finish_message
|