Skip to content

Commit b747391

Browse files
author
Benjamin Muskalla
committed
Improve error handling and refactor base path
1 parent c1b5565 commit b747391

File tree

1 file changed

+12
-5
lines changed

1 file changed

+12
-5
lines changed

java/ql/src/utils/model-generator/RegenerateModels.py

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,12 @@
1212
import sys
1313

1414

15+
defaultModelPath = "java/ql/lib/semmle/code/java/frameworks"
1516
lgtmSlugToModelFile = {
16-
# "apache/commons-beanutils": "java/ql/lib/semmle/code/java/frameworks/apache/BeanUtilsGenerated.qll",
17-
# "apache/commons-codec": "java/ql/lib/semmle/code/java/frameworks/apache/CodecGenerated.qll",
18-
# "apache/commons-lang": "java/ql/lib/semmle/code/java/frameworks/apache/Lang3Generated.qll",
19-
"apache/commons-io": "java/ql/lib/semmle/code/java/frameworks/apache/IOGenerated.qll",
17+
# "apache/commons-beanutils": "apache/BeanUtilsGenerated.qll",
18+
# "apache/commons-codec": "apache/CodecGenerated.qll",
19+
# "apache/commons-lang": "apache/Lang3Generated.qll",
20+
"apache/commons-io": "apache/IOGenerated.qll",
2021
}
2122

2223

@@ -30,7 +31,13 @@ def regenerateModel(lgtmSlug, extractedDb):
3031
print("============================================================")
3132
print("Generating models for " + lgtmSlug)
3233
print("============================================================")
33-
modelFile = lgtmSlugToModelFile[lgtmSlug]
34+
# check if lgtmSlug exists as key
35+
if lgtmSlug not in lgtmSlugToModelFile:
36+
print("ERROR: slug " + lgtmSlug +
37+
" is not mapped to a model file in script " + sys.argv[0])
38+
sys.exit(1)
39+
modelFile = defaultModelPath + \
40+
lgtmSlugToModelFile[lgtmSlug]
3441
codeQlRoot = findGitRoot()
3542
targetModel = codeQlRoot + "/" + modelFile
3643
subprocess.check_call([codeQlRoot + "/java/ql/src/utils/model-generator/GenerateFlowModel.py", extractedDb,

0 commit comments

Comments
 (0)