23 lines
570 B
Python
23 lines
570 B
Python
|
|
||
|
|
from transport.sql.common import BaseReader , BaseWriter
|
||
|
|
from psycopg2.extensions import register_adapter, AsIs
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
register_adapter(np.int64, AsIs)
|
||
|
|
|
||
|
|
class PG:
|
||
|
|
def __init__(self,**_args):
|
||
|
|
super().__init__(**_args)
|
||
|
|
def get_provider(self):
|
||
|
|
return "postgresql"
|
||
|
|
|
||
|
|
def get_default_port(self):
|
||
|
|
return "5432"
|
||
|
|
class Reader(PG,BaseReader) :
|
||
|
|
def __init__(self,**_args):
|
||
|
|
super().__init__(**_args)
|
||
|
|
class Writer(PG,BaseWriter):
|
||
|
|
def __init__(self,**_args):
|
||
|
|
super().__init__(**_args)
|
||
|
|
|