21 lines
560 B
Python
21 lines
560 B
Python
|
from psycopg2.extensions import adapt, new_type, register_adapter, register_type
|
||
|
|
from ..utils import SparseVector
|
||
|
|
|
||
|
|
|
||
|
|
class SparsevecAdapter:
|
||
|
|
def __init__(self, value):
|
||
|
|
self._value = value
|
||
|
|
|
||
|
|
def getquoted(self):
|
||
|
|
return adapt(SparseVector._to_db(self._value)).getquoted()
|
||
|
|
|
||
|
|
|
||
|
|
def cast_sparsevec(value, cur):
|
||
|
|
return SparseVector._from_db(value)
|
||
|
|
|
||
|
|
|
||
|
|
def register_sparsevec_info(oid, scope):
|
||
|
|
sparsevec = new_type((oid,), 'SPARSEVEC', cast_sparsevec)
|
||
|
|
register_type(sparsevec, scope)
|
||
|
|
register_adapter(SparseVector, SparsevecAdapter)
|