addftool 0.0.2__py3-none-any.whl → 0.0.4__py3-none-any.whl

addftool/blob.py

@@ -1,11 +1,12 @@
 import argparse
 import requests
-import subprocess
 import os
 import json
 import yaml
 from cryptography.fernet import Fernet
 
+from .util import execute_command
+
 
 def add_api(parser):
     parser.add_argument("-k", "--key", help="f key", required=True)
@@ -23,17 +24,6 @@ def get_ubuntu_version():
     return "22.04"
 
 
-def execute_command(command, to_file=None):
-    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
-
-
 def create_dir_for_current_user(dir_path, sudo=False):
     command = f"mkdir -p {dir_path}"
     if sudo:
@@ -45,7 +35,29 @@ def create_dir_for_current_user(dir_path, sudo=False):
     execute_command(command)
 
 
+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_main(args):
+
+    to_install = []
+    for package in args.packages.split():
+        if check_package_installed(package):
+            print(f"{package} is already installed")
+            continue
+        to_install.append(package)
+    
+    if len(to_install) == 0:
+        print("All packages are already installed")
+        return
+    else:
+        args.packages = " ".join(to_install)
+
     # generate install script
     if args.output_script is not None:
         script_writer = open(args.output_script, "w")

addftool/tool.py

@@ -0,0 +1,79 @@
+import argparse
+import os
+
+from .util import execute_command
+
+
+def init_vscode():
+    to_install = [
+        "github.copilot",
+        "github.copilot-chat",
+        "ms-python.debugpy",
+        "ms-python.python",
+        "ms-python.vscode-pylance",
+        "ms-toolsai.jupyter",
+        "ms-toolsai.jupyter-keymap",
+        "ms-toolsai.jupyter-renderers",
+        "ms-toolsai.vscode-jupyter-cell-tags",
+        "ms-toolsai.vscode-jupyter-slideshow",
+    ]
+    for ext in to_install:
+        execute_command(f"code --install-extension {ext}")
+
+
+def add_alias():
+    # add to ~/.bashrc
+
+    home_dir = os.path.expanduser("~")
+    bashrc_path = os.path.join(home_dir, ".bashrc")
+
+    # check if already added
+    with open(bashrc_path) as f:
+        for line in f:
+            if line.startswith("# Addf's alias"):
+                print("Already added")
+                return
+
+    to_add = [
+        "alias pull='git pull origin'",
+        "alias dupwd='du -h --max-depth=1 ./'",
+    ]
+
+    with open(bashrc_path, "a") as f:
+        f.write("\n")
+        f.write("# Addf's alias\n")
+        for line in to_add:
+            f.write(line + "\n")
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Addf's tool")
+
+    parser.add_argument('--init', default="", type=str, help="init")
+
+    args = parser.parse_args()
+
+    if args.init:
+        to_do = args.init.strip()
+        if to_do == 'vscode':
+            init_vscode()
+        elif to_do == 'git':
+            execute_command('git config --global user.name "addf400"')
+            execute_command('git config --global user.email "addf400@foxmail.com"')
+        elif to_do == 'alias':
+            add_alias()
+        elif to_do == 'all':
+            init_vscode()
+            execute_command('git config --global user.name "addf400"')
+            execute_command('git config --global user.email "addf400@foxmail.com"')
+            add_alias()
+        else:
+            print("No such init option")
+
+
+def pull():
+    execute_command("git pull origin")
+
+
+if __name__ == "__main__":
+    main()

addftool/util.py

@@ -1,32 +1,12 @@
-def rangeexpand(txt):
-    lst = []
-    for r in txt.split(','):
-        if '-' in r[1:]:
-            r0, r1 = r[1:].split('-', 1)
-            r1 = r1.strip()
-            lst += range(int(r[0] + r0), int(r1) + 1)
-        elif not r.strip():
-            continue
-        else:
-            lst.append(int(r.strip()))
-    return lst
-
-
-def test_rangeexpand():
-    import pytest
-
-    assert rangeexpand('1, 3-5, 7') == [1, 3, 4, 5, 7]
-    assert rangeexpand('1, 3-5, 7, 10-12') == [1, 3, 4, 5, 7, 10, 11, 12]
-    assert rangeexpand('1-5') == [1, 2, 3, 4, 5]
-    assert rangeexpand('1-5, 7-10') == [1, 2, 3, 4, 5, 7, 8, 9, 10]
-    assert rangeexpand('') == []
-    assert rangeexpand('1') == [1]
-    assert rangeexpand('1-1') == [1]
-    assert rangeexpand('1-1, 3-3, 5-5') == [1, 3, 5]
-    assert rangeexpand('1-1, 3-3, 5-5, 7-7') == [1, 3, 5, 7]
-    
-    with pytest.raises(ValueError):
-        rangeexpand('1-2-3')
-
-    with pytest.raises(ValueError):
-        rangeexpand('1 7-9')
+import subprocess
+
+
+def execute_command(command, to_file=None):
+    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

{addftool-0.0.2.dist-info → addftool-0.0.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: addftool
-Version: 0.0.2
+Version: 0.0.4
 License-File: LICENSE
 Requires-Dist: cryptography
 Requires-Dist: requests

addftool-0.0.4.dist-info/RECORD

@@ -0,0 +1,10 @@
+addftool/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+addftool/blob.py,sha256=5gdXzv_8RQOjCEo1ihsP3S3e7nrfqKWwfpxGw7pipXs,6690
+addftool/tool.py,sha256=EuKQ2t2InN7yB-_oYLcdsA7vRqzRGTunwIxplUSqEG0,2054
+addftool/util.py,sha256=Q3A68vJDxgfeNiEFmk54HuMuworVndocXpSbVpvGMfc,362
+addftool-0.0.4.dist-info/LICENSE,sha256=7EI8xVBu6h_7_JlVw-yPhhOZlpY9hP8wal7kHtqKT_E,1074
+addftool-0.0.4.dist-info/METADATA,sha256=If4J7R9OXA8e0OXs02shLes0NUmSK5pk4hsYyG48qKg,149
+addftool-0.0.4.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+addftool-0.0.4.dist-info/entry_points.txt,sha256=h5TlQy4AQw-J55HqI7FRkIVRUfw-x3pStFWCoYtoCTM,78
+addftool-0.0.4.dist-info/top_level.txt,sha256=jqj56-plrBbyzY0tIxB6wPzjAA8kte4hUlajyyQygN4,9
+addftool-0.0.4.dist-info/RECORD,,

{addftool-0.0.2.dist-info → addftool-0.0.4.dist-info}/entry_points.txt

@@ -1,2 +1,3 @@
 [console_scripts]
 addfblob = addftool.blob:main
+addftool = addftool.tool:main

addftool-0.0.2.dist-info/RECORD

@@ -1,9 +0,0 @@
-addftool/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-addftool/blob.py,sha256=bPtnt0T79jdkMjdJFp_wExvaoxlclbhw9cTw7Iug3rY,6437
-addftool/util.py,sha256=gx-pqNJk31tmWtRJvZrIkI15ER7QZg9FtaM_OyP0JqU,975
-addftool-0.0.2.dist-info/LICENSE,sha256=7EI8xVBu6h_7_JlVw-yPhhOZlpY9hP8wal7kHtqKT_E,1074
-addftool-0.0.2.dist-info/METADATA,sha256=qv43J1fLZpinNN5oGkExEcoVsDCHAuots7BrWTkw1So,149
-addftool-0.0.2.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
-addftool-0.0.2.dist-info/entry_points.txt,sha256=H3Z3iWsLB1CFdckedIA652_3tWaNS5Rl7fFWgFaFRIc,48
-addftool-0.0.2.dist-info/top_level.txt,sha256=jqj56-plrBbyzY0tIxB6wPzjAA8kte4hUlajyyQygN4,9
-addftool-0.0.2.dist-info/RECORD,,