async def add_user(self, user_to_add: user.User) -> user.User:
session = Session()
session.add(user_to_add)
session.commit()
session.refresh(user_to_add)
session.close()
return user_to_add
async def get_bot_instances(
self, user_id: int) -> List[bot_instance.BotInstance]:
session = Session()
bot_instances_from_db = session.query(
bot_instance.BotInstance).filter(user.User.id == user_id).all()
session.close()
return bot_instances_from_db
async def get_exchange_instances(
self, user_id: int) -> List[exchange_instance.ExchangeInstance]:
session = Session()
exchange_instances_from_db = session.query(
exchange_instance.ExchangeInstance).filter(
user.User.id == user_id).all()
session.close()
return exchange_instances_from_db
async def get_bot_instance(
self, user_id: int,
bot_instance_id: int) -> bot_instance.BotInstance:
session = Session()
bot_instance_from_db = session.query(bot_instance.BotInstance).filter(
bot_instance.BotInstance.id == bot_instance_id).first()
session.close()
return bot_instance_from_db
async def add_bot_instance(
self, user_id: int, bot_instance_to_add: bot_instance.BotInstance
) -> bot_instance.BotInstance:
session = Session()
session.add(bot_instance_to_add)
session.commit()
session.refresh(bot_instance_to_add)
session.close()
return bot_instance_to_add
async def get_exchange_instance(
self, user_id: int,
exchange_instance_id: int) -> exchange_instance.ExchangeInstance:
session = Session()
exchange_instance_from_db = session.query(
exchange_instance.ExchangeInstance).filter(
exchange_instance.ExchangeInstance.id ==
exchange_instance_id).first()
session.close()
return exchange_instance_from_db
async def add_exchange_instance(
self, user_id: int,
exchange_instance_to_add: exchange_instance.ExchangeInstance
) -> exchange_instance.ExchangeInstance:
session = Session()
session.add(exchange_instance_to_add)
session.commit()
session.refresh(exchange_instance_to_add)
session.close()
return exchange_instance_to_add
async def remove_bot_instance(
self, user_id: int, bot_instance_to_remove: bot_instance.BotInstance
) -> bot_instance.BotInstance:
session = Session()
session.delete(bot_instance_to_remove)
session.commit()
session.close()
return bot_instance_to_remove
async def remove_exchange_instance(
self, user_id: int,
exchange_instance_to_remove: exchange_instance.ExchangeInstance
) -> exchange_instance.ExchangeInstance:
session = Session()
session.delete(exchange_instance_to_remove)
session.commit()
session.close()
return exchange_instance_to_remove
async def get_all_bots(self) -> List[bot.Bot]:
session = Session()
bots_from_db = session.query(bot.Bot).all()
session.close()
return bots_from_db
async def get_all_exchanges(self) -> List[exchange.Exchange]:
session = Session()
exchanges_from_db = session.query(exchange.Exchange).all()
session.close()
return exchanges_from_db
async def get_user_from_email(self, email: str) -> user.User:
session = Session()
user_from_db = session.query(
user.User).filter(user.User.email == email).first()
session.close()
return user_from_db