coryHawkvelt
adfeb1bd0f
Remove UniqueConstraint('id') from all create_table calls in the initial
migration (40 occurrences) and the bulk constraint additions from the
superadmin migration (43 create + 43 drop). These were redundant with
PrimaryKeyConstraint('id') which already guarantees uniqueness.
Also removes duplicate unique enforcement on superadmins.email and
superadmin_sessions.token (kept the unique indexes, dropped the
table-level UniqueConstraints).
Fixes the root cause in BaseModel by removing unique=True from the id
column definition, which was causing Alembic autogenerate to produce
these redundant constraints.
Renames idx_cert_audit_org to ix_certificate_audit_logs_organization_id
to follow Alembic naming conventions.
91 lines
2.6 KiB
Python
91 lines
2.6 KiB
Python
"""Base model with common fields and functionality."""
|
|
import uuid
|
|
from datetime import datetime, timezone
|
|
from gatehouse_app.extensions import db
|
|
|
|
|
|
class BaseModel(db.Model):
|
|
"""Base model class with common fields."""
|
|
|
|
__abstract__ = True
|
|
|
|
id = db.Column(
|
|
db.String(36),
|
|
primary_key=True,
|
|
default=lambda: str(uuid.uuid4()),
|
|
nullable=False,
|
|
)
|
|
created_at = db.Column(db.DateTime, nullable=False, default=lambda: datetime.now(timezone.utc))
|
|
updated_at = db.Column(
|
|
db.DateTime, nullable=False, default=lambda: datetime.now(timezone.utc), onupdate=lambda: datetime.now(timezone.utc)
|
|
)
|
|
deleted_at = db.Column(db.DateTime, nullable=True)
|
|
|
|
@classmethod
|
|
def create(cls, **kwargs):
|
|
"""Create and save a new model instance.
|
|
|
|
Args:
|
|
**kwargs: Model field values
|
|
|
|
Returns:
|
|
The created model instance
|
|
"""
|
|
instance = cls(**kwargs)
|
|
db.session.add(instance)
|
|
db.session.commit()
|
|
return instance
|
|
|
|
def save(self):
|
|
"""Save the model instance to database."""
|
|
db.session.add(self)
|
|
db.session.commit()
|
|
return self
|
|
|
|
def delete(self, soft=True):
|
|
"""
|
|
Delete the model instance.
|
|
|
|
Args:
|
|
soft: If True, performs soft delete. If False, hard delete.
|
|
"""
|
|
if soft:
|
|
self.deleted_at = datetime.now(timezone.utc)
|
|
db.session.commit()
|
|
else:
|
|
db.session.delete(self)
|
|
db.session.commit()
|
|
|
|
def update(self, **kwargs):
|
|
"""Update model fields."""
|
|
for key, value in kwargs.items():
|
|
if hasattr(self, key):
|
|
setattr(self, key, value)
|
|
self.updated_at = datetime.now(timezone.utc)
|
|
db.session.commit()
|
|
return self
|
|
|
|
def to_dict(self, exclude=None):
|
|
"""
|
|
Convert model to dictionary.
|
|
|
|
Args:
|
|
exclude: List of fields to exclude from output
|
|
|
|
Returns:
|
|
Dictionary representation of the model
|
|
"""
|
|
exclude = exclude or []
|
|
result = {}
|
|
for column in self.__table__.columns:
|
|
if column.name not in exclude:
|
|
value = getattr(self, column.name)
|
|
if isinstance(value, datetime):
|
|
if value.tzinfo is None:
|
|
result[column.name] = value.isoformat() + "Z"
|
|
else:
|
|
result[column.name] = value.astimezone(timezone.utc).isoformat().replace("+00:00", "Z")
|
|
else:
|
|
result[column.name] = value
|
|
return result
|