1from src import db 

2from .content_model import ContentType 

3 

4 

5class LinkedGenreModel(db.Model): 

6 """ 

7 LinkedGenreModel Model for storing similars track 

8 """ 

9 __tablename__ = "linked_genre" 

10 

11 genre_id0 = db.Column(db.Integer, db.ForeignKey( 

12 "genre.genre_id"), primary_key=True) 

13 genre_id1 = db.Column(db.Integer, db.ForeignKey( 

14 "genre.genre_id"), primary_key=True) 

15 ratio = db.Column(db.Float) 

16 

17 

18class GenreModel(db.Model): 

19 """ 

20 Genre Model for storing group related details 

21 """ 

22 __tablename__ = "genre" 

23 

24 genre_id = db.Column(db.Integer, primary_key=True, autoincrement=True) 

25 name = db.Column(db.String(45), nullable=False) 

26 count = db.Column(db.Integer) 

27 content_type = db.Column(db.Enum(ContentType)) 

28 

29 # Loaded immediately after loading Track, but when querying multiple tracks, you will not get additional queries. 

30 linked_genres = db.relationship("LinkedGenreModel", secondary=LinkedGenreModel.__table__, 

31 primaryjoin=genre_id == LinkedGenreModel.genre_id0, 

32 secondaryjoin=genre_id == LinkedGenreModel.genre_id1, 

33 lazy="dynamic")