Rearrange arguments to allow multiple rule files
This commit is contained in:
parent
94dd62ffc4
commit
15e558c90b
|
@ -1,6 +1,6 @@
|
|||
[metadata]
|
||||
name = ProcruSQL
|
||||
version = 0.0.2
|
||||
version = 0.0.3
|
||||
author = Peter J. Holzer
|
||||
author_email = hjp@hjp.at
|
||||
description = Make a database fit its description
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
import argparse
|
||||
import logging
|
||||
import sys
|
||||
|
||||
|
@ -247,7 +248,17 @@ def dump_dot():
|
|||
|
||||
def main():
|
||||
logging.basicConfig(format="%(asctime)s %(levelname)s %(name)s %(lineno)d | %(message)s", level=logging.DEBUG)
|
||||
with open(sys.argv[1]) as rf:
|
||||
ap = argparse.ArgumentParser()
|
||||
ap.add_argument("--dbname")
|
||||
ap.add_argument("--dbuser")
|
||||
ap.add_argument("files", nargs="+")
|
||||
args = ap.parse_args()
|
||||
|
||||
db = psycopg2.connect(dbname=args.dbname, user=args.dbuser)
|
||||
|
||||
rules = []
|
||||
for f in args.files:
|
||||
with open(f) as rf:
|
||||
text = rf.read()
|
||||
ps = parser.ParseState(text)
|
||||
|
||||
|
@ -256,7 +267,7 @@ def main():
|
|||
if not ps2:
|
||||
ps.printerror()
|
||||
sys.exit(1)
|
||||
rules.extend(ps2.ast)
|
||||
|
||||
db = psycopg2.connect(sys.argv[2])
|
||||
fit(db, ps2.ast)
|
||||
fit(db, rules)
|
||||
|
||||
|
|
Loading…
Reference in New Issue