wrolpi/modules/map/models.py

from sqlalchemy import Column, Integer, Boolean, BigInteger

from modules.map import lib
from wrolpi import dates
from wrolpi.common import Base, ModelHelper
from wrolpi.media_path import MediaPathType


class MapFile(Base, ModelHelper):
    __tablename__ = 'map_file'
    id = Column(Integer, primary_key=True)

    path = Column(MediaPathType, unique=True, nullable=False)
    imported = Column(Boolean, default=False, nullable=False)
    size = Column(BigInteger)

    def __repr__(self):
        return f'<MapFile {self.path} imported={self.imported}>'

    def __json__(self):
        return {
            'id': self.id,
            'imported': self.imported,
            'path': self.path,
            'seconds_to_import': (seconds := lib.seconds_to_import(self.size)),
            'size': self.size,
            'time_to_import': dates.seconds_to_timestamp(seconds),
        }