def search_account(account_id, bank=None): account = Account.select().where(Account.label.contains(account_id)) if bank is not None: bank = Bank.get_or_none(name=bank) account = account.where(Account.bank == bank) return account
def get_account_by_balance(number, bank=None): accounts = Account.select().where(Account.balance <= number) if bank is not None: bank = Bank.get_or_none(name=bank) if bank is not None: accounts = accounts.where(Account.bank == bank) return accounts
def get_banks(bank_name=None): banks = Bank.select() if bank_name is not None: db_bank = Bank.get_or_none(name=bank_name) if db_bank is None: raise BankNotFoundError(bank_name) else: return [db_bank.get_small_data()] return [bank.get_small_data() for bank in banks]
def get_accounts(bank=None): accounts = Account.select() if bank is not None: db_bank = Bank.get_or_none(name=bank) if db_bank is None: raise BankNotFoundError(bank) accounts = accounts.where(Account.bank == db_bank) return accounts
def get_accounts_types(bank=None, db_user=None): accounts_types = Account.select(fn.Distinct(Account.label.strip(Account.account_id))) if db_user is not None: accounts_types = accounts_types.where(Account.user == db_user) if bank is not None: db_bank = Bank.get_or_none(name=bank) if db_bank is None: raise BankNotFoundError(bank) accounts_types = accounts_types.where(Account.bank == db_bank) return [account.strip("()") for account in accounts_types]
def get_transactions_by_account_type(account_type, bank=None, db_user=None): transactions = (Transaction.select() .join(Account) .where(Account.label.contains(account_type))) if db_user is not None: transactions = transactions.where(Account.user == db_user) if bank is not None: db_bank = Bank.get_or_none(name=bank) if db_bank is None: raise BankNotFoundError(bank) transactions = transactions.where(Account.bank == db_bank) return transactions
def get_accounts(bank=None, name=None, db_user=None): accounts = Account.select() if name is not None: accounts = accounts.where(Account.label.contains(name)) if db_user is not None: accounts = accounts.where(Account.user == db_user) if bank is not None: db_bank = Bank.get_or_none(name=bank) if db_bank is None: raise BankNotFoundError(bank) accounts = accounts.where(Account.bank == db_bank) return accounts
def get_bank_users(bank_name=None): nb_users = (User.select(fn.count(fn.Distinct(User.id)), Bank.name).join( Account, on=(User.id == Account.user_id)).join(Bank).group_by(Bank)) if bank_name is not None: db_bank = Bank.get_or_none(name=bank_name) if db_bank is None: raise BankNotFoundError(bank_name) else: nb_users = nb_users.where(Bank.name == bank_name) print(nb_users) return [{ "name": nb_user.account.bank.name, "number of users": nb_user.count } for nb_user in nb_users]
def get_bank_transactions(bank_name=None): transactions = (Transaction.select(fn.COUNT(fn.Distinct( Transaction.id)), fn.sum(Transaction.amount), Bank.name).join( Account, on=(Transaction.account_id == Account.id)).join(Bank).group_by( Bank.name)) if bank_name is not None: db_bank = Bank.get_or_none(name=bank_name) if db_bank is None: raise BankNotFoundError(bank_name) else: transactions = transactions.where(Bank.name == bank_name) return [{ "name": transaction.account.bank.name, "number of transactions": transaction.count, "total amount": transaction.sum } for transaction in transactions]