kopia lustrzana https://github.com/bellingcat/auto-archiver
added log diretory and file creation
rodzic
8f62e8b7c6
commit
e18a9779db
|
@ -5,15 +5,20 @@ import requests
|
||||||
import shutil
|
import shutil
|
||||||
import gspread
|
import gspread
|
||||||
from loguru import logger
|
from loguru import logger
|
||||||
from dotenv import load_dotenv
|
# from dotenv import load_dotenv
|
||||||
from selenium import webdriver
|
from selenium import webdriver
|
||||||
import traceback
|
import traceback
|
||||||
|
|
||||||
import archivers
|
import archivers
|
||||||
from storages import S3Storage, S3Config
|
from storages import S3Storage, S3Config
|
||||||
from utils import GWorksheet, mkdir_if_not_exists
|
from utils import GWorksheet, mkdir_if_not_exists
|
||||||
|
import sys
|
||||||
|
|
||||||
load_dotenv()
|
logger.add("logs/1trace.log", level="TRACE")
|
||||||
|
logger.add("logs/2info.log", level="INFO")
|
||||||
|
logger.add("logs/3success.log", level="SUCCESS")
|
||||||
|
logger.add("logs/4warning.log", level="WARNING")
|
||||||
|
logger.add("logs/5error.log", level="ERROR")
|
||||||
|
|
||||||
|
|
||||||
def update_sheet(gw, row, result: archivers.ArchiveResult):
|
def update_sheet(gw, row, result: archivers.ArchiveResult):
|
||||||
|
@ -86,7 +91,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}" header={header}')
|
logger.info(f'Opening worksheet ii={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'):
|
||||||
|
@ -151,8 +156,9 @@ def process_sheet(sheet, header=1, columns=GWorksheet.COLUMN_NAMES):
|
||||||
logger.success(f'Finshed worksheet {wks.title}')
|
logger.success(f'Finshed worksheet {wks.title}')
|
||||||
driver.quit()
|
driver.quit()
|
||||||
|
|
||||||
|
@logger.catch
|
||||||
def main():
|
def main():
|
||||||
|
logger.debug(f'Passed args:{sys.argv}')
|
||||||
parser = argparse.ArgumentParser(
|
parser = argparse.ArgumentParser(
|
||||||
description='Automatically archive social media videos from a Google Sheets document')
|
description='Automatically archive social media videos from a Google Sheets document')
|
||||||
parser.add_argument('--sheet', action='store', dest='sheet', help='the name of the google sheets document', required=True)
|
parser.add_argument('--sheet', action='store', dest='sheet', help='the name of the google sheets document', required=True)
|
||||||
|
|
Ładowanie…
Reference in New Issue