diff options
Diffstat (limited to 'python/process_queue.py')
| -rw-r--r-- | python/process_queue.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/python/process_queue.py b/python/process_queue.py index 5760883e..c004b6da 100644 --- a/python/process_queue.py +++ b/python/process_queue.py @@ -3,7 +3,7 @@ import blpapi import csv import datetime import logging -import psycopg2 +import psycopg import pathlib import re import redis @@ -70,7 +70,7 @@ def process_indicative( trade_type: str, upload: bool, session: blpapi.session.Session, - conn: psycopg2.extensions.connection, + conn: psycopg.connection, ) -> None: process_fun = globals().get( f"{trade_type}_trade_process", lambda conn, session, trade: trade @@ -812,7 +812,7 @@ if __name__ == "__main__": os.environ["SERENITAS_APP_NAME"] = "process_queue" from functools import partial - from serenitas.utils.db2 import dawn_pool + from serenitas.utils.pool import dawn_pool parser = argparse.ArgumentParser() parser.add_argument( |