rr_verification_bot/app/database.py

"""Database functons"""

from app import SESSION
from app.models import Player, TelegramAccount, TelegramHandle, PlayerTelegram, TelegramVerification


def get_telegram_account(telegram_id):
    """Get Telegram account"""
    session = SESSION()
    telegram_account = _get_telegram_account(session, telegram_id)
    session.close()
    return telegram_account

def _get_telegram_account(session, telegram_id):
    """Return telegram_account"""
    return session.query(TelegramAccount).get(telegram_id)

def get_rr_accounts(telegram_id):
    """Get Rival Region accounts associated with Telegram account"""
    session = SESSION()
    telegram_account = _get_telegram_account(session, telegram_id)
    rr_accounts = session.query(Player) \
        .join(Player.player_telegram) \
        .filter(PlayerTelegram.telegram_id == telegram_account.id) \
        .all()
    session.close()
    return rr_accounts