def remove_customer(location_id): init_db() session = db_session.create_session() location = session.query(Location) \ .filter(Location.id == location_id) \ .first() session.delete(location) session.commit() session.close()
def get_server_urls() -> List[Location]: init_db() session = db_session.create_session() locations = session.query(Location). \ all() session.close() return locations
def sorted_server_urls(): init_db() session = db_session.create_session() locations = session.query(Location).\ order_by(Location.ping.desc()).\ all() session.close() return locations
def does_customer_exist(location_id): init_db() session = db_session.create_session() exists = session.query( sqlalchemy.exists().where(Location.id == location_id)).scalar() if exists: print(f'Customer {location_id} exists.') if not exists: print(f'Customer {location_id} does not exists.') return exists
def update_server(friendly_name, anonymize_name, response_time, ping, location_status, url, status_color, type): init_db() session = db_session.create_session() try: s = session.query(Location).filter(Location.id == friendly_name).one() # print(f'Location {friendly_name} exists.') s.id = friendly_name s.anonymized_name = anonymize_name s.response_time = response_time s.ping = ping s.status = location_status s.url = url s.status_color = status_color s.type = type session.commit() except NoResultFound: # print(f'Location {friendly_name} does not exists.') s = Location() s.id = friendly_name s.anonymized_name = anonymize_name s.url = url s.status = location_status s.response_time = response_time s.ping = ping s.status_color = status_color s.type = type session = db_session.create_session() session.add(s) session.commit()