initial_commit

svc.py

@@ -0,0 +1,89 @@
+#!/bin/python3
+
+###########
+# Modules #
+###########
+
+import os
+import subprocess
+import sys
+
+#############
+# Functions #
+#############
+
+# Access check
+def sucheck():
+    user = subprocess.run('whoami', shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+    if user.stdout[:-1] == 'root':
+        return True
+    else:
+        return 'Run script as root user'
+
+#Args check
+def argcheck():
+    if len(sys.orig_argv[1:]) == 3:
+        return True
+    else:
+        return 'Invalid usage'
+
+#Options check
+def optcheck(option):
+    path = ''
+    allopts_list = []
+    allopts = { 'enable': ['enable', 'e', 'on'], 'disable': ['disable','d','off'], 'down': ['down','dt']}
+    for i in allopts.keys():
+        allopts_list = allopts_list + allopts[i]
+    if not(option in allopts_list):
+        print(f'Invalid option: {option}')
+        sys.exit()
+    while path == '':
+        if option in allopts['enable']:
+            option = 'enable'
+            path = '/etc/sv'
+        if option in allopts['disable']:
+            option = 'disable'
+            path = '/var/service'
+        if option in allopts['down']:
+            option = 'down'
+            path = '/etc/sv'
+    return path, option
+
+#Service check
+def svcheck(service,path):
+    list = os.listdir(path)
+    if service in list:
+        return True
+    else:
+        return f'Service not found: {service}'
+
+#Control service
+def action(option,service):
+    if option == 'enable':
+        subprocess.run(f'ln -s /etc/sv/{service} /var/service', shell=True)
+        return f'{service} enabled'
+    if option == 'disable':
+        subprocess.run(f'rm /var/service/{service}', shell=True)
+        return f'{service} disabled'
+    if option == 'down':
+        subprocess.run(f'touch /etc/sv/{service}/down', shell=True)
+        return f'Down-file created for {service}'
+
+########
+# Exec #
+########
+
+ac = argcheck()
+if ac == True:
+    suc = sucheck()
+    if suc == True:
+        oc = optcheck(sys.argv[1])
+        sc = svcheck(sys.argv[2],oc[0])
+        if sc == True:
+            print(action(oc[1],sys.argv[2]))
+        else:
+            print(sc)
+    else:
+        print(suc)
+else:
+    print(ac)