accounting/frappe_docker
2
0
Fork 0
mirror of https://github.com/frappe/frappe_docker.git synced 2025-01-05 15:12:13 +00:00
Code Issues Releases Activity

feat: optional bucket directory for push-backup (#784)

Browse Source
This commit is contained in:
Revant Nandgaonkar 2022-04-20 05:09:56 +05:30 committed by GitHub
parent c9258ae843
commit 5fbd9bdda5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
images/worker/push_backup.py
View File

@ -22,6 +22,7 @@ class Arguments(argparse.Namespace):
endpoint_url: str endpoint_url: str
aws_access_key_id: str aws_access_key_id: str
aws_secret_access_key: str aws_secret_access_key: str
bucket_directory: str
def _get_files_from_previous_backup(site_name: str) -> list[Path]: def _get_files_from_previous_backup(site_name: str) -> list[Path]:
@ -59,9 +60,13 @@ def get_bucket(args: Arguments) -> _Bucket:
).Bucket(args.bucket) ).Bucket(args.bucket)
def upload_file(path: Path, site_name: str, bucket: _Bucket) -> None: def upload_file(
path: Path, site_name: str, bucket: _Bucket, bucket_directory: str = None
) -> None:
filename = str(path.absolute()) filename = str(path.absolute())
key = str(Path(site_name) / path.name) key = str(Path(site_name) / path.name)
if bucket_directory:
key = bucket_directory + "/" + key
print(f"Uploading {key}") print(f"Uploading {key}")
bucket.upload_file(Filename=filename, Key=key) bucket.upload_file(Filename=filename, Key=key)
os.remove(path) os.remove(path)
@ -74,7 +79,12 @@ def push_backup(args: Arguments) -> None:
bucket = get_bucket(args) bucket = get_bucket(args)
for path in files: for path in files:
upload_file(path=path, site_name=args.site, bucket=bucket) upload_file(
path=path,
site_name=args.site,
bucket=bucket,
bucket_directory=args.bucket_directory,
)
print("Done!") print("Done!")
@ -94,6 +104,7 @@ def parse_args(args: list[str]) -> Arguments:
required=True, required=True,
default=os.getenv("AWS_SECRET_ACCESS_KEY"), default=os.getenv("AWS_SECRET_ACCESS_KEY"),
) )
parser.add_argument("--bucket-directory")
return parser.parse_args(args, namespace=Arguments()) return parser.parse_args(args, namespace=Arguments())