From 1920b9c9f0b2602646aa0beb74384448523bdb4a Mon Sep 17 00:00:00 2001
From: jimmy <git@jimmy.nz>
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__":