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.
29 lines
714 B
Python
29 lines
714 B
Python
import importlib
|
|
|
|
from .. import errors
|
|
from ..protocol import CompressionMethodByte
|
|
|
|
|
|
def get_compressor_cls(alg):
|
|
try:
|
|
module = importlib.import_module('.' + alg, __name__)
|
|
return module.Compressor
|
|
|
|
except ImportError:
|
|
raise errors.UnknownCompressionMethod(
|
|
"Unknown compression method: '{}'".format(alg)
|
|
)
|
|
|
|
|
|
def get_decompressor_cls(method_type):
|
|
if method_type == CompressionMethodByte.LZ4:
|
|
module = importlib.import_module('.lz4', __name__)
|
|
|
|
elif method_type == CompressionMethodByte.ZSTD:
|
|
module = importlib.import_module('.zstd', __name__)
|
|
|
|
else:
|
|
raise errors.UnknownCompressionMethod()
|
|
|
|
return module.Decompressor
|