def test_get_month_interval_for_year(self): months = list(get_month_intervals_for_year(2012)) self.assertTrue(len(months), 12) self.assertTrue(months[0], (datetime.datetime(2012, 1, 1), datetime.datetime(2012, 1, 31))) self.assertTrue(months[11], (datetime.datetime(2012, 12, 1), datetime.datetime(2012, 12, 31)))
def get_data(self): sheets = {} for account in Account.get_children_for(self.store, parent=None): if sysparam.compare_object('IMBALANCE_ACCOUNT', account): continue columns = [] for start, end in get_month_intervals_for_year(self.year): column = [] self._prepare_items(column, account, start, end) columns.append(column) # Skip empty sheets if sum(item[1] for c in columns for item in c) == 0: continue sheets[account.description] = columns return sheets
def get_data(self): sysparam_ = sysparam(self.store) sheets = {} for account in Account.get_children_for(self.store, parent=None): if account.id == sysparam_.IMBALANCE_ACCOUNT.id: continue columns = [] for start, end in get_month_intervals_for_year(self.year): column = [] self._prepare_items(column, account, start, end) columns.append(column) # Skip empty sheets if sum(item[1] for c in columns for item in c) == 0: continue sheets[account.description] = columns return sheets