mirror of
https://github.com/github/codeql.git
synced 2025-12-17 01:03:14 +01:00
91fd83a554
This patch introduces the basic infrastructure of the code generation suite and the `dbscheme` generator. Notice that the checked in `schema.yml` should reflect swift 5.6 but might need some tweaking. Closes https://github.com/github/codeql-c-team/issues/979
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
import argparse
|
|
import collections
|
|
import logging
|
|
import pathlib
|
|
import sys
|
|
|
|
from . import paths
|
|
|
|
options = collections.defaultdict(list)
|
|
|
|
|
|
class Option:
|
|
def __init__(self, *args, tags=None, **kwargs):
|
|
tags = tags or []
|
|
self.args = args
|
|
self.kwargs = kwargs
|
|
if tags:
|
|
for t in tags:
|
|
options[t].append(self)
|
|
else:
|
|
options["*"].append(self)
|
|
|
|
def add_to(self, parser: argparse.ArgumentParser):
|
|
parser.add_argument(*self.args, **self.kwargs)
|
|
|
|
|
|
Option("--check", "-c", action="store_true")
|
|
Option("--verbose", "-v", action="store_true")
|
|
Option("--schema", tags=["schema"], type=pathlib.Path, default=paths.swift_dir / "codegen/schema.yml")
|
|
Option("--dbscheme", tags=["dbscheme"], type=pathlib.Path, default=paths.swift_dir / "ql/lib/swift.dbscheme")
|
|
|
|
|
|
def _parse(*tags):
|
|
parser = argparse.ArgumentParser()
|
|
if not tags:
|
|
opts = [o for os in options.values() for o in os]
|
|
else:
|
|
opts = options["*"]
|
|
for t in tags:
|
|
opts.extend(options[t])
|
|
for opt in opts:
|
|
opt.add_to(parser)
|
|
ret = parser.parse_args()
|
|
log_level = logging.DEBUG if ret.verbose else logging.INFO
|
|
logging.basicConfig(format="{levelname} {message}", style='{', level=log_level)
|
|
return ret
|
|
|
|
|
|
def run(*generate, tags=()):
|
|
opts = _parse(*tags)
|
|
done_something = False
|
|
for g in generate:
|
|
if g(opts):
|
|
done_something = True
|
|
sys.exit(1 if opts.check and done_something else 0)
|