from sqlalchemy import types
from sqlalchemy.engine import Engine, Inspector
from sqlalchemy.sql.sqltypes import STRINGTYPE

from sqlalchemy_dremio import flight

# SqlAlchemy < 2.0 doesn't have a DOUBLE type, but using Float here would be misleading and can be dangerous for the openmetadata users
class DOUBLE(types.Float):
    __visit_name__ = "DOUBLE"


# monkey patching the sql types of sqlalchemy_dremio package
flight._type_map.update({
    'double': DOUBLE,
    'DOUBLE': DOUBLE,
    'CHARACTER VARYING': STRINGTYPE,
    'BINARY VARYING': types.LargeBinary,
})