You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
104 lines
3.2 KiB
104 lines
3.2 KiB
from collections import namedtuple |
|
|
|
from .helpers import excel_value_as_number |
|
|
|
KIND_OBLIGO = "obligo" |
|
KIND_EXPENSES = "expenses" |
|
|
|
VALID_MATERIAL_IDS = { |
|
"0216 - Sachmittel", |
|
"0227 - Verbrauchsmaterial", |
|
"0843 - Verbrauch/Stoffe", |
|
"2 - Sachmittel", |
|
} |
|
|
|
OverviewBudgetEntry = namedtuple( |
|
"OverviewBudgetEntry", ["description", "kind", "amount"] |
|
) |
|
|
|
|
|
class ProjectOverview: |
|
def __init__(self, budget_data): |
|
"""initializes the class""" |
|
self.budget_data = budget_data |
|
self.entries = [] |
|
self.found = False |
|
|
|
@property |
|
def project(self): |
|
"""returns the project number""" |
|
return self.budget_data.project |
|
|
|
@property |
|
def row(self): |
|
"""returns the excel row number""" |
|
return self.budget_data.row |
|
|
|
@property |
|
def budget(self): |
|
"""returns the budget""" |
|
return self.budget_data.budget |
|
|
|
@property |
|
def expenses(self): |
|
"""returns the accumulated expenses""" |
|
numbers = (excel_value_as_number(e.amount) for e in self.entries) |
|
values = (abs(entry) for entry in numbers) |
|
return sum(values) |
|
|
|
@property |
|
def available(self): |
|
"""returns the still available rest budget""" |
|
return (self.budget_data.budget or 0) - self.expenses |
|
|
|
def add(self, description, kind, amount): |
|
"""adds an entry that modifies the available budget""" |
|
entry = OverviewBudgetEntry(description, kind, amount) |
|
self.entries.append(entry) |
|
|
|
def __repr__(self): |
|
"""returns an informative string representation""" |
|
return ( |
|
f"<ProjectOverview {self.project}:" |
|
f"{self.budget} - {self.expenses} = {self.available}>" |
|
) |
|
|
|
|
|
def _create_overview_map(budget_list): |
|
"""returns a dictonary with project as key and overview as value""" |
|
mapping = {} |
|
for budget_data in budget_list: |
|
overview = ProjectOverview(budget_data) |
|
mapping[str(overview.project)] = overview |
|
return mapping |
|
|
|
|
|
def _filter_superx_material_expenses(superx_export): |
|
"""filters superx data to only contain material entries""" |
|
return (i for i in superx_export.data if i.kind in VALID_MATERIAL_IDS) |
|
|
|
|
|
def _create_entries_from_superx(overview_map, superx_export_data): |
|
"""adds overview entries from superx data""" |
|
for line in superx_export_data: |
|
if line.project in overview_map: |
|
overview = overview_map[line.project] |
|
overview.add(line.kind, KIND_OBLIGO, line.obligo) |
|
overview.add(line.kind, KIND_EXPENSES, line.expenses) |
|
return overview_map |
|
|
|
|
|
def _set_found_state(overview_map, superx_export): |
|
superx_projects = {line.project for line in superx_export.data} |
|
found_projects = superx_projects.intersection(set(overview_map.keys())) |
|
for project in found_projects: |
|
overview_map[project].found = True |
|
return overview_map |
|
|
|
|
|
def create_overview(budget_list, superx_export): |
|
"""create a overview map with budget entries from the parsed raw data""" |
|
tmp_map = _create_overview_map(budget_list) |
|
overview_map = _set_found_state(tmp_map, superx_export) |
|
material_expenses = _filter_superx_material_expenses(superx_export) |
|
return _create_entries_from_superx(overview_map, material_expenses)
|
|
|