pull/18/head
msramalho 2022-03-12 20:04:08 +01:00
rodzic 6e5e7212c2
commit 69483d432c
2 zmienionych plików z 7 dodań i 5 usunięć

Wyświetl plik

@ -79,7 +79,7 @@ def process_sheet(sheet, header=1, columns=GWorksheet.COLUMN_NAMES):
# loop through worksheets to check
for ii, wks in enumerate(sh.worksheets()):
logger.info(f'Opening worksheet {ii}: "{wks.title}"')
logger.info(f'Opening worksheet {ii}: "{wks.title}" header={header}')
gw = GWorksheet(wks, header_row=header, columns=columns)
if not gw.col_exists('url'):

Wyświetl plik

@ -18,8 +18,10 @@ class GWorksheet:
def __init__(self, worksheet, columns=COLUMN_NAMES, header_row=1):
self.wks = worksheet
self.values = self.wks.get_values()[header_row-1:]
self.values = self.wks.get_values()[header_row - 1:]
self.headers = [v.lower() for v in self.values[0]]
print(self.headers)
self.row_offset = header_row - 1
self.columns = columns
def _check_col_exists(self, col: str):
@ -54,7 +56,7 @@ class GWorksheet:
col_index = self._col_index(col)
if fresh:
return self.wks.cell(row, col_index + 1).value
return self.wks.cell(row + self.row_offset, col_index + 1).value
if type(row) == int:
row = self.get_row(row)
@ -65,7 +67,7 @@ class GWorksheet:
def set_cell(self, row: int, col: str, val):
# row is 1-based
col_index = self._col_index(col) + 1
self.wks.update_cell(row, col_index, val)
self.wks.update_cell(row + self.row_offset, col_index, val)
def batch_set_cell(self, cell_updates):
"""
@ -82,4 +84,4 @@ class GWorksheet:
def to_a1(self, row: int, col: str):
# row is 1-based
return utils.rowcol_to_a1(row, self._col_index(col) + 1)
return utils.rowcol_to_a1(row + self.row_offset, self._col_index(col) + 1)