addftool 0.2.0__py3-none-any.whl → 0.2.2__py3-none-any.whl

addftool/addf_portal.py

@@ -1,5 +1,8 @@
 import argparse
 from addftool.process import add_killer_args, killer_main
+from addftool.sync import add_sync_args, sync_main
+from addftool.deploy import add_deploy_args, deploy_main
+from addftool.broadcast_folder import add_broadcast_folder_args, broadcast_folder_main
 
 
 def get_args():
@@ -7,6 +10,9 @@ def get_args():
 
     subparsers = parser.add_subparsers(dest='command', help='Sub-command help')
     add_killer_args(subparsers)
+    add_sync_args(subparsers)
+    add_deploy_args(subparsers)
+    add_broadcast_folder_args(subparsers)
 
     return parser.parse_args()
 
@@ -15,6 +21,12 @@ def main():
     args = get_args()
     if args.command == "kill":
         killer_main(args)
+    elif args.command == "sync":
+        sync_main(args)
+    elif args.command == "deploy":
+        deploy_main(args)
+    elif args.command == "broadcast-folder":
+        broadcast_folder_main(args)
     else:
         print("Unknown command: ", args.command)
 

addftool/util.py

@@ -1,12 +1,75 @@
+import os
 import subprocess
 
 
-def execute_command(command, to_file=None):
+def execute_command(command, to_file=None, only_stdout=True, hide=False):
     if to_file is not None:
         to_file.write(command + "\n")
         return None
     else:
-        print("Execute command: ", command)
-        result = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE).stdout.read().decode()
-        print("Result: ", result)
-        return result
+        if not hide:
+            print("Execute command: ", command)
+        result = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE)
+        result.wait()
+        print(f"Return code: {result.returncode}")
+        if result.stdout is not None:
+            stdout = result.stdout.read().decode()
+            print(f"Stdout: {stdout}")
+        else:
+            stdout = None
+        if only_stdout:
+            if not hide and stdout is not None:
+                print(stdout)
+            return stdout
+        if result.stderr is not None:
+            stderr = result.stderr.read().decode()
+            print(f"Stderr: {stderr}")
+        else:
+            stderr = None
+
+        return {'stdout': stdout, 'stderr': stderr, 'returncode': result.returncode}
+
+
+def need_sudo():
+    return os.name == 'posix' and os.getuid() != 0
+
+
+def is_running_in_docker():
+    return os.path.exists('/.dockerenv') or \
+           any('docker' in line for line in open('/proc/self/cgroup', 'r')) if os.path.exists('/proc/self/cgroup') else False or \
+           os.environ.get('container') == 'docker' or \
+           os.environ.get('DOCKER') == 'true' or \
+           os.environ.get('DOCKER_CONTAINER') == 'yes'
+
+
+def get_ubuntu_version():  
+    with open("/etc/os-release") as f:
+        for line in f:
+            if line.startswith("VERSION_ID="):
+                version = line.split("=")[1].strip().strip('"')
+                return version
+    return "22.04"
+
+
+def check_package_installed(package):
+    command = f"dpkg -l | grep {package}"
+    result = execute_command(command)
+    if result is not None and package in result:
+        return True
+    return False
+
+
+def install_packages(package_list):
+    to_install = []
+    for package in package_list:
+        if check_package_installed(package):
+            print(f"{package} is already installed")
+            continue
+        to_install.append(package)
+    
+    if len(to_install) > 0:
+        packages = " ".join(to_install)
+        command = f"apt-get install -y {packages}"
+        if need_sudo():
+            command = "sudo " + command
+        execute_command(command)

{addftool-0.2.0.dist-info → addftool-0.2.2.dist-info}/METADATA

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

{addftool-0.2.0.dist-info → addftool-0.2.2.dist-info}/RECORD

@@ -1,10 +1,10 @@
 addftool/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-addftool/addf_portal.py,sha256=U52TdNcwWOEvv_C5r-guWYxn3ntzwYI2eBzJIE7IdcY,493
+addftool/addf_portal.py,sha256=pw6ivnwMfmK4WC-1Gs5cMndXKQRbMm6VvDyxXJDNrm4,991
 addftool/blob.py,sha256=NZOItDyFUIdV1tfhJZJJBEzGy296CE5NCictTzP4OPc,8282
 addftool/broadcast_folder.py,sha256=bEOr-8Q14DhZFS658hab4U-9HvAW3EGluEriilIdEXQ,11976
 addftool/sync.py,sha256=ZpYxbM8uiPFrV7ODmOaM7asVPCWaxBixA-arVc-1kfs,14045
 addftool/tool.py,sha256=EuKQ2t2InN7yB-_oYLcdsA7vRqzRGTunwIxplUSqEG0,2054
-addftool/util.py,sha256=Q3A68vJDxgfeNiEFmk54HuMuworVndocXpSbVpvGMfc,362
+addftool/util.py,sha256=zlNLu8Be8cGIpNRqBw8_0q7nFxWlsJ9cToN62ohjdXE,2335
 addftool/deploy/__init__.py,sha256=UL8b-Idt7lStlMiOm8oTZ65fdzYz99Fgzq2Gaw8WsZc,1544
 addftool/deploy/azure.py,sha256=_o_9Eh8cVwLDAqvfyRYBtQRHs_Gul-nCs2ZXttwO1bk,1301
 addftool/deploy/ssh_server.py,sha256=f2T8fgwACVljPfdcimMywUjsFnLCWRde7iWPAILpRz8,5463
@@ -12,8 +12,8 @@ addftool/deploy/vscode_server.py,sha256=9KoWQdK3ESgKtdACEHTng2OuQKlIZjh_R_qJNT-X
 addftool/process/__init__.py,sha256=Dze8OrcyjQlAbPrjE_h8bMi8W4b3OJyZOjTucPrkJvM,3721
 addftool/process/utils.py,sha256=JldxnwanLJOgxaPgmCJh7SeBRaaj5rFxWWxh1hpsvbA,2609
 addftool/ssh/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-addftool-0.2.0.dist-info/METADATA,sha256=Ja75t6IT9XdIMjCUsZKLsx-QJJ9Yo2UfYtpVUC1AYPw,170
-addftool-0.2.0.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-addftool-0.2.0.dist-info/entry_points.txt,sha256=9lkmuWMInwUAtev8w8poNkNd7iML9Bjd5CBCFVxg2b8,111
-addftool-0.2.0.dist-info/top_level.txt,sha256=jqj56-plrBbyzY0tIxB6wPzjAA8kte4hUlajyyQygN4,9
-addftool-0.2.0.dist-info/RECORD,,
+addftool-0.2.2.dist-info/METADATA,sha256=UAlfwCvT_leVPhIku998LlT2f23Gd4ugq9Yh4Dzlt3E,170
+addftool-0.2.2.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+addftool-0.2.2.dist-info/entry_points.txt,sha256=9lkmuWMInwUAtev8w8poNkNd7iML9Bjd5CBCFVxg2b8,111
+addftool-0.2.2.dist-info/top_level.txt,sha256=jqj56-plrBbyzY0tIxB6wPzjAA8kte4hUlajyyQygN4,9
+addftool-0.2.2.dist-info/RECORD,,