meerschaum 2.9.4__py3-none-any.whl → 3.0.0__py3-none-any.whl

meerschaum/utils/dtypes/sql.py

@@ -8,6 +8,8 @@ Utility functions for working with SQL data types.
 
 from __future__ import annotations
 from meerschaum.utils.typing import Dict, Union, Tuple, Optional
+from meerschaum._internal.static import STATIC_CONFIG as _STATIC_CONFIG
+from meerschaum.utils.dtypes import MRSM_PRECISION_UNITS_ABBREVIATIONS as _MRSM_PRECISION_UNITS_ABBREVIATIONS
 
 NUMERIC_PRECISION_FLAVORS: Dict[str, Tuple[int, int]] = {
     'mariadb': (38, 20),
@@ -67,6 +69,9 @@ for _flavor, (_precision, _scale) in NUMERIC_PRECISION_FLAVORS.items():
         'DECIMAL': f"DECIMAL({_precision}, {_scale})",
     })
 
+_default_precision_unit = _STATIC_CONFIG['dtypes']['datetime']['default_precision_unit']
+_default_precision_abbreviation = _MRSM_PRECISION_UNITS_ABBREVIATIONS[_default_precision_unit]
+
 DB_TO_PD_DTYPES: Dict[str, Union[str, Dict[str, str]]] = {
     'FLOAT': 'float64[pyarrow]',
     'REAL': 'float64[pyarrow]',
@@ -80,15 +85,21 @@ DB_TO_PD_DTYPES: Dict[str, Union[str, Dict[str, str]]] = {
     'NUMERIC': 'numeric',
     'GEOMETRY': 'geometry',
     'GEOMETRY(GEOMETRY)': 'geometry',
-    'TIMESTAMP': 'datetime64[ns]',
-    'TIMESTAMP WITHOUT TIMEZONE': 'datetime64[ns]',
-    'TIMESTAMP WITH TIMEZONE': 'datetime64[ns, UTC]',
-    'TIMESTAMP WITH TIME ZONE': 'datetime64[ns, UTC]',
-    'TIMESTAMPTZ': 'datetime64[ns, UTC]',
-    'DATE': 'datetime64[ns]',
-    'DATETIME': 'datetime64[ns]',
-    'DATETIME2': 'datetime64[ns]',
-    'DATETIMEOFFSET': 'datetime64[ns, UTC]',
+    'TIMESTAMP': f'datetime64[{_default_precision_abbreviation}]',
+    'TIMESTAMP WITHOUT TIMEZONE': f'datetime64[{_default_precision_abbreviation}]',
+    'TIMESTAMP WITH TIMEZONE': f'datetime64[{_default_precision_abbreviation}, UTC]',
+    'TIMESTAMP WITH TIME ZONE': f'datetime64[{_default_precision_abbreviation}, UTC]',
+    'TIMESTAMPTZ': f'datetime64[{_default_precision_abbreviation}, UTC]',
+    'DATE': 'date',
+    'DATETIME': f'datetime64[{_default_precision_abbreviation}]',
+    'DATETIME2': 'datetime64[us]',
+    'DATETIME2(6)': 'datetime64[us]',
+    'DATETIME2(3)': 'datetime64[ms]',
+    'DATETIME2(0)': 'datetime64[s]',
+    'DATETIMEOFFSET': 'datetime64[us, UTC]',
+    'DATETIMEOFFSET(6)': 'datetime64[us, UTC]',
+    'DATETIMEOFFSET(3)': 'datetime64[ms, UTC]',
+    'DATETIMEOFFSET(0)': 'datetime64[s, UTC]',
     'TEXT': 'string[pyarrow]',
     'VARCHAR': 'string[pyarrow]',
     'CLOB': 'string[pyarrow]',
@@ -109,9 +120,9 @@ DB_TO_PD_DTYPES: Dict[str, Union[str, Dict[str, str]]] = {
     'VARBINARY(MAX)': 'bytes',
     'substrings': {
         'CHAR': 'string[pyarrow]',
-        'TIMESTAMP': 'datetime64[ns]',
-        'TIME': 'datetime64[ns]',
-        'DATE': 'datetime64[ns]',
+        'TIMESTAMP': f'datetime64[{_default_precision_abbreviation}]',
+        'TIME': f'datetime64[{_default_precision_abbreviation}]',
+        'DATE': 'date',
         'DOUBLE': 'double[pyarrow]',
         'DECIMAL': 'numeric',
         'NUMERIC': 'numeric',
@@ -131,6 +142,7 @@ DB_TO_PD_DTYPES: Dict[str, Union[str, Dict[str, str]]] = {
 PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     'int': {
         'timescaledb': 'BIGINT',
+        'timescaledb-ha': 'BIGINT',
         'postgresql': 'BIGINT',
         'postgis': 'BIGINT',
         'mariadb': 'BIGINT',
@@ -145,6 +157,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'uint': {
         'timescaledb': 'BIGINT',
+        'timescaledb-ha': 'BIGINT',
         'postgresql': 'BIGINT',
         'postgis': 'BIGINT',
         'mariadb': 'BIGINT',
@@ -159,6 +172,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'int8': {
         'timescaledb': 'SMALLINT',
+        'timescaledb-ha': 'SMALLINT',
         'postgresql': 'SMALLINT',
         'postgis': 'SMALLINT',
         'mariadb': 'SMALLINT',
@@ -173,6 +187,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'uint8': {
         'timescaledb': 'SMALLINT',
+        'timescaledb-ha': 'SMALLINT',
         'postgresql': 'SMALLINT',
         'postgis': 'SMALLINT',
         'mariadb': 'SMALLINT',
@@ -187,6 +202,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'int16': {
         'timescaledb': 'SMALLINT',
+        'timescaledb-ha': 'SMALLINT',
         'postgresql': 'SMALLINT',
         'postgis': 'SMALLINT',
         'mariadb': 'SMALLINT',
@@ -201,6 +217,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'int32': {
         'timescaledb': 'INT',
+        'timescaledb-ha': 'INT',
         'postgresql': 'INT',
         'postgis': 'INT',
         'mariadb': 'INT',
@@ -215,6 +232,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'int64': {
         'timescaledb': 'BIGINT',
+        'timescaledb-ha': 'BIGINT',
         'postgresql': 'BIGINT',
         'postgis': 'BIGINT',
         'mariadb': 'BIGINT',
@@ -229,6 +247,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'float': {
         'timescaledb': 'DOUBLE PRECISION',
+        'timescaledb-ha': 'DOUBLE PRECISION',
         'postgresql': 'DOUBLE PRECISION',
         'postgis': 'DOUBLE PRECISION',
         'mariadb': 'DOUBLE PRECISION',
@@ -243,6 +262,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'double': {
         'timescaledb': 'DOUBLE PRECISION',
+        'timescaledb-ha': 'DOUBLE PRECISION',
         'postgresql': 'DOUBLE PRECISION',
         'postgis': 'DOUBLE PRECISION',
         'mariadb': 'DOUBLE PRECISION',
@@ -256,41 +276,134 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
         'default': 'DOUBLE',
     },
     'datetime64[ns]': {
-        'timescaledb': 'TIMESTAMP',
-        'postgresql': 'TIMESTAMP',
-        'postgis': 'TIMESTAMP',
+        'timescaledb': 'TIMESTAMP(6)',
+        'timescaledb-ha': 'TIMESTAMP(6)',
+        'postgresql': 'TIMESTAMP(6)',
+        'postgis': 'TIMESTAMP(6)',
         'mariadb': 'DATETIME',
         'mysql': 'DATETIME',
-        'mssql': 'DATETIME2',
+        'mssql': 'DATETIME2(7)',
         'oracle': 'TIMESTAMP(9)',
         'sqlite': 'DATETIME',
-        'duckdb': 'TIMESTAMP',
-        'citus': 'TIMESTAMP',
-        'cockroachdb': 'TIMESTAMP',
+        'duckdb': 'TIMESTAMP(6)',
+        'citus': 'TIMESTAMP(6)',
+        'cockroachdb': 'TIMESTAMP(6)',
         'default': 'DATETIME',
     },
     'datetime64[ns, UTC]': {
-        'timescaledb': 'TIMESTAMPTZ',
-        'postgresql': 'TIMESTAMPTZ',
-        'postgis': 'TIMESTAMPTZ',
+        'timescaledb': 'TIMESTAMPTZ(6)',
+        'timescaledb-ha': 'TIMESTAMPTZ(6)',
+        'postgresql': 'TIMESTAMPTZ(9)',
+        'postgis': 'TIMESTAMPTZ(6)',
         'mariadb': 'DATETIME',
         'mysql': 'DATETIME',
-        'mssql': 'DATETIMEOFFSET',
+        'mssql': 'DATETIMEOFFSET(7)',
         'oracle': 'TIMESTAMP(9)',
         'sqlite': 'TIMESTAMP',
         'duckdb': 'TIMESTAMPTZ',
-        'citus': 'TIMESTAMPTZ',
-        'cockroachdb': 'TIMESTAMPTZ',
+        'citus': 'TIMESTAMPTZ(6)',
+        'cockroachdb': 'TIMESTAMPTZ(6)',
+        'default': 'TIMESTAMPTZ',
+    },
+    'datetime64[us]': {
+        'timescaledb': 'TIMESTAMP(6)',
+        'timescaledb-ha': 'TIMESTAMP(6)',
+        'postgresql': 'TIMESTAMP(6)',
+        'postgis': 'TIMESTAMP(6)',
+        'mariadb': 'DATETIME',
+        'mysql': 'DATETIME',
+        'mssql': 'DATETIME2(6)',
+        'oracle': 'TIMESTAMP(6)',
+        'sqlite': 'DATETIME',
+        'duckdb': 'TIMESTAMP(6)',
+        'citus': 'TIMESTAMP(6)',
+        'cockroachdb': 'TIMESTAMP',
+        'default': 'DATETIME',
+    },
+    'datetime64[us, UTC]': {
+        'timescaledb': 'TIMESTAMPTZ(6)',
+        'timescaledb-ha': 'TIMESTAMPTZ(6)',
+        'postgresql': 'TIMESTAMPTZ(6)',
+        'postgis': 'TIMESTAMPTZ(6)',
+        'mariadb': 'DATETIME',
+        'mysql': 'DATETIME',
+        'mssql': 'DATETIMEOFFSET(6)',
+        'oracle': 'TIMESTAMP(6)',
+        'sqlite': 'TIMESTAMP',
+        'duckdb': 'TIMESTAMPTZ',
+        'citus': 'TIMESTAMPTZ(6)',
+        'cockroachdb': 'TIMESTAMPTZ(6)',
         'default': 'TIMESTAMPTZ',
     },
+    'datetime64[ms]': {
+        'timescaledb': 'TIMESTAMP(3)',
+        'timescaledb-ha': 'TIMESTAMP(3)',
+        'postgresql': 'TIMESTAMP(3)',
+        'postgis': 'TIMESTAMP(3)',
+        'mariadb': 'DATETIME',
+        'mysql': 'DATETIME',
+        'mssql': 'DATETIME2(3)',
+        'oracle': 'TIMESTAMP(3)',
+        'sqlite': 'DATETIME',
+        'duckdb': 'TIMESTAMP(3)',
+        'citus': 'TIMESTAMP(3)',
+        'cockroachdb': 'TIMESTAMP(3)',
+        'default': 'DATETIME',
+    },
+    'datetime64[ms, UTC]': {
+        'timescaledb': 'TIMESTAMPTZ(3)',
+        'timescaledb-ha': 'TIMESTAMPTZ(3)',
+        'postgresql': 'TIMESTAMPTZ(3)',
+        'postgis': 'TIMESTAMPTZ(3)',
+        'mariadb': 'DATETIME',
+        'mysql': 'DATETIME',
+        'mssql': 'DATETIMEOFFSET(3)',
+        'oracle': 'TIMESTAMP(3)',
+        'sqlite': 'TIMESTAMP',
+        'duckdb': 'TIMESTAMPTZ',
+        'citus': 'TIMESTAMPTZ(3)',
+        'cockroachdb': 'TIMESTAMPTZ(3)',
+        'default': 'TIMESTAMPTZ',
+    },
+    'datetime64[s]': {
+        'timescaledb': 'TIMESTAMP(0)',
+        'timescaledb-ha': 'TIMESTAMP(0)',
+        'postgresql': 'TIMESTAMP(0)',
+        'postgis': 'TIMESTAMP(0)',
+        'mariadb': 'DATETIME',
+        'mysql': 'DATETIME',
+        'mssql': 'DATETIME2(0)',
+        'oracle': 'TIMESTAMP(0)',
+        'sqlite': 'DATETIME',
+        'duckdb': 'TIMESTAMP(0)',
+        'citus': 'TIMESTAMP(0)',
+        'cockroachdb': 'TIMESTAMP(0)',
+        'default': 'DATETIME',
+    },
+    'datetime64[s, UTC]': {
+        'timescaledb': 'TIMESTAMPTZ(0)',
+        'timescaledb-ha': 'TIMESTAMPTZ(0)',
+        'postgresql': 'TIMESTAMPTZ(0)',
+        'postgis': 'TIMESTAMPTZ(0)',
+        'mariadb': 'DATETIME',
+        'mysql': 'DATETIME',
+        'mssql': 'DATETIMEOFFSET(0)',
+        'oracle': 'TIMESTAMP(0)',
+        'sqlite': 'TIMESTAMP',
+        'duckdb': 'TIMESTAMPTZ',
+        'citus': 'TIMESTAMPTZ(0)',
+        'cockroachdb': 'TIMESTAMPTZ(0)',
+        'default': 'TIMESTAMPTZ(0)',
+    },
     'datetime': {
         'timescaledb': 'TIMESTAMPTZ',
+        'timescaledb-ha': 'TIMESTAMPTZ',
         'postgresql': 'TIMESTAMPTZ',
         'postgis': 'TIMESTAMPTZ',
         'mariadb': 'DATETIME',
         'mysql': 'DATETIME',
         'mssql': 'DATETIMEOFFSET',
-        'oracle': 'TIMESTAMP(9)',
+        'oracle': 'TIMESTAMP',
         'sqlite': 'TIMESTAMP',
         'duckdb': 'TIMESTAMPTZ',
         'citus': 'TIMESTAMPTZ',
@@ -299,20 +412,67 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'datetimetz': {
         'timescaledb': 'TIMESTAMPTZ',
+        'timescaledb-ha': 'TIMESTAMPTZ',
         'postgresql': 'TIMESTAMPTZ',
         'postgis': 'TIMESTAMPTZ',
         'mariadb': 'DATETIME',
         'mysql': 'DATETIME',
         'mssql': 'DATETIMEOFFSET',
-        'oracle': 'TIMESTAMP(9)',
+        'oracle': 'TIMESTAMP',
         'sqlite': 'TIMESTAMP',
         'duckdb': 'TIMESTAMPTZ',
         'citus': 'TIMESTAMPTZ',
         'cockroachdb': 'TIMESTAMPTZ',
         'default': 'TIMESTAMPTZ',
     },
+    'date': {
+        'timescaledb': 'DATE',
+        'timescaledb-ha': 'DATE',
+        'postgresql': 'DATE',
+        'postgis': 'DATE',
+        'mariadb': 'DATE',
+        'mysql': 'DATE',
+        'mssql': 'DATE',
+        'oracle': 'DATE',
+        'sqlite': 'DATE',
+        'duckdb': 'DATE',
+        'citus': 'DATE',
+        'cockroachdb': 'DATE',
+        'default': 'DATE',
+    },
     'bool': {
         'timescaledb': 'BOOLEAN',
+        'timescaledb-ha': 'BOOLEAN',
+        'postgresql': 'BOOLEAN',
+        'postgis': 'BOOLEAN',
+        'mariadb': 'BOOLEAN',
+        'mysql': 'BOOLEAN',
+        'mssql': 'BIT',
+        'oracle': 'INTEGER',
+        'sqlite': 'FLOAT',
+        'duckdb': 'BOOLEAN',
+        'citus': 'BOOLEAN',
+        'cockroachdb': 'BOOLEAN',
+        'default': 'BOOLEAN',
+    },
+    'bool[pyarrow]': {
+        'timescaledb': 'BOOLEAN',
+        'timescaledb-ha': 'BOOLEAN',
+        'postgresql': 'BOOLEAN',
+        'postgis': 'BOOLEAN',
+        'mariadb': 'BOOLEAN',
+        'mysql': 'BOOLEAN',
+        'mssql': 'BIT',
+        'oracle': 'INTEGER',
+        'sqlite': 'FLOAT',
+        'duckdb': 'BOOLEAN',
+        'citus': 'BOOLEAN',
+        'cockroachdb': 'BOOLEAN',
+        'default': 'BOOLEAN',
+    },
+    'boolean': {
+        'timescaledb': 'BOOLEAN',
+        'timescaledb-ha': 'BOOLEAN',
         'postgresql': 'BOOLEAN',
         'postgis': 'BOOLEAN',
         'mariadb': 'BOOLEAN',
@@ -327,6 +487,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'object': {
         'timescaledb': 'TEXT',
+        'timescaledb-ha': 'TEXT',
         'postgresql': 'TEXT',
         'postgis': 'TEXT',
         'mariadb': 'TEXT',
@@ -341,6 +502,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'string': {
         'timescaledb': 'TEXT',
+        'timescaledb-ha': 'TEXT',
         'postgresql': 'TEXT',
         'postgis': 'TEXT',
         'mariadb': 'TEXT',
@@ -355,6 +517,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'unicode': {
         'timescaledb': 'TEXT',
+        'timescaledb-ha': 'TEXT',
         'postgresql': 'TEXT',
         'postgis': 'TEXT',
         'mariadb': 'TEXT',
@@ -369,6 +532,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'json': {
         'timescaledb': 'JSONB',
+        'timescaledb-ha': 'JSONB',
         'postgresql': 'JSONB',
         'postgis': 'JSONB',
         'mariadb': 'TEXT',
@@ -383,6 +547,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'numeric': {
         'timescaledb': 'NUMERIC',
+        'timescaledb-ha': 'NUMERIC',
         'postgresql': 'NUMERIC',
         'postgis': 'NUMERIC',
         'mariadb': f'DECIMAL{NUMERIC_PRECISION_FLAVORS["mariadb"]}',
@@ -397,6 +562,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'uuid': {
         'timescaledb': 'UUID',
+        'timescaledb-ha': 'UUID',
         'postgresql': 'UUID',
         'postgis': 'UUID',
         'mariadb': 'CHAR(36)',
@@ -412,6 +578,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'bytes': {
         'timescaledb': 'BYTEA',
+        'timescaledb-ha': 'BYTEA',
         'postgresql': 'BYTEA',
         'postgis': 'BYTEA',
         'mariadb': 'BLOB',
@@ -426,6 +593,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'geometry': {
         'timescaledb': 'TEXT',
+        'timescaledb-ha': 'GEOMETRY',
         'postgresql': 'TEXT',
         'postgis': 'GEOMETRY',
         'mariadb': 'TEXT',
@@ -440,6 +608,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'geography': {
         'timescaledb': 'TEXT',
+        'timescaledb-ha': 'GEOGRAPHY',
         'postgresql': 'TEXT',
         'postgis': 'GEOGRAPHY',
         'mariadb': 'TEXT',
@@ -456,6 +625,7 @@ PD_TO_DB_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
 PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     'int': {
         'timescaledb': 'BigInteger',
+        'timescaledb-ha': 'BigInteger',
         'postgresql': 'BigInteger',
         'postgis': 'BigInteger',
         'mariadb': 'BigInteger',
@@ -470,6 +640,7 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'uint': {
         'timescaledb': 'BigInteger',
+        'timescaledb-ha': 'BigInteger',
         'postgresql': 'BigInteger',
         'postgis': 'BigInteger',
         'mariadb': 'BigInteger',
@@ -484,6 +655,7 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'int8': {
         'timescaledb': 'SmallInteger',
+        'timescaledb-ha': 'SmallInteger',
         'postgresql': 'SmallInteger',
         'postgis': 'SmallInteger',
         'mariadb': 'SmallInteger',
@@ -498,6 +670,7 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'uint8': {
         'timescaledb': 'SmallInteger',
+        'timescaledb-ha': 'SmallInteger',
         'postgresql': 'SmallInteger',
         'postgis': 'SmallInteger',
         'mariadb': 'SmallInteger',
@@ -512,6 +685,7 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'int16': {
         'timescaledb': 'SmallInteger',
+        'timescaledb-ha': 'SmallInteger',
         'postgresql': 'SmallInteger',
         'postgis': 'SmallInteger',
         'mariadb': 'SmallInteger',
@@ -526,6 +700,7 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'int32': {
         'timescaledb': 'Integer',
+        'timescaledb-ha': 'Integer',
         'postgresql': 'Integer',
         'postgis': 'Integer',
         'mariadb': 'Integer',
@@ -540,6 +715,7 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'int64': {
         'timescaledb': 'BigInteger',
+        'timescaledb-ha': 'BigInteger',
         'postgresql': 'BigInteger',
         'postgis': 'BigInteger',
         'mariadb': 'BigInteger',
@@ -554,6 +730,7 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'float': {
         'timescaledb': 'Float',
+        'timescaledb-ha': 'Float',
         'postgresql': 'Float',
         'postgis': 'Float',
         'mariadb': 'Float',
@@ -568,12 +745,13 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'datetime': {
         'timescaledb': 'DateTime(timezone=True)',
+        'timescaledb-ha': 'DateTime(timezone=True)',
         'postgresql': 'DateTime(timezone=True)',
         'postgis': 'DateTime(timezone=True)',
         'mariadb': 'DateTime(timezone=True)',
         'mysql': 'DateTime(timezone=True)',
         'mssql': 'sqlalchemy.dialects.mssql.DATETIMEOFFSET',
-        'oracle': 'sqlalchemy.dialects.oracle.TIMESTAMP(timezone=True)',
+        'oracle': 'sqlalchemy.dialects.oracle.TIMESTAMP',
         'sqlite': 'DateTime(timezone=True)',
         'duckdb': 'DateTime(timezone=True)',
         'citus': 'DateTime(timezone=True)',
@@ -582,6 +760,7 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'datetime64[ns]': {
         'timescaledb': 'DateTime',
+        'timescaledb-ha': 'DateTime',
         'postgresql': 'DateTime',
         'postgis': 'DateTime',
         'mariadb': 'DateTime',
@@ -596,20 +775,157 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'datetime64[ns, UTC]': {
         'timescaledb': 'DateTime(timezone=True)',
+        'timescaledb-ha': 'DateTime(timezone=True)',
         'postgresql': 'DateTime(timezone=True)',
         'postgis': 'DateTime(timezone=True)',
         'mariadb': 'DateTime(timezone=True)',
         'mysql': 'DateTime(timezone=True)',
         'mssql': 'sqlalchemy.dialects.mssql.DATETIMEOFFSET',
-        'oracle': 'sqlalchemy.dialects.oracle.TIMESTAMP(timezone=True)',
+        'oracle': 'sqlalchemy.dialects.oracle.TIMESTAMP',
         'sqlite': 'DateTime(timezone=True)',
         'duckdb': 'DateTime(timezone=True)',
         'citus': 'DateTime(timezone=True)',
         'cockroachdb': 'DateTime(timezone=True)',
         'default': 'DateTime(timezone=True)',
     },
+    'datetime64[us]': {
+        'timescaledb': 'DateTime',
+        'timescaledb-ha': 'DateTime',
+        'postgresql': 'DateTime',
+        'postgis': 'DateTime',
+        'mariadb': 'DateTime',
+        'mysql': 'DateTime',
+        'mssql': 'sqlalchemy.dialects.mssql.DATETIME2',
+        'oracle': 'DateTime',
+        'sqlite': 'DateTime',
+        'duckdb': 'DateTime',
+        'citus': 'DateTime',
+        'cockroachdb': 'DateTime',
+        'default': 'DateTime',
+    },
+    'datetime64[us, UTC]': {
+        'timescaledb': 'DateTime(timezone=True)',
+        'timescaledb-ha': 'DateTime(timezone=True)',
+        'postgresql': 'DateTime(timezone=True)',
+        'postgis': 'DateTime(timezone=True)',
+        'mariadb': 'DateTime(timezone=True)',
+        'mysql': 'DateTime(timezone=True)',
+        'mssql': 'sqlalchemy.dialects.mssql.DATETIMEOFFSET',
+        'oracle': 'sqlalchemy.dialects.oracle.TIMESTAMP',
+        'sqlite': 'DateTime(timezone=True)',
+        'duckdb': 'DateTime(timezone=True)',
+        'citus': 'DateTime(timezone=True)',
+        'cockroachdb': 'DateTime(timezone=True)',
+        'default': 'DateTime(timezone=True)',
+    },
+    'datetime64[ms]': {
+        'timescaledb': 'DateTime',
+        'timescaledb-ha': 'DateTime',
+        'postgresql': 'DateTime',
+        'postgis': 'DateTime',
+        'mariadb': 'DateTime',
+        'mysql': 'DateTime',
+        'mssql': 'sqlalchemy.dialects.mssql.DATETIME2',
+        'oracle': 'DateTime',
+        'sqlite': 'DateTime',
+        'duckdb': 'DateTime',
+        'citus': 'DateTime',
+        'cockroachdb': 'DateTime',
+        'default': 'DateTime',
+    },
+    'datetime64[ms, UTC]': {
+        'timescaledb': 'DateTime(timezone=True)',
+        'timescaledb-ha': 'DateTime(timezone=True)',
+        'postgresql': 'DateTime(timezone=True)',
+        'postgis': 'DateTime(timezone=True)',
+        'mariadb': 'DateTime(timezone=True)',
+        'mysql': 'DateTime(timezone=True)',
+        'mssql': 'sqlalchemy.dialects.mssql.DATETIMEOFFSET',
+        'oracle': 'sqlalchemy.dialects.oracle.TIMESTAMP',
+        'sqlite': 'DateTime(timezone=True)',
+        'duckdb': 'DateTime(timezone=True)',
+        'citus': 'DateTime(timezone=True)',
+        'cockroachdb': 'DateTime(timezone=True)',
+        'default': 'DateTime(timezone=True)',
+    },
+    'datetime64[s]': {
+        'timescaledb': 'DateTime',
+        'timescaledb-ha': 'DateTime',
+        'postgresql': 'DateTime',
+        'postgis': 'DateTime',
+        'mariadb': 'DateTime',
+        'mysql': 'DateTime',
+        'mssql': 'sqlalchemy.dialects.mssql.DATETIME2',
+        'oracle': 'DateTime',
+        'sqlite': 'DateTime',
+        'duckdb': 'DateTime',
+        'citus': 'DateTime',
+        'cockroachdb': 'DateTime',
+        'default': 'DateTime',
+    },
+    'datetime64[s, UTC]': {
+        'timescaledb': 'DateTime(timezone=True)',
+        'timescaledb-ha': 'DateTime(timezone=True)',
+        'postgresql': 'DateTime(timezone=True)',
+        'postgis': 'DateTime(timezone=True)',
+        'mariadb': 'DateTime(timezone=True)',
+        'mysql': 'DateTime(timezone=True)',
+        'mssql': 'sqlalchemy.dialects.mssql.DATETIMEOFFSET',
+        'oracle': 'sqlalchemy.dialects.oracle.TIMESTAMP',
+        'sqlite': 'DateTime(timezone=True)',
+        'duckdb': 'DateTime(timezone=True)',
+        'citus': 'DateTime(timezone=True)',
+        'cockroachdb': 'DateTime(timezone=True)',
+        'default': 'DateTime(timezone=True)',
+    },
+    'date': {
+        'timescaledb': 'Date',
+        'timescaledb-ha': 'Date',
+        'postgresql': 'Date',
+        'postgis': 'Date',
+        'mariadb': 'Date',
+        'mysql': 'Date',
+        'mssql': 'Date',
+        'oracle': 'Date',
+        'sqlite': 'Date',
+        'duckdb': 'Date',
+        'citus': 'Date',
+        'cockroachdb': 'Date',
+        'default': 'Date',
+    },
     'bool': {
         'timescaledb': 'Boolean',
+        'timescaledb-ha': 'Boolean',
+        'postgresql': 'Boolean',
+        'postgis': 'Boolean',
+        'mariadb': 'Integer',
+        'mysql': 'Integer',
+        'mssql': 'sqlalchemy.dialects.mssql.BIT',
+        'oracle': 'Integer',
+        'sqlite': 'Float',
+        'duckdb': 'Boolean',
+        'citus': 'Boolean',
+        'cockroachdb': 'Boolean',
+        'default': 'Boolean',
+    },
+    'bool[pyarrow]': {
+        'timescaledb': 'Boolean',
+        'timescaledb-ha': 'Boolean',
+        'postgresql': 'Boolean',
+        'postgis': 'Boolean',
+        'mariadb': 'Integer',
+        'mysql': 'Integer',
+        'mssql': 'sqlalchemy.dialects.mssql.BIT',
+        'oracle': 'Integer',
+        'sqlite': 'Float',
+        'duckdb': 'Boolean',
+        'citus': 'Boolean',
+        'cockroachdb': 'Boolean',
+        'default': 'Boolean',
+    },
+    'boolean': {
+        'timescaledb': 'Boolean',
+        'timescaledb-ha': 'Boolean',
         'postgresql': 'Boolean',
         'postgis': 'Boolean',
         'mariadb': 'Integer',
@@ -624,6 +940,7 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'object': {
         'timescaledb': 'UnicodeText',
+        'timescaledb-ha': 'UnicodeText',
         'postgresql': 'UnicodeText',
         'postgis': 'UnicodeText',
         'mariadb': 'UnicodeText',
@@ -638,6 +955,7 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'string': {
         'timescaledb': 'UnicodeText',
+        'timescaledb-ha': 'UnicodeText',
         'postgresql': 'UnicodeText',
         'postgis': 'UnicodeText',
         'mariadb': 'UnicodeText',
@@ -652,6 +970,7 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'json': {
         'timescaledb': 'sqlalchemy.dialects.postgresql.JSONB',
+        'timescaledb-ha': 'sqlalchemy.dialects.postgresql.JSONB',
         'postgresql': 'sqlalchemy.dialects.postgresql.JSONB',
         'postgis': 'sqlalchemy.dialects.postgresql.JSONB',
         'mariadb': 'UnicodeText',
@@ -666,6 +985,7 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'numeric': {
         'timescaledb': 'Numeric',
+        'timescaledb-ha': 'Numeric',
         'postgresql': 'Numeric',
         'postgis': 'Numeric',
         'mariadb': 'Numeric',
@@ -680,6 +1000,7 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'uuid': {
         'timescaledb': 'Uuid',
+        'timescaledb-ha': 'Uuid',
         'postgresql': 'Uuid',
         'postgis': 'Uuid',
         'mariadb': 'sqlalchemy.dialects.mysql.CHAR(36)',
@@ -692,8 +1013,24 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
         'cockroachdb': 'Uuid',
         'default': 'Uuid',
     },
+    'binary[pyarrow]': {
+        'timescaledb': 'LargeBinary',
+        'timescaledb-ha': 'LargeBinary',
+        'postgresql': 'LargeBinary',
+        'postgis': 'LargeBinary',
+        'mariadb': 'LargeBinary',
+        'mysql': 'LargeBinary',
+        'mssql': 'LargeBinary',
+        'oracle': 'LargeBinary',
+        'sqlite': 'LargeBinary',
+        'duckdb': 'LargeBinary',
+        'citus': 'LargeBinary',
+        'cockroachdb': 'LargeBinary',
+        'default': 'LargeBinary',
+    },
     'bytes': {
         'timescaledb': 'LargeBinary',
+        'timescaledb-ha': 'LargeBinary',
         'postgresql': 'LargeBinary',
         'postgis': 'LargeBinary',
         'mariadb': 'LargeBinary',
@@ -708,6 +1045,7 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'geometry': {
         'timescaledb': 'UnicodeText',
+        'timescaledb-ha': 'geoalchemy2.Geometry',
         'postgresql': 'UnicodeText',
         'postgis': 'geoalchemy2.Geometry',
         'mariadb': 'UnicodeText',
@@ -722,6 +1060,7 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
     },
     'geography': {
         'timescaledb': 'UnicodeText',
+        'timescaledb-ha': 'geoalchemy2.Geography',
         'postgresql': 'UnicodeText',
         'postgis': 'geoalchemy2.Geography',
         'mariadb': 'UnicodeText',
@@ -738,6 +1077,7 @@ PD_TO_SQLALCHEMY_DTYPES_FLAVORS: Dict[str, Dict[str, str]] = {
 
 AUTO_INCREMENT_COLUMN_FLAVORS: Dict[str, str] = {
     'timescaledb': 'GENERATED BY DEFAULT AS IDENTITY',
+    'timescaledb-ha': 'GENERATED BY DEFAULT AS IDENTITY',
     'postgresql': 'GENERATED BY DEFAULT AS IDENTITY',
     'postgis': 'GENERATED BY DEFAULT AS IDENTITY',
     'mariadb': 'AUTO_INCREMENT',