kopia lustrzana https://github.com/glidernet/ogn-python
90 wiersze
3.3 KiB
Python
90 wiersze
3.3 KiB
Python
from sqlalchemy import Date
|
|
from sqlalchemy import and_, insert, update, exists, between
|
|
from sqlalchemy.sql import func, null
|
|
|
|
from app.model import AircraftBeacon, ReceiverCoverage
|
|
from app.utils import date_to_timestamps
|
|
|
|
from app 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
|