mirror of
https://github.com/hanzo-ds/driver.git
synced 2026-10-30 20:32:40 +00:00
zeekay
eb74c996a6
Imports read `datastore_driver`, distribution `datastore-driver`, so nothing has to import another product's name to talk to Hanzo Datastore. Cut from upstream 0.2.9 — the version this stack already ran — rather than from upstream HEAD, so it is a PURE RENAME with no behaviour change. Forking from HEAD would have dragged versions along with the name (connect alone jumped 0.8.17 -> 1.6.0, a major), which is a different and much riskier change wearing a rename's clothes. X-ClickHouse-* headers are untouched: they are wire protocol the server sends and expects, and a different token shape from the module name.
59 lines
2.1 KiB
Python
59 lines
2.1 KiB
Python
from ..service import aliases
|
|
from ... import errors
|
|
from .datecolumn import NumpyDateColumn
|
|
from .datetimecolumn import create_numpy_datetime_column
|
|
from .floatcolumn import NumpyFloat32Column, NumpyFloat64Column
|
|
from .intcolumn import (
|
|
NumpyInt8Column, NumpyInt16Column, NumpyInt32Column, NumpyInt64Column,
|
|
NumpyUInt8Column, NumpyUInt16Column, NumpyUInt32Column, NumpyUInt64Column
|
|
)
|
|
from .boolcolumn import NumpyBoolColumn
|
|
from .lowcardinalitycolumn import create_numpy_low_cardinality_column
|
|
from .stringcolumn import create_string_column
|
|
from .tuplecolumn import create_tuple_column
|
|
from ..nullablecolumn import create_nullable_column
|
|
|
|
column_by_type = {c.ch_type: c for c in [
|
|
NumpyDateColumn,
|
|
NumpyFloat32Column, NumpyFloat64Column,
|
|
NumpyInt8Column, NumpyInt16Column, NumpyInt32Column, NumpyInt64Column,
|
|
NumpyUInt8Column, NumpyUInt16Column, NumpyUInt32Column, NumpyUInt64Column,
|
|
NumpyBoolColumn
|
|
]}
|
|
|
|
|
|
def get_numpy_column_by_spec(spec, column_options):
|
|
def create_column_with_options(x):
|
|
return get_numpy_column_by_spec(x, column_options)
|
|
|
|
if spec == 'String' or spec.startswith('FixedString'):
|
|
return create_string_column(spec, column_options)
|
|
|
|
elif spec.startswith('DateTime'):
|
|
return create_numpy_datetime_column(spec, column_options)
|
|
|
|
elif spec.startswith('Tuple'):
|
|
return create_tuple_column(
|
|
spec, create_column_with_options, column_options
|
|
)
|
|
|
|
elif spec.startswith('Nullable'):
|
|
return create_nullable_column(spec, create_column_with_options)
|
|
|
|
elif spec.startswith('LowCardinality'):
|
|
return create_numpy_low_cardinality_column(
|
|
spec, create_column_with_options, column_options
|
|
)
|
|
else:
|
|
for alias, primitive in aliases:
|
|
if spec.startswith(alias):
|
|
return create_column_with_options(
|
|
primitive + spec[len(alias):]
|
|
)
|
|
|
|
if spec in column_by_type:
|
|
cls = column_by_type[spec]
|
|
return cls(**column_options)
|
|
|
|
raise errors.UnknownTypeError('Unknown type {}'.format(spec))
|