diff options
Diffstat (limited to 'python/markit/__main__.py')
| -rw-r--r-- | python/markit/__main__.py | 12 |
1 files changed, 5 insertions, 7 deletions
diff --git a/python/markit/__main__.py b/python/markit/__main__.py index c85ef0bc..5fcaaf30 100644 --- a/python/markit/__main__.py +++ b/python/markit/__main__.py @@ -15,9 +15,8 @@ from .loans import ( from .rates import downloadMarkitIRData from .import_quotes import * from pandas.tseries.offsets import BDay -from sqlalchemy import create_engine -from utils import SerenitasFileHandler -from utils.db import serenitas_pool +from serenitas.utils import SerenitasFileHandler +from serenitas.utils.db import serenitas_pool # parse arguments parser = argparse.ArgumentParser() @@ -116,11 +115,10 @@ elif args.cds: remove_curves(conn, workdate) insert_cds(conn, workdate) copy_curves_forward(conn, workdate) - serenitas_pool.putconn(conn) if not args.insert_only: - engine = create_engine("postgresql://serenitas_user@debian/serenitasdb") - insert_index(engine, workdate) - insert_tranche(engine, workdate) + insert_index(conn, workdate) + insert_tranche(conn, workdate) + serenitas_pool.putconn(conn) elif args.rates: conn = serenitas_pool.getconn() |