Restore 0.1.5 version from stash
This commit is contained in:
@@ -0,0 +1,21 @@
|
||||
import numpy as np
|
||||
from psycopg2.extensions import adapt, new_type, register_adapter, register_type
|
||||
from ..utils import Vector
|
||||
|
||||
|
||||
class VectorAdapter:
|
||||
def __init__(self, value):
|
||||
self._value = value
|
||||
|
||||
def getquoted(self):
|
||||
return adapt(Vector._to_db(self._value)).getquoted()
|
||||
|
||||
|
||||
def cast_vector(value, cur):
|
||||
return Vector._from_db(value)
|
||||
|
||||
|
||||
def register_vector_info(oid, scope):
|
||||
vector = new_type((oid,), 'VECTOR', cast_vector)
|
||||
register_type(vector, scope)
|
||||
register_adapter(np.ndarray, VectorAdapter)
|
||||
Reference in New Issue
Block a user