You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
75 lines
2.2 KiB
Python
75 lines
2.2 KiB
Python
import click
|
|
|
|
from sqlalchemy import create_engine, Column, Integer, String, ForeignKey
|
|
from sqlalchemy.orm import relationship, sessionmaker, Mapped, backref
|
|
|
|
from .show import Base, Show
|
|
from .track import Track
|
|
|
|
from ffx.media_descriptor import MediaDescriptor
|
|
from ffx.show_descriptor import ShowDescriptor
|
|
|
|
class Pattern(Base):
|
|
|
|
__tablename__ = 'patterns'
|
|
|
|
# v1.x
|
|
id = Column(Integer, primary_key=True)
|
|
pattern = Column(String)
|
|
|
|
# v2.0
|
|
# id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
|
# pattern: Mapped[str] = mapped_column(String, nullable=False)
|
|
|
|
# v1.x
|
|
show_id = Column(Integer, ForeignKey('shows.id', ondelete="CASCADE"))
|
|
show = relationship(Show, back_populates='patterns', lazy='joined')
|
|
|
|
# v2.0
|
|
# show_id: Mapped[int] = mapped_column(ForeignKey("shows.id", ondelete="CASCADE"))
|
|
# show: Mapped["Show"] = relationship(back_populates="patterns")
|
|
|
|
tracks = relationship('Track', back_populates='pattern', cascade="all, delete", lazy='joined')
|
|
|
|
|
|
media_tags = relationship('MediaTag', back_populates='pattern', cascade="all, delete", lazy='joined')
|
|
|
|
|
|
def getId(self):
|
|
return int(self.id)
|
|
|
|
def getShowId(self):
|
|
return int(self.show_id)
|
|
|
|
def getShowDescriptor(self) -> ShowDescriptor:
|
|
click.echo(f"self.show {self.show} id={self.show_id}")
|
|
return self.show.getDescriptor()
|
|
|
|
def getId(self):
|
|
return int(self.id)
|
|
|
|
def getPattern(self):
|
|
return str(self.pattern)
|
|
|
|
def getTags(self):
|
|
return {str(t.key):str(t.value) for t in self.media_tags}
|
|
|
|
|
|
def getMediaDescriptor(self):
|
|
|
|
kwargs = {}
|
|
kwargs[MediaDescriptor.TAGS_KEY] = self.getTags()
|
|
|
|
kwargs[MediaDescriptor.TRACK_DESCRIPTOR_LIST_KEY] = []
|
|
|
|
# Set ordered subindices
|
|
subIndexCounter = {}
|
|
for track in self.tracks:
|
|
trackType = track.getType()
|
|
if not trackType in subIndexCounter.keys():
|
|
subIndexCounter[trackType] = 0
|
|
kwargs[MediaDescriptor.TRACK_DESCRIPTOR_LIST_KEY].append(track.getDescriptor(subIndex = subIndexCounter[trackType]))
|
|
subIndexCounter[trackType] += 1
|
|
|
|
return MediaDescriptor(**kwargs)
|