45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
from sqlalchemy import Column, Integer, String, DateTime, Float, ForeignKey
|
|
from sqlalchemy.orm import relationship
|
|
from .database import Base
|
|
import datetime
|
|
|
|
|
|
class DuplicateReport(Base):
|
|
__tablename__ = "duplicate_reports"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
created_at = Column(DateTime, default=datetime.datetime.utcnow)
|
|
status = Column(String)
|
|
groups = relationship("DuplicateFileGroup", back_populates="report")
|
|
|
|
|
|
class VideoFile(Base):
|
|
__tablename__ = "video_files"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
filepath = Column(String, unique=True, index=True)
|
|
filename = Column(String)
|
|
size = Column(Integer)
|
|
duration = Column(Float)
|
|
phash = Column(String)
|
|
|
|
|
|
class DuplicateFileGroup(Base):
|
|
__tablename__ = "duplicate_file_groups"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
report_id = Column(Integer, ForeignKey("duplicate_reports.id"))
|
|
score = Column(Float)
|
|
report = relationship("DuplicateReport", back_populates="groups")
|
|
files = relationship("DuplicateFile", back_populates="group")
|
|
|
|
|
|
class DuplicateFile(Base):
|
|
__tablename__ = "duplicate_files"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
group_id = Column(Integer, ForeignKey("duplicate_file_groups.id"))
|
|
video_file_id = Column(Integer, ForeignKey("video_files.id"))
|
|
group = relationship("DuplicateFileGroup", back_populates="files")
|
|
video_file = relationship("VideoFile")
|