addftool 0.1.0__py3-none-any.whl → 0.1.2__py3-none-any.whl

addftool/blob.py

@@ -30,7 +30,11 @@ def create_dir_for_current_user(dir_path, sudo=False):
     if sudo:
         command = "sudo " + command
     execute_command(command)
-    command = f"chown $USER {dir_path}"
+    username = os.environ.get("USER")
+    if username is None or username == "":
+        command = f"chown {os.getuid()} {dir_path}"
+    else:
+        command = f"chown $USER {dir_path}"
     if sudo:
         command = "sudo " + command
     execute_command(command)
@@ -163,22 +167,22 @@ def mount_main(args):
 
 
 def get_token(args, info=False):
-    api_url = args.api
+    return get_sas_token(args.api, args.name, args.container, args.key, info=info)
+
 
-    f = Fernet(args.key.encode())
+# create a new function for function calling
+def get_sas_token(api_url, name, container, key, info=False):
+    f = Fernet(key.encode())
 
-    an = f.encrypt(args.name.encode()).decode()
-    cn = f.encrypt(args.container.encode()).decode()
+    an = f.encrypt(name.encode()).decode()
+    cn = f.encrypt(container.encode()).decode()
 
     if info:
         print("Get token from: ", api_url)
         print("Name: ", an)
         print("Container: ", cn)
 
-    params = {
-        "an": an,
-        "cn": cn, 
-    }
+    params = {"an": an, "cn": cn}
     response = requests.get(api_url, params=params)
     if response.status_code == 200:
         data = json.loads(response.text)

{addftool-0.1.0.dist-info → addftool-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: addftool
-Version: 0.1.0
+Version: 0.1.2
 Requires-Dist: cryptography
 Requires-Dist: requests
 Requires-Dist: PyYAML

addftool-0.1.2.dist-info/RECORD

@@ -0,0 +1,9 @@
+addftool/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+addftool/blob.py,sha256=NZOItDyFUIdV1tfhJZJJBEzGy296CE5NCictTzP4OPc,8282
+addftool/tool.py,sha256=EuKQ2t2InN7yB-_oYLcdsA7vRqzRGTunwIxplUSqEG0,2054
+addftool/util.py,sha256=Q3A68vJDxgfeNiEFmk54HuMuworVndocXpSbVpvGMfc,362
+addftool-0.1.2.dist-info/METADATA,sha256=gmakYu_DywA1K1z3yEoLN0h43nBUFbtNy0JmKcEbTgw,126
+addftool-0.1.2.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+addftool-0.1.2.dist-info/entry_points.txt,sha256=h5TlQy4AQw-J55HqI7FRkIVRUfw-x3pStFWCoYtoCTM,78
+addftool-0.1.2.dist-info/top_level.txt,sha256=jqj56-plrBbyzY0tIxB6wPzjAA8kte4hUlajyyQygN4,9
+addftool-0.1.2.dist-info/RECORD,,

addftool-0.1.0.dist-info/RECORD

@@ -1,9 +0,0 @@
-addftool/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-addftool/blob.py,sha256=P4Fj6ghLsEAw3C1NpEKEe3N4_LS3EedS1OYm5C6yzPw,8006
-addftool/tool.py,sha256=EuKQ2t2InN7yB-_oYLcdsA7vRqzRGTunwIxplUSqEG0,2054
-addftool/util.py,sha256=Q3A68vJDxgfeNiEFmk54HuMuworVndocXpSbVpvGMfc,362
-addftool-0.1.0.dist-info/METADATA,sha256=EbzSeyvThjLaTnuLShjdJ_3skYisyb2NYaWlBv7qMd8,126
-addftool-0.1.0.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-addftool-0.1.0.dist-info/entry_points.txt,sha256=h5TlQy4AQw-J55HqI7FRkIVRUfw-x3pStFWCoYtoCTM,78
-addftool-0.1.0.dist-info/top_level.txt,sha256=jqj56-plrBbyzY0tIxB6wPzjAA8kte4hUlajyyQygN4,9
-addftool-0.1.0.dist-info/RECORD,,