mirror of
https://github.com/beak-insights/felicity-lims.git
synced 2025-02-24 17:02:55 +08:00
53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
from datetime import datetime
|
|
from typing import Optional
|
|
from pydantic import BaseModel
|
|
|
|
from sqlalchemy.ext.declarative import declared_attr
|
|
from sqlalchemy import Column, Integer, ForeignKey
|
|
from sqlalchemy.orm import relationship
|
|
|
|
from felicity.database.base_class import DBModel
|
|
from felicity.apps.user.models import User
|
|
from felicity.apps.user.schemas import User as UserSchema
|
|
from felicity.apps.audit.mixin import AuditableMixin
|
|
|
|
|
|
class TrailMixin(object):
|
|
|
|
@declared_attr
|
|
def created_by_uid(self):
|
|
return Column(Integer, ForeignKey('user.uid'), nullable=True)
|
|
|
|
@declared_attr
|
|
def created_by(self):
|
|
return relationship(User, foreign_keys=[self.created_by_uid])
|
|
|
|
@declared_attr
|
|
def updated_by_uid(self):
|
|
return Column(Integer, ForeignKey('user.uid'), nullable=True)
|
|
|
|
@declared_attr
|
|
def updated_by(self):
|
|
return relationship(User, foreign_keys=[self.updated_by_uid])
|
|
|
|
|
|
class BaseAuditDBModel(DBModel, TrailMixin):
|
|
"""With creator and updater"""
|
|
__abstract__ = True
|
|
|
|
|
|
class Auditable(BaseAuditDBModel, AuditableMixin):
|
|
"""With Audit Log"""
|
|
__abstract__ = True
|
|
|
|
|
|
# Pydantic
|
|
class BaseAuditModel(BaseModel):
|
|
created_at: Optional[datetime] = None
|
|
created_by_uid: Optional[str] = None
|
|
created_by: Optional[UserSchema] = None
|
|
updated_at: Optional[datetime] = None
|
|
updated_by_uid: Optional[str] = None
|
|
updated_by: Optional[UserSchema] = None
|
|
|
|
|