blob: 0f83f3c6a70a4964dee4da09b589f151fdcf2f03 (
about) (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
from sqlalchemy.dialects.postgresql.base import ischema_names
from sqlalchemy.types import UserDefinedType, Float
class BIT(UserDefinedType):
cache_ok = True
def __init__(self, length=None):
super(UserDefinedType, self).__init__()
self.length = length
def get_col_spec(self, **kw):
if self.length is None:
return 'BIT'
return 'BIT(%d)' % self.length
class comparator_factory(UserDefinedType.Comparator):
def hamming_distance(self, other):
return self.op('<~>', return_type=Float)(other)
def jaccard_distance(self, other):
return self.op('<%>', return_type=Float)(other)
# for reflection
ischema_names['bit'] = BIT
|