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.
68 lines
1.5 KiB
68 lines
1.5 KiB
""" Budget Parser """ |
|
|
|
from collections import namedtuple |
|
|
|
from .helpers import get_sheet_of_file, strip_excel_value, is_empty_excel_value |
|
from .exceptions import BudgetParserError |
|
|
|
EXPECTED_TABLE_HEADERS = [ |
|
"Nr.", |
|
"Projekt", |
|
"Laufzeit", |
|
"BA", |
|
"Budget", |
|
"Ausgaben", |
|
"Rest", |
|
] |
|
|
|
|
|
ExcelRow = namedtuple("ExcelRow", ["row", "data"]) |
|
|
|
BudgetData = namedtuple( |
|
"BudgetData", |
|
[ |
|
"row", |
|
"number", |
|
"project_name", |
|
"period", |
|
"project", |
|
"budget", |
|
"expenses", |
|
"rest", |
|
], |
|
) |
|
|
|
|
|
def _check_table_header(xl_row): |
|
fields = [c.strip() for c in xl_row.data[:7]] |
|
print(fields) |
|
print(EXPECTED_TABLE_HEADERS) |
|
if fields != EXPECTED_TABLE_HEADERS: |
|
raise BudgetParserError(f"unexpected headers: '{xl_row.data}'") |
|
|
|
|
|
def _skip_empty_lines(rows): |
|
for xl_row in rows: |
|
first_cell = xl_row.data[0] |
|
if is_empty_excel_value(first_cell): |
|
continue |
|
yield xl_row |
|
|
|
|
|
def _parse_data_table(rows): |
|
for xl_row in _skip_empty_lines(rows): |
|
data = [strip_excel_value(value) for value in xl_row.data[:7]] |
|
yield BudgetData(xl_row.row, *data) |
|
|
|
|
|
def parse_budget_data(xls_sheet): |
|
""" parses the budget data """ |
|
rows = (ExcelRow(i, v) for i, v in enumerate(xls_sheet.values, start=1)) |
|
_check_table_header(next(rows)) |
|
return list(_parse_data_table(rows)) |
|
|
|
|
|
def parse_budget_file(file_path): |
|
""" parses the budget file """ |
|
sheet = get_sheet_of_file(file_path, sheet=None) |
|
return parse_budget_data(sheet)
|
|
|