From 8af35183f1f25e2f9d1d111cac85a48622f22673 Mon Sep 17 00:00:00 2001 From: Trygve Laugstøl Date: Tue, 27 Dec 2016 22:49:58 +0100 Subject: o Replacing open_database with with_database to make sure it is closed after use. --- trygvis/eda/cli/add_to_db.py | 23 ++++++++++++----------- 1 file changed, 12 insertions(+), 11 deletions(-) (limited to 'trygvis/eda/cli/add_to_db.py') diff --git a/trygvis/eda/cli/add_to_db.py b/trygvis/eda/cli/add_to_db.py index 0511850..e3b07c1 100644 --- a/trygvis/eda/cli/add_to_db.py +++ b/trygvis/eda/cli/add_to_db.py @@ -2,16 +2,17 @@ import trygvis.eda.cli as cli def run(files, path, args): - g = cli.open_database(path) + def load(g): + s = 0 + for f in files: + cli.info("Adding %s" % f) + pre = len(g) + g.load(f, format="turtle") + post = len(g) + diff = post - pre + s += diff + cli.info("Loaded %d tuples" % diff) - s = 0 - for f in files: - cli.info("Adding %s" % f) - pre = len(g) - g.load(f, format="turtle") - post = len(g) - diff = post - pre - s += diff - cli.info("Loaded %d tuples" % diff) + cli.info("Done. Loaded %d tuples" % s) - cli.info("Done. Loaded %d tuples" % s) + cli.with_database(path, load) -- cgit v1.2.3