diff --git a/main.py b/main.py index dba3165..3492dd9 100644 --- a/main.py +++ b/main.py @@ -44,7 +44,7 @@ def main(driver: WebDriver, logger = logging.getLogger('main')): except Exception as e: logger.critical('Unable to load starup page', exc_info=e) return 2 - + try: driver.switch_to.new_window('tab') driver.set_page_load_timeout(parameters['timeout']) @@ -141,17 +141,16 @@ def main(driver: WebDriver, logger = logging.getLogger('main')): except Exception as e: logger.critical('Unable to load input data', exc_info=e) return 6 - + try: file = Path('template.xlsx').resolve() template = openpyxl.load_workbook(file) + header = next(data) + source = template.active + output = BytesIO() except Exception as e: logger.critical('Unable to load template excel', exc_info=e) return 7 - - header = next(data, None) - source = template.active - output = BytesIO() def preprocess(data: str): try: return float(data) @@ -172,7 +171,7 @@ def main(driver: WebDriver, logger = logging.getLogger('main')): source[coord] = value sheet = template.copy_worksheet(source) - sheet.title = 'Copy' + sheet.title = "COPY" logger.debug('%s', sheet.title) if not len(template.worksheets) > 1: