diff options
Diffstat (limited to 'python/handle_default.py')
| -rw-r--r-- | python/handle_default.py | 21 |
1 files changed, 11 insertions, 10 deletions
diff --git a/python/handle_default.py b/python/handle_default.py index 25f4a7a8..c61f071f 100644 --- a/python/handle_default.py +++ b/python/handle_default.py @@ -1,4 +1,4 @@ -from db import dbconn +from db import serenitas_pool import datetime from sys import argv @@ -44,7 +44,7 @@ def update_indexmembers(newids, company_id, conn): and company_id != %s)""", (newid, oldid, company_id)) conn.commit() -def update_redcodes(index_type): +def update_redcodes(index_type, conn): if index_type == 'HY': index_subfamily = 'CDX.NA.HY' elif index_type == 'IG': @@ -73,11 +73,12 @@ if __name__=="__main__": For instance: python handle_default.py 210065 2015-02-19 15.875 100""") else: - with dbconn('serenitasdb') as serenitasdb: - company_id = int(argv[1]) - lastdate = datetime.datetime.strptime(argv[2], "%Y-%m-%d") - recovery = float(argv[3]) - n_issuers = float(argv[4]) - recordslist = affected_indices(company_id, serenitasdb) - newids = create_newindices(recordslist, lastdate, n_issuers, serenitasdb) - update_indexmembers(newids, company_id, serenitasdb) + conn = serenitas_pool.getconn(__name__) + company_id = int(argv[1]) + lastdate = datetime.datetime.strptime(argv[2], "%Y-%m-%d") + recovery = float(argv[3]) + n_issuers = float(argv[4]) + recordslist = affected_indices(company_id, serenitasdb) + newids = create_newindices(recordslist, lastdate, n_issuers, serenitasdb) + update_indexmembers(newids, company_id, serenitasdb) + serenitas_pool.putconn(conn, __name__) |