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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
|
#!/usr/bin/env python3
import os
import os.path
print("It's happy to see you!\n")
# get script dir in relative path
script_dir = os.path.dirname(__file__)
template_dir = script_dir
project_dir = os.path.normpath(os.path.join(script_dir, "../"))
print("First let's check all the templates...")
# get all filenames ending with .template
filenames = [os.path.basename(f)[:-len('.template')] for f in os.listdir(
template_dir) if f.endswith(".template")]
print("I have found following template files:")
for filename in filenames:
print(filename)
print("")
print("Now let's check if they are already generated...")
conflict = False
# check if there exists any generated files
for filename in filenames:
if os.path.exists(os.path.join(project_dir, filename)):
print(f"Found {filename}")
conflict = True
if conflict:
print("It seems there are some files already generated. Do you want to overwrite them? (y/N)")
if input() != "y":
print("Great! Check the existing files and see you next time!")
exit()
else:
print("No conflict found. Let's go on!\n")
print("Check for existing config file...")
config_path = os.path.join(project_dir, "config")
# check if there exists a config file
if not os.path.exists(config_path):
print("No existing config file found. Don't worry. Let's create one! Just tell me your domain name:")
domain = input()
my_uid = os.getuid()
my_gid = os.getgid()
halo_db_password = os.urandom(8).hex()
config_content = f"CRUPEST_DOMAIN={domain}\nCRUPEST_UID={my_uid}\nCRUPEST_GID={my_gid}\nCRUPEST_HALO_DB_PASSWORD={halo_db_password}\n"
# write config file
with open(config_path, "w") as f:
f.write(config_content)
print(
f"Everything else is auto generated. The config file is written into {config_path}. And here is the content:")
print(config_content)
print("If you think it's not ok, you can stop here and edit it. Or let's go on? (Y/n)")
if input() == "n":
print("Great! Check the config file and see you next time!")
exit()
else:
print("Looks like you have already had a config file. Let's check the content:")
with open(config_path, "r") as f:
print(f.read())
print("Is it good enough? (Y/n)")
if input() == "n":
print("Great! Check the config file and see you next time!")
exit()
# Parse config file
with open(config_path, "r") as f:
config = {}
# read line with line number
for line_number, line in enumerate(f.readlines()):
# check if it's a comment
if line.startswith("#"):
continue
# check if there is a '='
if line.find("=") == -1:
print(
f"Invalid config file. Please check line {line_number + 1}. There is even no '='! Aborted!")
# split at first '='
key, value = line.split("=", 1)
key = key.strip()
value = value.strip()
config[key] = value
required_config_keys = ["CRUPEST_DOMAIN", "CRUPEST_UID",
"CRUPEST_GID", "CRUPEST_HALO_DB_PASSWORD"]
# check if all required keys are in config
for key in required_config_keys:
if key not in config:
print(
f"Invalid config file. Please check if {key} is in the config file. Aborted!")
exit()
print("Finally, everything is ready. Let's generate the files:")
# generate files
for filename in filenames:
print(f"Generating {filename}...")
with open(os.path.join(template_dir, filename + ".template"), "r") as f:
content = f.read()
for key in config:
content = content.replace("{{" + key + "}}", config[key])
with open(os.path.join(project_dir, filename), "w") as f:
f.write(content)
print("\n🍻All done! See you next time!")
|
