Merge remote-tracking branch 'origin/initscript' into master

merge initscript to master

init.py

@@ -0,0 +1,74 @@
+#!/bin/python3
+
+import yaml
+import os
+from os.path import expanduser
+from shutil import copyfile
+from bs4 import BeautifulSoup
+import sys
+import subprocess
+
+# Install dependencies
+subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'bs4'])
+
+# get home directory
+home = expanduser("~")
+
+# get config path
+config_dir = home + '/.config/StartTree'
+config_path = home + '/.config/StartTree/config.yaml'
+
+# get cache path
+cache_dir = home + '/.cache/StartTree'
+
+def prettifyHTML(html):
+    soup = BeautifulSoup(html, 'html.parser')
+    prettyHTML = soup.prettify()
+    return prettyHTML
+
+def setup():
+    # check if .config path exists
+    if not os.path.isdir(home + '/.config'):
+        print("The directory '~/.config' does not exist, or you do not have permissions to edit it.")
+        exit()
+
+    if not os.path.isdir(home + '/.cache'):
+        print("The directory '~/.cache' does not exist, or you do not have permissions to edit it.")
+        exit()
+
+    # check if .config/StartTree exists, create it and config if not
+    if not os.path.isdir(config_dir):
+        print("Creating '~/.config/StartTree'...")
+        os.mkdir(config_dir)
+        print("Copying config.yaml")
+        copyfile("./config.yaml", config_path)
+        print("")
+
+    # check if config.yaml exists
+    if not os.path.exists(config_path):
+        print("No config.yaml found in '~/.config/StartTree'")
+        print("Copy the example config with:")
+        print("\tcp ./config.yaml $HOME/.config/StartTree/config.yaml")
+        print("or create your own in that directory.")
+        exit()
+
+    # create directory structure in .cache
+    if not os.path.isdir(cache_dir):
+        print("Creating '" + cache_dir + "'...")
+        os.mkdir(cache_dir)
+
+        print("Creating '" + cache_dir + "/styles'")
+        os.mkdir(cache_dir + '/styles')
+
+    print("Creating style.css")
+    copyfile("./skeletons/style.css", cache_dir + '/styles/style.css')
+
+    print("Creating Hack.ttf")
+    copyfile("./skeletons/Hack.ttf", cache_dir + '/styles/Hack.ttf')
+
+
+def main():
+    setup()
+
+if __name__ == '__main__':
+    main()

install.py

@@ -21,51 +21,10 @@ def prettifyHTML(html):
     prettyHTML = soup.prettify()
     return prettyHTML
 
-def setup():
-    # check if .config path exists
-    if not os.path.isdir(home + '/.config'):
-        print("The directory '~/.config' does not exist, or you do not have permissions to edit it.")
-        exit()
-
-    if not os.path.isdir(home + '/.cache'):
-        print("The directory '~/.cache' does not exist, or you do not have permissions to edit it.")
-        exit()
-
-    # check if .config/StartTree exists, create it and config if not
-    if not os.path.isdir(config_dir):
-        print("Creating '~/.config/StartTree'...")
-        os.mkdir(config_dir)
-        print("Copying config.yaml")
-        copyfile("./config.yaml", config_path)
-        print("")
-
-    # check if config.yaml exists
-    if not os.path.exists(config_path):
-        print("No config.yaml found in '~/.config/StartTree'")
-        print("Copy the example config with:")
-        print("\tcp ./config.yaml $HOME/.config/StartTree/config.yaml")
-        print("or create your own in that directory.")
-        exit()
-
-    # create directory structure in .cache
-    if not os.path.isdir(cache_dir):
-        print("Creating '" + cache_dir + "'...")
-        os.mkdir(cache_dir)
-
-        print("Creating '" + cache_dir + "/styles'")
-        os.mkdir(cache_dir + '/styles')
-
-    print("Creating style.css")
-    copyfile("./skeletons/style.css", cache_dir + '/styles/style.css')
-
-    print("Creating Hack.ttf")
-    copyfile("./skeletons/Hack.ttf", cache_dir + '/styles/Hack.ttf')
-
 def parse_yaml():
     with open(config_path, mode='r') as file:
         file_dict = yaml.full_load(file)
         file.close()
-
     return file_dict
 
 def print_keys(dictionary):
@@ -140,7 +99,6 @@ def gen_html(file_dict):
     print("Done!")
 
 def main():
-    setup()
     file_dict = parse_yaml()
     gen_html(file_dict)