mirror of
https://github.com/bellingcat/auto-archiver.git
synced 2026-06-11 12:48:28 +03:00
31 lines
921 B
Python
31 lines
921 B
Python
from loguru import logger
|
|
from abc import ABC, abstractmethod
|
|
from pathlib import Path
|
|
|
|
|
|
class Storage(ABC):
|
|
TMP_FOLDER = "tmp/"
|
|
|
|
@abstractmethod
|
|
def __init__(self, config): pass
|
|
|
|
@abstractmethod
|
|
def get_cdn_url(self, key): pass
|
|
|
|
@abstractmethod
|
|
def exists(self, key): pass
|
|
|
|
@abstractmethod
|
|
def uploadf(self, file, key, **kwargs): pass
|
|
|
|
def upload(self, filename: str, key: str, **kwargs):
|
|
logger.debug(f'[{self.__class__.__name__}] uploading file {filename} with key {key}')
|
|
with open(filename, 'rb') as f:
|
|
self.uploadf(f, key, **kwargs)
|
|
|
|
#TODO: is this really necessary if only use os.path operations
|
|
def _clean_path(self, folder, default="", add_forward_slash=True):
|
|
if folder is None or type(folder) != str or len(folder.strip()) == 0:
|
|
return default
|
|
return str(Path(folder)) + ("/" if add_forward_slash else "")
|