bellingcat/auto-archiver
1
0
Fork 0
mirror of https://github.com/bellingcat/auto-archiver.git synced 2026-06-11 12:48:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

simplifies access to google sheets, single get_values

Browse Source
This commit is contained in:
msramalho
2022-03-09 12:17:51 +01:00
parent 544e7578a6
commit ff874fe0d3
2 changed files with 9 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
utils/gworksheet.py
View File

@@ -18,7 +18,8 @@ class GWorksheet:
def __init__(self, worksheet, columns=COLUMN_NAMES, header_row=1):
self.wks = worksheet
self.headers = [v.lower() for v in self.wks.row_values(header_row)]
self.values = self.wks.get_values()
self.headers = [v.lower() for v in self.values[header_row - 1]]
self.columns = columns
def _check_col_exists(self, col: str):
@@ -34,14 +35,14 @@ class GWorksheet:
return self.columns[col] in self.headers
def count_rows(self):
return len(self.wks.get_values())
return len(self.values)
def get_row(self, row: int):
# row is 1-based
return self.wks.row_values(row)
return self.values[row - 1]
def get_values(self):
return self.wks.get_values()
return self.values
def get_cell(self, row, col: str):
"""