def _generate_subscription(user=None, with_broker_queue=False):
subscription = make_subscription(user=user)
db_save(session, subscription)
if with_broker_queue:
broker.create_queue_for_topic(subscription.queue,
subscription.topic.name)
return subscription
def create_route(route: Route) -> Route:
"""
Creates a new Route object in DB.
:param route:
:return:
"""
return db_save(db.session, route)
def update_route(route: Route) -> Route:
"""
Updates an Route object in DB.
:param route:
:return:
"""
return db_save(db.session, route)
def update_route_segment(route_segment: RouteSegment) -> RouteSegment:
"""
Updates an RouteSegment object in DB.
:param route_segment:
:return:
"""
return db_save(db.session, route_segment)
def update_point(point: Point) -> Point:
"""
Updates an Point object in DB.
:param point:
:return:
"""
return db_save(db.session, point)
def create_route_segment(route_segment: RouteSegment) -> RouteSegment:
"""
Creates a new RouteSegment object in DB.
:param route_segment:
:return:
"""
return db_save(db.session, route_segment)
def create_point(point: Point) -> Point:
"""
Creates a new Point object in DB.
:param point:
:return:
"""
return db_save(db.session, point)
def update_airport_heliport(
airport_heliport: AirportHeliport) -> AirportHeliport:
"""
Updates an AirportHeliport object in DB.
:param airport_heliport:
:return:
"""
return db_save(db.session, airport_heliport)
def update_topic(topic: Topic) -> Topic:
return db_save(db.session, topic)
def save_user(user: User) -> User:
return db_save(db.session, user)
def _generate_route_segment():
route_segment = make_route_segment(POINT_TYPE.NAVAID)
return db_save(session, route_segment)
def update_subscription(subscription: Subscription) -> Subscription:
return db_save(db.session, subscription)
def test_user(session):
user = make_user(uuid4().hex, DEFAULT_LOGIN_PASS)
return db_save(session, user)
def _save(obj):
try:
db_save(db.session, obj)
except Exception as e:
_logger.error(f"Couldn't save object {obj}: {str(e)}")
def _generate_subscription(user=None):
subscription = make_subscription(user=user)
return db_save(session, subscription)
def _generate_topic(name, user=None):
topic = make_topic(name, user=user)
return db_save(session, topic)
def _generate_topic(name):
topic = make_topic(name)
return db_save(session, topic)
def _generate_user(username, password, is_admin=False):
user = make_user(username=username,
password=password,
is_admin=is_admin)
return db_save(session, user)
def _generate_airport_heliport():
airport_heliport = make_airport_heliport()
return db_save(session, airport_heliport)
def _generate_user():
user = make_user()
return db_save(session, user)
def _generate_point(point_type=POINT_TYPE.DESIGNATED_POINT):
point = make_point(point_type)
return db_save(session, point)
def _generate_route():
route = make_route()
return db_save(session, route)
def _generate_user(is_admin=False):
user = make_user(is_admin=is_admin)
return db_save(session, user)
def test_admin_user(session):
user = make_user(uuid4().hex, DEFAULT_LOGIN_PASS, is_admin=True)
return db_save(session, user)
def create_subscription(subscription: Subscription) -> Subscription:
if subscription.queue is None:
subscription.queue = generate_queue()
return db_save(db.session, subscription)
