diff options
Diffstat (limited to 'python/load_indicative.py')
| -rw-r--r-- | python/load_indicative.py | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/python/load_indicative.py b/python/load_indicative.py index 85f02e47..a692c534 100644 --- a/python/load_indicative.py +++ b/python/load_indicative.py @@ -7,11 +7,12 @@ import pdb import sys from common import root, sanitize_float from db import conn +import logging def convertToNone(s): return None if s in ["", "-", "NR"] else s -def upload_cusip_data(conn, filename, logger=None): +def upload_cusip_data(conn, filename): dealupdate = {} with open( filename, "r") as fh: dr = csv.DictReader(fh, dialect='excel-tab') @@ -69,8 +70,8 @@ def upload_cusip_data(conn, filename, logger=None): with conn.cursor() as c: c.execute(sqlstring, line) except psycopg2.DataError as e: - logger.info(e) - print("uploaded: {0}".format(line['CUSIP'])) + logging.error(e) + logging.debug("uploaded: {0}".format(line['CUSIP'])) conn.commit() for dealname in deals_to_update: with conn.cursor() as c: @@ -136,10 +137,10 @@ def upload_deal_data(conn, filename): c.execute(sqlstring, line) deallist[dealname] = line['Latest Update'] except psycopg2.DataError as detail: - print(detail) + logging.error(detail) pdb.set_trace() except KeyError as detail: - print(detail) + logging.error(detail) pdb.set_trace() conn.commit() |