kopia lustrzana https://github.com/bellingcat/auto-archiver
adds logs
rodzic
6e5e7212c2
commit
69483d432c
|
@ -79,7 +79,7 @@ def process_sheet(sheet, header=1, columns=GWorksheet.COLUMN_NAMES):
|
||||||
|
|
||||||
# loop through worksheets to check
|
# loop through worksheets to check
|
||||||
for ii, wks in enumerate(sh.worksheets()):
|
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)
|
gw = GWorksheet(wks, header_row=header, columns=columns)
|
||||||
|
|
||||||
if not gw.col_exists('url'):
|
if not gw.col_exists('url'):
|
||||||
|
|
|
@ -18,8 +18,10 @@ class GWorksheet:
|
||||||
|
|
||||||
def __init__(self, worksheet, columns=COLUMN_NAMES, header_row=1):
|
def __init__(self, worksheet, columns=COLUMN_NAMES, header_row=1):
|
||||||
self.wks = worksheet
|
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]]
|
self.headers = [v.lower() for v in self.values[0]]
|
||||||
|
print(self.headers)
|
||||||
|
self.row_offset = header_row - 1
|
||||||
self.columns = columns
|
self.columns = columns
|
||||||
|
|
||||||
def _check_col_exists(self, col: str):
|
def _check_col_exists(self, col: str):
|
||||||
|
@ -54,7 +56,7 @@ class GWorksheet:
|
||||||
col_index = self._col_index(col)
|
col_index = self._col_index(col)
|
||||||
|
|
||||||
if fresh:
|
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:
|
if type(row) == int:
|
||||||
row = self.get_row(row)
|
row = self.get_row(row)
|
||||||
|
|
||||||
|
@ -65,7 +67,7 @@ class GWorksheet:
|
||||||
def set_cell(self, row: int, col: str, val):
|
def set_cell(self, row: int, col: str, val):
|
||||||
# row is 1-based
|
# row is 1-based
|
||||||
col_index = self._col_index(col) + 1
|
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):
|
def batch_set_cell(self, cell_updates):
|
||||||
"""
|
"""
|
||||||
|
@ -82,4 +84,4 @@ class GWorksheet:
|
||||||
|
|
||||||
def to_a1(self, row: int, col: str):
|
def to_a1(self, row: int, col: str):
|
||||||
# row is 1-based
|
# 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)
|
||||||
|
|
Ładowanie…
Reference in New Issue