diff options
Diffstat (limited to 'python/db.py')
| -rw-r--r-- | python/db.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/python/db.py b/python/db.py index 3c937ba1..0ba5f7b3 100644 --- a/python/db.py +++ b/python/db.py @@ -3,7 +3,7 @@ import os import psycopg2 from psycopg2.extras import DictCursor from psycopg2 import IntegrityError -from psycopg2.extensions import register_adapter +from psycopg2.extensions import register_adapter, AsIs from sqlalchemy import create_engine from sqlalchemy.engine.url import URL import numpy as np @@ -20,6 +20,7 @@ class InfDateAdapter: return psycopg2.extensions.DateFromPy(self.wrapped).getquoted() register_adapter(datetime.date, InfDateAdapter) +register_adapter(np.int64, lambda x: AsIs(x)) def dbconn(dbname): if dbname == 'etdb': |