Add b2 setup. Change repourl

This commit is contained in:
jimmy 2021-06-12 08:40:41 +00:00
parent acbe9a7751
commit 1920b9c9f0
1 changed files with 20 additions and 1 deletions

View File

@ -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__":