def _get_data(self):
from dal.dao import Dao
from tests.ledger_data.test_data import invoices_q2_pre_query
from models.employee import Employee
from models.department import Department
from models.grant_admin import GrantAdmin
from models.invoice import Invoice
from models.assignment import Assignment
dao = Dao(db_path=self.db_path, stateful=True)
self._emp_rex = Employee.get_all(dao)
self._dept_rex = Department.get_all(dao)
self._grant_admin_rex = GrantAdmin.get_all(dao)
invoices_unpaid = Invoice.get_rex(dao)
# invoices_unpaid = invoices_q2
self._invoices_sent = [
invoice for invoice in invoices_unpaid if invoice.sent
]
self._invoices_unsent = [
invoice for invoice in invoices_unpaid if not invoice.sent
]
self._asn_rex = Assignment.get_billables(dao)
dao.close()
