from datetime import datetime

from sqlalchemy import Column, String, DateTime


class OperationTrackMixin:
    update_user_id = Column('update_user_id', String(64), comment='最后更新人id')
    update_user_name = Column('update_user_name', String(64), comment='最后更新人名称')
    update_time = Column('update_time', DateTime, default=datetime.now, comment='最后更新时间')
    create_user_id = Column('create_user_id', String(64), comment='创建人id')
    create_user_name = Column('create_user_name', String(64), comment='创建人名称')
    create_time = Column('create_time', DateTime, default=datetime.now, comment='创建时间')
    delete_user_id = Column('delete_user_id', String(64), comment='删除人id')
    delete_user_name = Column('delete_user_name', String(64), comment='删除人名称')
    delete_time = Column('delete_time', DateTime, comment='删除时间')

    def track_delete(self, user_id, user_name=None):
        self.delete_user_id = user_id
        self.delete_user_name = user_name
        self.delete_time = datetime.now()

    def track_create(self, user_id, user_name=None):
        self.create_user_id = user_id
        self.create_user_name = user_name
        self.create_time = datetime.now()

    def track_update(self, user_id, user_name=None):
        self.update_user_id = user_id
        self.update_user_name = user_name
        self.update_time = datetime.now()