from datetime import datetime

from app.main.data_management.data_base import *


class PV5MissionDay(Schema.base):
    __tablename__ = 'pv5_mission_day'

    id = Column(INTEGER(11), primary_key=True, nullable=False)
    score1 = Column(INTEGER(11))
    score2 = Column(INTEGER(11))
    stars = Column(INTEGER(11))
    status = Column(INTEGER(11))
    start_date = Column(DateTime, nullable=True)
    finish_date = Column(DateTime, nullable=True)
    mission_id = Column(ForeignKey('pv5_mission.id'), nullable=False, index=True)
    day_id = Column(ForeignKey('pv5_day.id'), nullable=False, index=True)

    mission = relationship('PV5Mission')
    day = relationship('PV5Day')

    def __init__(self, score1, score2, stars, status, mission_id, day_id):
        self.score1 = score1
        self.score2 = score2
        self.stars = stars
        self.status = status
        self.mission_id = mission_id
        self.day_id = day_id
        self.start_date = None
        self.finish_date = None

    def set_start_date(self):
        self.start_date = datetime.now()

    def set_finish_date(self):
        self.finish_date = datetime.now()