functions.get_balance

  1from dictionary.sql import (
  2    total_expense_query,
  3    total_revenue_query,
  4    last_expense_query,
  5    last_revenue_query,
  6    max_expense_query,
  7    max_revenue_query,
  8    future_accounts_expenses_query,
  9    future_accounts_revenue_query,
 10    accounts_revenue_query,
 11    accounts_expenses_query,
 12    accounts_query,
 13)
 14from dictionary.vars import to_remove_list, today
 15from functions.query_executor import QueryExecutor
 16from functions.login import Login
 17
 18
 19class GetBalance:
 20    """
 21    Classe responsável pela consulta dos valores do balannço de contas.
 22    """
 23
 24    def get_values(self):
 25        """
 26        Faz a consulta dos valores de despesas e receitas.
 27
 28        Returns
 29        -------
 30        total_revenues : float
 31            O valor total das receitas.
 32        total_expenses : float
 33            O valor total das despesas.
 34        """
 35
 36        user_name, user_document = Login().get_user_data(return_option="user_doc_name")
 37
 38        str_total_expenses = QueryExecutor().simple_consult_query(query=total_expense_query, params=(user_name, user_document))
 39        str_total_expenses = QueryExecutor().treat_simple_result(str_total_expenses, to_remove_list)
 40
 41        if str_total_expenses == 'None':
 42            total_expenses = 0.00
 43        else:
 44            total_expenses = float(str_total_expenses)
 45
 46        str_total_revenues = QueryExecutor().simple_consult_query(query=total_revenue_query, params=(user_name, user_document))
 47        str_total_revenues = QueryExecutor().treat_simple_result(str_total_revenues, to_remove_list)
 48
 49        if str_total_revenues == 'None':
 50            total_revenues = 0.00
 51
 52        else:
 53            total_revenues = float(str_total_revenues)
 54
 55        return total_revenues, total_expenses
 56
 57    def balance(self):
 58        """
 59        Calcula o valor do balanço.
 60
 61        Returns
 62        -------
 63        balance : float
 64            O valor do balanço.
 65        """
 66        total_revenues, total_expenses = self.get_values()
 67
 68        if total_revenues is not None and total_expenses is not None:
 69            balance = round((total_revenues - total_expenses), 2)
 70            return balance
 71        else:
 72            return None
 73
 74    def accounts_balance(self):
 75        """
 76        Realiza o cálculo do balanço das contas.
 77
 78        Returns
 79        -------
 80        accounts : list
 81            A lista com o nome das contas.
 82        balance_list : list
 83            A lista com o balanço das contas.
 84        future_balance_list : list
 85            A lista com o balanço futuro das contas.
 86        """
 87        user_name, user_document = Login().get_user_data(return_option="user_doc_name")
 88
 89        accounts_expenses = QueryExecutor().complex_consult_query(query=accounts_expenses_query, params=(today, user_name, user_document))
 90        accounts_expenses = QueryExecutor().treat_numerous_simple_result(accounts_expenses, to_remove_list)
 91
 92        accounts_revenues = QueryExecutor().complex_consult_query(query=accounts_revenue_query, params=(today, user_name, user_document))
 93        accounts_revenues = QueryExecutor().treat_numerous_simple_result(accounts_revenues, to_remove_list)
 94
 95        future_accounts_expenses = QueryExecutor().complex_consult_query(query=future_accounts_expenses_query, params=(today, user_name, user_document))
 96        future_accounts_expenses = QueryExecutor().treat_numerous_simple_result(future_accounts_expenses, to_remove_list)
 97
 98        future_accounts_revenues = QueryExecutor().complex_consult_query(query=future_accounts_revenue_query, params=(user_name, user_document))
 99        future_accounts_revenues = QueryExecutor().treat_numerous_simple_result(future_accounts_revenues, to_remove_list)
100
101        accounts = QueryExecutor().complex_consult_query(query=accounts_query, params=(user_name, user_document))
102        accounts = QueryExecutor().treat_numerous_simple_result(accounts, to_remove_list)
103
104        balance_list = []
105        future_balance_list = []
106
107        if len(accounts_revenues) == len(accounts_expenses):
108            for i in range(0, len(accounts_revenues)):
109                revenue = float(accounts_revenues[i])
110                expense = float(accounts_expenses[i])
111                balance_list.append(revenue - expense)
112
113        if len(future_accounts_expenses) == len(future_accounts_revenues):
114            for i in range(0, len(future_accounts_revenues)):
115                future_revenue = float(future_accounts_revenues[i])
116                future_expense = float(future_accounts_expenses[i])
117                future_balance_list.append(
118                    (future_revenue - future_expense) + balance_list[i])
119
120        return accounts, balance_list, future_balance_list
121
122    def list_values(self):
123        """
124        Faz a consultas das maiores e últimas despesas e receitas.
125
126        Returns
127        -------
128        last_revenues : list
129            Lista com as últimas receitas.
130        last_expenses : list
131            Lista com as últimas despesas.
132        max_revenues : list
133            Lista com as maiores receitas.
134        max_expenses : list
135            Lista com as maiores despesas.
136        """
137
138        user_name, user_document = Login().get_user_data(return_option="user_doc_name")
139        last_expenses = QueryExecutor().complex_consult_query(query=last_expense_query, params=(user_name, user_document))
140        last_revenues = QueryExecutor().complex_consult_query(query=last_revenue_query, params=(today, user_name, user_document))
141        max_revenues = QueryExecutor().complex_consult_query(query=max_revenue_query, params=(today, user_name, user_document))
142        max_expenses = QueryExecutor().complex_consult_query(query=max_expense_query, params=(user_name, user_document))
143
144        return last_revenues, last_expenses, max_revenues, max_expenses