mdBookShelf/manager.py at main · Hanthebot/mdBookShelf · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
""" Module for Controlling the Junctions and Exporting the Books """

import sys
import os
import json
import shutil
import glob

CONFIGS = ["src", "covers"]

def delete_junctions(directory_: str):
    """ delete all custom subdir """
    delete_path = os.path.join(directory_, "*/")
    for dirs in glob.glob(delete_path):
        if any(config in dirs for config in CONFIGS):
            continue
        try:
            os.unlink(dirs[:-1])
        except PermissionError:
            shutil.rmtree(dirs[:-1])
    for dirs in glob.glob(os.path.join(directory_, "covers/*.png")):
        os.remove(dirs)

def reset_js(data_: dict, js: str = "./src/data.js"):
    """ overwrite the JS data file """
    # perform deep copy here
    temp = data_.copy()
    temp["books"] = {book_code_: val.copy() for book_code_, val in data_["books"].items()}
    for book_code_ in temp["books"]:
        temp["books"][book_code_]["path"] = "./" + book_code_
    with open(js, "w", encoding = "utf-8") as file:
        file.write("var data = " + json.dumps(temp, indent = 4) + ";")

def clear_setting(input_dir: str = "./src/data.json",
                  setup_dir: str = "./") -> dict:
    """ clear the setup and return the data """
    with open(input_dir, "r", encoding = "utf-8") as file:
        data_ = json.load(file)
    reset_js(data_)
    delete_junctions(setup_dir)
    return data_

def serve(book_dir: str, book_code_: str):
    """ Soft copying all books by junctions and ready to serve """
    link_dir_ = os.path.join("./", book_code_)
    cover_path_ = os.path.join(book_dir, "../cover.png")
    os.system(f"mklink /J \"{link_dir_}\" \"{book_dir}\"")
    shutil.copy(cover_path_, f"covers/{book_code_}.png")

def export(book_dir: str, book_code_: str):
    """ Hard copying all books and ready to export """
    link_dir_ = os.path.join("./", book_code_)
    cover_path_ = os.path.join(book_dir, "../cover.png")
    shutil.copytree(book_dir, link_dir_)
    shutil.copy(cover_path_, f"covers/{book_code_}.png")

if __name__ == "__main__":
    if len(sys.argv) == 1 or sys.argv[1].lower() not in ["serve", "export", "clear"]:
        print("Correct usage: manager.py [serve / export / clear]")
        sys.exit()
    command = sys.argv[1].lower()
    for config in CONFIGS:
        if not os.path.exists(config):
            os.mkdir(config)
    data = clear_setting()
    if command == "clear":
        sys.exit()
    i = 0
    for book_code, vals in data["books"].items():
        cover_dir = os.path.join(vals["path"], "../cover.png")
        if not os.path.exists(cover_dir):
            shutil.copy2("src/cover.png", cover_dir)
        link_dir = os.path.join("./", book_code)
        if command == "serve":
            serve(vals["path"], book_code)
        else:
            export(vals["path"], book_code)
        # if the book is not in dir/cover.png, create one
        # copy it to covers/book_code.png
        i += 1
        print(f"Created {i}/{len(data['books'].keys())} links for {book_code}", end = "\t\t\r")
    print()