ogn-python/ogn_python/commands/export.py

114 wiersze
3.9 KiB
Python

from flask.cli import AppGroup
import click
import datetime
import re
import csv
from aerofiles.igc import Writer
from ogn_python.model import *
from ogn_python import db
user_cli = AppGroup('export')
user_cli.help = "Export data in several file formats."
@user_cli.command('cup')
def cup():
"""Export receiver waypoints as '.cup'."""
sql = """
SELECT
'OGN-' || sq.name AS name,
sq.name AS code,
c.iso2 AS country,
CASE WHEN sq.lat_deg < 10 THEN '0' ELSE '' END || CAST((sq.lat_deg*100 + sq.lat_min) AS decimal(18, 5)) || sq.lat_sig AS lat,
CASE WHEN sq.lon_deg < 10 THEN '00' WHEN sq.lon_deg < 100 THEN '0' ELSE '' END || CAST(sq.lon_deg*100 + sq.lon_min AS decimal(18, 5)) || sq.lon_sig AS lon,
altitude || 'm' AS elev,
'8' AS style,
'' AS rwdir,
'' AS rwlen,
'' AS freq,
'lastseen: ' || sq.lastseen::date || ', version: ' || sq.version || ', platform: ' || sq.platform AS desc
FROM (
SELECT
st_y(location) as lat,
CASE WHEN ST_Y(location) > 0 THEN 'N' ELSE 'S' END AS lat_sig,
FLOOR(ABS(ST_Y(location))) AS lat_deg,
60*(ABS(ST_Y(location)) - FLOOR(ABS(ST_Y(location)))) AS lat_min,
st_x(location) AS lon,
CASE WHEN ST_X(location) > 0 THEN 'E' ELSE 'W' END AS lon_sig,
FLOOR(ABS(ST_X(location))) AS lon_deg,
60*(ABS(ST_X(location)) - FLOOR(ABS(ST_X(location)))) AS lon_min
, *
FROM receivers
WHERE lastseen - firstseen > INTERVAL'3 MONTH' AND lastseen > '2018-01-01 00:00:00' AND name NOT LIKE 'FNB%'
) sq
INNER JOIN countries c ON c.gid = sq.country_id
ORDER BY sq.name;
"""
results = db.session.execute(sql)
with open('receivers.cup', 'w') as outfile:
outcsv = csv.writer(outfile)
outcsv.writerow(results.keys())
outcsv.writerows(results.fetchall())
@user_cli.command('igc')
@click.argument('address')
@click.argument('date')
def igc(address, date):
"""Export igc file for <address> at <date>."""
if not re.match('.{6}', address):
print("Address {} not valid.".format(address))
return
if not re.match('\d{4}-\d{2}-\d{2}', date):
print("Date {} not valid.".format(date))
return
device_id = db.session.query(Device.id) \
.filter(Device.address == address) \
.first()
if (device_id is None):
print("Device with address '{}' not found.".format(address))
return
with open('sample.igc', 'wb') as fp:
writer = Writer(fp)
writer.write_headers({
'manufacturer_code': 'OGN',
'logger_id': 'OGN',
'date': datetime.date(1987, 2, 24),
'fix_accuracy': 50,
'pilot': 'Konstantin Gruendger',
'copilot': '',
'glider_type': 'Duo Discus',
'glider_id': 'D-KKHH',
'firmware_version': '2.2',
'hardware_version': '2',
'logger_type': 'LXNAVIGATION,LX8000F',
'gps_receiver': 'uBLOX LEA-4S-2,16,max9000m',
'pressure_sensor': 'INTERSEMA,MS5534A,max10000m',
'competition_id': '2H',
'competition_class': 'Doubleseater',
})
points = db.session.query(AircraftBeacon) \
.filter(AircraftBeacon.device_id == device_id) \
.filter(AircraftBeacon.timestamp > date + ' 00:00:00') \
.filter(AircraftBeacon.timestamp < date + ' 23:59:59') \
.order_by(AircraftBeacon.timestamp)
for point in points.all():
writer.write_fix(
point.timestamp.time(),
latitude=point.location.latitude,
longitude=point.location.longitude,
valid=True,
pressure_alt=point.altitude,
gps_alt=point.altitude,
)