diff options
| -rw-r--r-- | python/markit/loans.py | 3 | ||||
| -rw-r--r-- | python/markit/rates.py | 12 |
2 files changed, 7 insertions, 8 deletions
diff --git a/python/markit/loans.py b/python/markit/loans.py index 16c256b0..f4f13d4e 100644 --- a/python/markit/loans.py +++ b/python/markit/loans.py @@ -1,10 +1,9 @@ import csv import logging -import os import requests from . import with_connection -from psycopg2 import IntegrityError +from psycopg import IntegrityError from . import DATA_DIR logger = logging.getLogger(__name__) diff --git a/python/markit/rates.py b/python/markit/rates.py index 4884d626..a105648a 100644 --- a/python/markit/rates.py +++ b/python/markit/rates.py @@ -2,8 +2,8 @@ import datetime from itertools import chain from serenitas.utils.env import DATA_DIR from io import BytesIO -from psycopg2 import sql -from psycopg2.extras import DateRange +from psycopg import sql +from psycopg.types.range import Range import requests import xml.etree.ElementTree as ET import zipfile @@ -169,7 +169,7 @@ def update_bbg_members(conn, session, curve_type, download_date): "UPDATE bbg_curves SET in_effect=%s " "WHERE in_effect=%s AND curve_type=%s", ( - DateRange(download_date, in_effect.upper), + Range(download_date, in_effect.upper), in_effect, curve_type.value, ), @@ -182,7 +182,7 @@ def update_bbg_members(conn, session, curve_type, download_date): ( curve_type.value, new_members, - DateRange(download_date, in_effect.lower), + Range(download_date, in_effect.lower), ), ) conn.commit() @@ -192,7 +192,7 @@ def update_bbg_members(conn, session, curve_type, download_date): c.execute( "UPDATE bbg_curves SET in_effect=%s WHERE id=%s", ( - DateRange(in_effect.lower, download_date), + Range(in_effect.lower, download_date), curve_id, ), ) @@ -202,7 +202,7 @@ def update_bbg_members(conn, session, curve_type, download_date): ( curve_type.value, new_members, - DateRange(download_date, in_effect.upper), + Range(download_date, in_effect.upper), ), ) conn.commit() |