"""Worksheet Correction Model."""
from sqlalchemy import Column
from sqlalchemy import Integer
from sqlalchemy import Numeric

from moneyhub.models.mysql_base import BaseModel


class WorksheetCorrection(BaseModel):
    """Worksheet Correction Model."""

    __tablename__ = 'worksheet_correction'

    worksheet_correction_id = Column(Integer, nullable=False, primary_key=True)  # noqa: E501
    account_id = Column(Integer, nullable=False)
    contract_id = Column(Integer, nullable=False)
    statement_period_id = Column(Integer, nullable=False)
    correction_statement_period_id = Column(Integer, nullable=False)
    mechanical_deduction_total = Column(Numeric(20, 2), nullable=True)
    mechanical_deduction_admin_fee_total = Column(Numeric(20, 2), nullable=True)