-
Notifications
You must be signed in to change notification settings - Fork 0
/
advanced_python.py
104 lines (78 loc) · 2.79 KB
/
advanced_python.py
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
import os
import json
import shutil
from subprocess import PIPE, run
import sys
GAME_DIR_PATTERN = "game"
GAME_CODE_EXTENSION = ".go"
GAME_COMPILE_COMMAND = ["go", "build"]
def find_all_game_paths(source):
game_paths = []
for root, dirs, files in os.walk(source):
for directory in dirs:
if GAME_DIR_PATTERN in directory.lower():
path = os.path.join(source, directory)
game_paths.append(path)
break
return game_paths
def get_name_from_paths(paths, to_strip):
new_names = []
for path in paths:
_, dir_name = os.path.split(path)
new_dir_name = dir_name.replace(to_strip, "")
new_names.append(new_dir_name)
return new_names
def create_dir(path):
if not os.path.exists(path):
os.mkdir(path)
def copy_and_overwrite(source, dest):
if os.path.exists(dest):
shutil.rmtree(dest)
shutil.copytree(source, dest)
def make_json_metadata_file(path, game_dirs):
data = {
"gameNames": game_dirs,
"numberOfGames": len(game_dirs)
}
with open(path, "w") as f:
json.dump(data, f)
def compile_game_code(path):
code_file_name = None
for root, dirs, files in os.walk(path):
for file in files:
if file.endswith(GAME_CODE_EXTENSION):
code_file_name = file
break
break
if code_file_name is None:
return
command = GAME_COMPILE_COMMAND + [code_file_name]
run_command(command, path)
def run_command(command, path):
cwd = os.getcwd()
os.chdir(path)
result = run(command, stdout=PIPE, stdin=PIPE, universal_newlines=True)
print("compile result", result)
os.chdir(cwd)
def main(source, target):
cwd = os.getcwd()
source_path = os.path.join(cwd, source)
target_path = os.path.join(cwd, target)
game_paths = find_all_game_paths(source_path)
new_game_dirs = get_name_from_paths(game_paths, "_game")
create_dir(target_path)
#zip combines mult arrays into a tuple
for src, dest in zip(game_paths, new_game_dirs):
dest_path = os.path.join(target_path, dest)
copy_and_overwrite(src, dest_path)
compile_game_code(dest_path)
json_path = os.path.join(target_path, "metadata.json")
make_json_metadata_file(json_path, new_game_dirs)
#only want to execute the main script if I'm running this python file directly
#prevents it from running the code any time a function or class is imported from this code
if __name__ == "__main__":
args = sys.argv
if len(args) != 3:
raise Exception("You must pass a source and target directory - only.")
source, target = args[1:]
main(source, target)