From 1920b9c9f0b2602646aa0beb74384448523bdb4a Mon Sep 17 00:00:00 2001 From: jimmy Date: Sat, 12 Jun 2021 08:40:41 +0000 Subject: [PATCH] Add b2 setup. Change repourl --- src/pack.py | 21 ++++++++++++++++++++- 1 file changed, 20 insertions(+), 1 deletion(-) diff --git a/src/pack.py b/src/pack.py index 4b780eb..a4e7543 100644 --- a/src/pack.py +++ b/src/pack.py @@ -4,16 +4,24 @@ import shutil import zipfile import hashlib from dotenv import load_dotenv +from b2sdk.v1 import InMemoryAccountInfo +from b2sdk.v1 import B2Api class Pack: def __init__(self): load_dotenv() self.repourl = os.environ.get("RESOURCE_PACK_URL") - command = "git clone git@github.com:" + self.repourl + command = "git clone " + self.repourl self.dir = re.split("/|\.", self.repourl)[1] status = os.system(command) print(status) + info = InMemoryAccountInfo() # store credentials, tokens and cache in memory + b2_api = B2Api(info) + b2_api.authorize_account("production", os.environ.get("APPLICATION_KEY_ID"), + os.environ.get("APPLICATION_KEY")) + self.bucket = b2_api.get_bucket_by_name(os.environ.get("BUCKET")) + def __del__(self): shutil.rmtree(self.dir) @@ -47,6 +55,17 @@ class Pack: with open("files/hash", 'w') as hashfile: hashfile.write(self.hash) + def upload(self): + bucket.upload_local_file( + local_file="files/pack.zip", + file_name="pack.zip") + + bucket.upload_local_file( + local_file="files/hash", + file_name="hash") + + + if __name__ == "__main__":