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

Add header argument; set up webdriver

Browse Source
This commit is contained in:
Logan Williams
2022-02-25 16:09:35 +01:00
parent 09dc5b5b81
commit 63a2847ac9
4 changed files with 74 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
utils/gworksheet.py
View File

@@ -3,7 +3,7 @@ from gspread import utils
class GWorksheet:
COLUMN_NAMES = {
'url': 'media url',
'url': 'link',
'archive': 'archive location',
'date': 'archive date',
'status': 'archive status',
@@ -11,12 +11,14 @@ class GWorksheet:
'thumbnail_index': 'thumbnail index',
'timestamp': 'upload timestamp',
'title': 'upload title',
'duration': 'duration'
'duration': 'duration',
'screenshot': 'screenshot',
'hash': 'hash'
}
def __init__(self, worksheet, columns=COLUMN_NAMES):
def __init__(self, worksheet, columns=COLUMN_NAMES, header_row=1):
self.wks = worksheet
self.headers = [v.lower() for v in self.wks.row_values(1)]
self.headers = [v.lower() for v in self.wks.row_values(header_row)]
self.columns = columns
def _check_col_exists(self, col: str):
@@ -38,6 +40,9 @@ class GWorksheet:
# row is 1-based
return self.wks.row_values(row)
def get_values(self):
return self.wks.get_values()
def get_cell(self, row, col: str):
"""
returns the cell value from (row, col),