diff options
Diffstat (limited to 'utils/nwztools/database/gen_db.py')
-rwxr-xr-x | utils/nwztools/database/gen_db.py | 15 |
1 files changed, 13 insertions, 2 deletions
diff --git a/utils/nwztools/database/gen_db.py b/utils/nwztools/database/gen_db.py index be4cc34db0..23f4e3f474 100755 --- a/utils/nwztools/database/gen_db.py +++ b/utils/nwztools/database/gen_db.py | |||
@@ -4,6 +4,17 @@ import os | |||
4 | import re | 4 | import re |
5 | import subprocess | 5 | import subprocess |
6 | import hashlib | 6 | import hashlib |
7 | import sys | ||
8 | |||
9 | # arguments | ||
10 | if len(sys.argv) != 2: | ||
11 | print("Usage: %s output_directory" % sys.argv[0]) | ||
12 | exit(1) | ||
13 | output_directory = sys.argv[1] | ||
14 | # check path is valid | ||
15 | if not os.path.isdir(output_directory): | ||
16 | print("Error: '%s' is not a valid directory" % output_directory) | ||
17 | exit(1) | ||
7 | 18 | ||
8 | # parse models.txt | 19 | # parse models.txt |
9 | g_models = [] | 20 | g_models = [] |
@@ -200,7 +211,7 @@ extern struct nwz_series_info_t nwz_series[NWZ_SERIES_COUNT]; | |||
200 | #endif /* __NWZ_DB_H__ */ | 211 | #endif /* __NWZ_DB_H__ */ |
201 | """ | 212 | """ |
202 | 213 | ||
203 | with open("nwz-db.h", "w") as fp: | 214 | with open(os.path.join(output_directory, "nwz-db.h"), "w") as fp: |
204 | fp.write(header_begin) | 215 | fp.write(header_begin) |
205 | # generate list of all nvp nodes | 216 | # generate list of all nvp nodes |
206 | for name in sorted(g_nvp_names): | 217 | for name in sorted(g_nvp_names): |
@@ -256,7 +267,7 @@ def by_name(nvp_entry): | |||
256 | def codename_to_c(codename): | 267 | def codename_to_c(codename): |
257 | return re.sub('[^a-zA-Z0-9]', '_', codename, 0) | 268 | return re.sub('[^a-zA-Z0-9]', '_', codename, 0) |
258 | 269 | ||
259 | with open("nwz-db.c", "w") as fp: | 270 | with open(os.path.join(output_directory, "nwz-db.c"), "w") as fp: |
260 | fp.write(impl_begin) | 271 | fp.write(impl_begin) |
261 | # generate model list (sort by mid) | 272 | # generate model list (sort by mid) |
262 | for model in sorted(g_models, key = by_mid): | 273 | for model in sorted(g_models, key = by_mid): |