StartTree/install.py

#!/bin/python3

import yaml
import os
from os.path import expanduser
from shutil import copyfile

# 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 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):
    for key in dictionary:
        print(key)
        if isinstance(dictionary[key], dict):
            print_keys(dictionary[key])

def gen_columns(html_file):
    print("here")

def gen_html():
    print("Generating index.html...")

    # open files
    skeleton_html = open('./skeletons/index.html', 'r')
    cache_html = open(cache_dir + '/index.html', 'w')

    # copy skeleton_html to cache_html until Column Start comment
    lines = skeleton_html.readlines()
    for line in lines:
        if line == "<!-- Columns start -->":
            gen_columns(cache_html)
        else:
            cache_html.write(line)

    # close files
    skeleton_html.close()
    cache_html.close()

    print("Done!")

if __name__ == '__main__':
    setup()
    file_dict = parse_yaml()
    gen_html()