def insert_user(skylines_key, ogn_address, add_device=False):
"""
Arguments:
skylines_key -- hexadecimal string (4 bytes)
ogn_address -- hexadecimal string (3 bytes)
add_device -- boolean
raises:
NoResultFound
IntegrityError
"""
validate_skylines_key(skylines_key)
validate_ogn_address(ogn_address)
try:
session.query(Device).filter(Device.ogn_address == ogn_address).one()
except NoResultFound as e:
if not add_device:
raise e
else:
device = Device(ogn_address=ogn_address)
session.add(device)
user = User(ogn_address=ogn_address, skylines_key=int(skylines_key, 16))
try:
session.add(user)
session.commit()
except IntegrityError as e:
raise e
return user.as_dict()
def import_ddb():
"""Import registered devices from the DDB (discards all devices before import)."""
session.query(Device).delete()
print("Import registered devices fom the DDB...")
for device in get_ddb():
if device['identified'] and device['tracked']:
session.add(Device(ogn_address=device['device_id']))
session.commit()
print("Imported {} devices.".format(session.query(Device).count()))
def show_users():
"""
raises:
NoResultFound
"""
users = session.query(User).all()
if not len(users):
raise NoResultFound
return list(map(lambda user: user.as_dict(), users))
def show_seen_devices():
"""
raises:
NoResultFound
"""
devices = session.query(Device).filter(Device.timestamp != None).all() # noqa
if not len(devices):
raise NoResultFound
return list(map(lambda device: device.as_dict(), devices))
def show_user(skylines_key):
"""
Arguments:
skylines_key -- hexadecimal string (4 bytes)
raises:
NoResultFound
"""
validate_skylines_key(skylines_key)
users = session.query(User).filter(User.skylines_key == int(skylines_key, 16)).all()
if not len(users):
raise NoResultFound
return list(map(lambda user: user.as_dict(), users))
def show_nearby_devices(lat, lon, r, n):
"""
Return nearby devices.
Arguments:
lat - Latitude in degrees
lon - Longitude in degrees
r - Search radius in km
n - output list length limit
"""
user_location = Location(lon=lon, lat=lat).to_wkt()
distance = ST_Distance_Sphere(Device.location_wkb, user_location) / 1000
direction = func.degrees(func.ST_Azimuth(user_location, Device.location_wkb))
devices = session.query(Device,
direction.label('direction'),
distance.label('distance')).filter(distance < r).order_by(distance.asc()).limit(n).all()
if not len(devices):
raise NoResultFound
return list(map(lambda device: {'device': device.Device.as_dict(),
'direction': round(device.direction, 2),
'distance': round(device.distance, 3)}, devices))
