from sqlalchemy import insert
from datetime import date
from dbapi.repositories.utils import DatabaseInterfasesMixin
from dbapi.tables import training


class TrainingRepository(DatabaseInterfasesMixin):
    """Training table repository"""

    def create_training(self, date: date) -> int:
        """Method for creating new instance of training
        Args:
            date: date of a training

        Returns:
            Primary key of created training entry
        """
        new_instance_statement = insert(training).values(Date=date)
        with self.engine.connect() as conn:
            result = conn.execute(new_instance_statement)
            inerted_entry_pk: int = result.inserted_primary_key[0]
            conn.commit()
        return inerted_entry_pk