mirror of
https://github.com/mealie-recipes/mealie.git
synced 2025-08-04 13:05:21 +02:00
7602c67449
Some checks are pending
Docker Nightly Production / Backend Server Tests (push) Waiting to run
Docker Nightly Production / Frontend and End-to-End Tests (push) Waiting to run
Docker Nightly Production / Build Tagged Release (push) Blocked by required conditions
Docker Nightly Production / Notify Discord (push) Blocked by required conditions
* refactor normalized search migration to use dummy default * changed group slug migration to use raw SQL * updated comment * added tests with anonymized backups (currently failing) * typo * fixed LDAP enum in test data * fix for adding label settings across groups * add migration data fixes * fix shopping list label settings test * re-run db init instead of just running alembic migration, to include fixes * intentionally broke SQLAlchemy GUID handling * safely convert between GUID types in different databases * restore original test data after testing backup restores * added missing group name update to migration
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
import uuid
|
|
from typing import Any
|
|
|
|
from sqlalchemy import Dialect
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
from sqlalchemy.types import CHAR, TypeDecorator
|
|
|
|
|
|
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
|
|
|
|
@staticmethod
|
|
def generate():
|
|
return uuid.uuid4()
|
|
|
|
@staticmethod
|
|
def convert_value_to_guid(value: Any, dialect: Dialect) -> str | None:
|
|
if value is None:
|
|
return value
|
|
elif dialect.name == "postgresql":
|
|
return str(value)
|
|
else:
|
|
if not isinstance(value, uuid.UUID):
|
|
return "%.32x" % uuid.UUID(value).int
|
|
else:
|
|
# hexstring
|
|
return "%.32x" % value.int
|
|
|
|
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):
|
|
return self.convert_value_to_guid(value, dialect)
|
|
|
|
def _uuid_value(self, value):
|
|
if value is None:
|
|
return value
|
|
else:
|
|
if not isinstance(value, uuid.UUID):
|
|
value = uuid.UUID(value)
|
|
return value
|
|
|
|
def process_result_value(self, value, dialect):
|
|
return self._uuid_value(value)
|
|
|
|
def sort_key_function(self, value):
|
|
return self._uuid_value(value)
|