sqlmodel/sqlmodel/sql/sqltypes.py at main · codassassin/sqlmodel · GitHub

Repository navigation

History

60 lines (47 loc) · 1.79 KB

Raw

import uuid

from typing import Any, cast

from sqlalchemy import types

from sqlalchemy.dialects.postgresql import UUID

from sqlalchemy.engine.interfaces import Dialect

from sqlalchemy.types import CHAR, TypeDecorator

class AutoString(types.TypeDecorator):

impl = types.String

cache_ok = True

mysql_default_length = 255

def load_dialect_impl(self, dialect: Dialect) -> "types.TypeEngine[Any]":

impl = cast(types.String, self.impl)

if impl.length is None and dialect.name == "mysql":

return dialect.type_descriptor(types.String(self.mysql_default_length)) # type: ignore

return super().load_dialect_impl(dialect)

# Reference form SQLAlchemy docs: https://docs.sqlalchemy.org/en/14/core/custom_types.html#backend-agnostic-guid-type

# with small modifications

class GUID(TypeDecorator):

"""Platform-independent GUID type.

Uses PostgreSQL's UUID type, otherwise uses

CHAR(32), storing as stringified hex values.

"""

impl = CHAR

cache_ok = True

def load_dialect_impl(self, dialect):

if dialect.name == "postgresql":

return dialect.type_descriptor(UUID())

else:

return dialect.type_descriptor(CHAR(32))

def process_bind_param(self, value, dialect):

if value is None:

return value

elif dialect.name == "postgresql":

return str(value)

else:

if not isinstance(value, uuid.UUID):

return f"{uuid.UUID(value).int:x}"

else:

# hexstring

return f"{value.int:x}"

def process_result_value(self, value, dialect):

if value is None:

return value

else:

if not isinstance(value, uuid.UUID):

value = uuid.UUID(value)

return value

sqlmodel/sqlmodel/sql/sqltypes.py at main · codassassin/sqlmodel · GitHub

Repository navigation

Expand file tree

Breadcrumbs

sqltypes.py

Latest commit

History

Breadcrumbs

sqltypes.py

File metadata and controls