mirror of
https://github.com/bellingcat/auto-archiver.git
synced 2026-06-11 12:48:28 +03:00
adds local storage
This commit is contained in:
@@ -1,3 +1,4 @@
|
||||
# we need to explicitly expose the available imports here
|
||||
from .base_storage import *
|
||||
from .local_storage import *
|
||||
from .s3_storage import *
|
||||
20
storages/local_storage.py
Normal file
20
storages/local_storage.py
Normal file
@@ -0,0 +1,20 @@
|
||||
import os
|
||||
from .base_storage import Storage
|
||||
|
||||
|
||||
class LocalStorage(Storage):
|
||||
def __init__(self, folder):
|
||||
self.folder = folder
|
||||
if len(self.folder) and self.folder[-1] != '/':
|
||||
self.folder += '/'
|
||||
|
||||
def get_cdn_url(self, key):
|
||||
return self.folder + key
|
||||
|
||||
def exists(self, key):
|
||||
return os.path.isfile(self.get_cdn_url(key))
|
||||
|
||||
def uploadf(self, file, key, **kwargs):
|
||||
path = self.get_cdn_url(key)
|
||||
with open(path, "wb") as outf:
|
||||
outf.write(file.read())
|
||||
Reference in New Issue
Block a user