hohn/codeql
1
0
Fork 0
mirror of https://github.com/github/codeql.git synced 2025-12-16 16:53:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

C#: Update model conversion queries

Browse Source
This commit is contained in:
Tom Hvitved
2023-11-02 11:07:57 +01:00
parent 6f4311d656
commit 74f483a6f5
8 changed files with 101 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
misc/scripts/models-as-data/convert_extensions.py
View File

@@ -29,10 +29,12 @@ def merge(*dicts):
return merged
def parseData(data):
rows = { }
rows = [{ }, { }]
for row in data:
d = map(quote_if_needed, row)
insert_update(rows, row[0], " - [" + ', '.join(d) + ']\n')
provenance = row[-1]
targetRows = rows[1] if provenance.endswith("generated") else rows[0]
insert_update(targetRows, row[0], " - [" + ', '.join(d) + ']\n')
return rows
@@ -57,9 +59,10 @@ class Converter:
def asAddsTo(self, rows, predicate):
extensions = { }
for key in rows:
extensions[key] = helpers.addsToTemplate.format(f"codeql/{self.language}-all", predicate, rows[key])
extensions = [{ }, { }]
for i in range(2):
for key in rows[i]:
extensions[i][key] = helpers.addsToTemplate.format(f"codeql/{self.language}-all", predicate, rows[i][key])
return extensions
@@ -75,7 +78,7 @@ class Converter:
sources = self.getAddsTo("ExtractSources.ql", helpers.sourceModelPredicate)
sinks = self.getAddsTo("ExtractSinks.ql", helpers.sinkModelPredicate)
neutrals = self.getAddsTo("ExtractNeutrals.ql", helpers.neutralModelPredicate)
return merge(sources, sinks, summaries, neutrals)
return [merge(sources[0], sinks[0], summaries[0], neutrals[0]), merge(sources[1], sinks[1], summaries[1], neutrals[1])]
def save(self, extensions):
@@ -85,9 +88,13 @@ class Converter:
# Create a file for each namespace and save models.
extensionTemplate = """extensions:
{0}"""
for entry in extensions:
for entry in extensions[0]:
with open(self.extDir + "/" + entry + self.modelFileExtension, "w") as f:
f.write(extensionTemplate.format(extensions[entry]))
f.write(extensionTemplate.format(extensions[0][entry]))
for entry in extensions[1]:
with open(self.extDir + "/generated/" + entry + self.modelFileExtension, "w") as f:
f.write(extensionTemplate.format(extensions[1][entry]))
def run(self):
extensions = self.makeContent()