class Store:
def __init__(self):
load_dotenv()
self.bucket_name = self.get_bucket_name()
self.finder = PathFinder()
def get_bucket_name(self):
return os.environ.get('S3_BUCKET') if not C.DEV else os.environ.get(
'S3_DEV_BUCKET')
def upload_file(self, path):
s3 = boto3.resource('s3')
bucket = s3.Bucket(self.bucket_name)
bucket.upload_file(path, path)
def upload_dir(self, path, truncate=False):
paths = self.finder.get_all_paths(path, truncate=truncate)
with Pool() as p:
p.map(self.upload_file, paths)
def delete_objects(self, keys):
if keys:
objects = [{'Key': key} for key in keys]
s3 = boto3.resource('s3')
bucket = s3.Bucket(self.bucket_name)
bucket.delete_objects(Delete={'Objects': objects})
def get_all_keys(self):
s3 = boto3.resource('s3')
bucket = s3.Bucket(self.bucket_name)
keys = [obj.key for obj in bucket.objects.filter()]
return keys
def key_exists(self, key, download=False):
try:
s3 = boto3.resource('s3')
bucket = s3.Bucket(self.bucket_name)
bucket.Object(key).load()
except ClientError:
return False
else:
return True
def download_file(self, key):
try:
self.finder.make_path(key)
s3 = boto3.resource('s3')
with open(key, 'wb') as file:
bucket = s3.Bucket(self.bucket_name)
bucket.download_fileobj(key, file)
except ClientError as e:
print(f'{key} does not exist in S3.')
os.remove(key)
raise e
def rename_key(self, old_key, new_key):
s3 = boto3.resource('s3')
s3.Object(
self.bucket_name,
new_key).copy_from(CopySource=f'{self.bucket_name}/{old_key}')
s3.Object(self.bucket_name, old_key).delete()
class Store:
def __init__(self):
load_dotenv(find_dotenv('config.env'))
self.bucket_name = self.get_bucket_name()
self.finder = PathFinder()
def get_bucket_name(self):
return os.environ.get('S3_BUCKET') if not C.DEV else os.environ.get(
'S3_DEV_BUCKET')
def get_bucket(self):
s3 = boto3.resource('s3')
bucket = s3.Bucket(self.bucket_name)
return bucket
def upload_file(self, path):
bucket = self.get_bucket()
key = path.replace('\\', '/')
bucket.upload_file(path, key)
def upload_dir(self, **kwargs):
paths = self.finder.get_all_paths(**kwargs)
with Pool() as p:
p.map(self.upload_file, paths)
def delete_objects(self, keys):
if keys:
objects = [{'Key': key.replace('\\', '/')} for key in keys]
bucket = self.get_bucket()
bucket.delete_objects(Delete={'Objects': objects})
def get_all_keys(self):
bucket = self.get_bucket()
keys = [obj.key for obj in bucket.objects.filter()]
return keys
def key_exists(self, key, download=False):
key = key.replace('\\', '/')
try:
if download:
self.download_file(key)
else:
bucket = self.get_bucket()
bucket.Object(key).load()
except ClientError:
return False
else:
return True
def download_file(self, key):
try:
self.finder.make_path(key)
with open(key, 'wb') as file:
bucket = self.get_bucket()
s3_key = key.replace('\\', '/')
bucket.download_fileobj(s3_key, file)
except ClientError as e:
print(f'{key} does not exist in S3.')
os.remove(key)
raise e
def copy_object(self, src, dst):
src = src.replace('\\', '/')
dst = dst.replace('\\', '/')
bucket = self.get_bucket()
copy_source = {'Bucket': self.bucket_name, 'Key': src}
bucket.copy(copy_source, dst)
def rename_key(self, old_key, new_key):
old_key = old_key.replace('\\', '/')
new_key = new_key.replace('\\', '/')
self.copy_object(old_key, new_key)
self.delete_objects([old_key])
def last_modified(self, key):
key = key.replace('\\', '/')
bucket = self.get_bucket()
obj = bucket.Object(key)
then = obj.last_modified.replace(tzinfo=None)
return then
def modified_delta(self, key):
key = key.replace('\\', '/')
then = self.last_modified(key)
now = datetime.utcnow()
return now - then
