Initial Commit

ah_tools/core.py

@@ -0,0 +1,53 @@
+import argparse
+import sys
+
+from .check_backups import main as check_backups
+from .helpers import decrypt, encrypt
+
+
+def dummy():
+    check_backups()
+
+
+def encrypt_cli():
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "-p", "--password",
+        type=str,
+        required=True,
+        help="password to decrypt the servers file"
+    )
+    parser.add_argument(
+        "-S", "--salt",
+        type=str,
+        required=True,
+        help="salt for decrypting the servers file"
+    )
+
+    args = parser.parse_args()
+
+    data = sys.stdin.read()
+    data = encrypt(args.salt, args.password, data)
+    print(data)
+
+
+def decrypt_cli():
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "-p", "--password",
+        type=str,
+        required=True,
+        help="password to decrypt the servers file"
+    )
+    parser.add_argument(
+        "-S", "--salt",
+        type=str,
+        required=True,
+        help="salt for decrypting the servers file"
+    )
+
+    args = parser.parse_args()
+
+    data = sys.stdin.read()
+    data = decrypt(args.salt, args.password, data)
+    print(data)