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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
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))
|
