diff --git a/main.py b/main.py index ff18145..8be5d09 100644 --- a/main.py +++ b/main.py @@ -150,16 +150,15 @@ def main(driver: WebDriver, logger = logging.getLogger('main')): for col in range(1, self.wb.active.max_column + 1): value = str(self.wb.active.cell(1, col).value) - if value in keys: self.headers[value] = col + self.headers[value] = col for row in range(1, self.wb.active.max_row + 1): - for key in keys: - col = self.headers[key] + for key, col in map(lambda k: (k, self.headers[k]), keys): value = str(self.wb.active.cell(row, col).value) self.indices.setdefault(key, dict())[value] = row - def at(self, by: str, index) -> dict: - if (a := self.indices.get(by)) is None or (b := dict.get(a, index)) is None: return dict() + def at(self, axis: str, index) -> dict: + if (a := self.indices.get(axis)) is None or (b := dict.get(a, index)) is None: return dict() return { k: self.wb.active.cell(b, c).value for k, c in self.headers.items() } try: