Coverage for otaku_info/db/MediaList.py: 86%

1"""LICENSE 

2Copyright 2020 Hermann Krumrey <hermann@krumreyh.com> 

3 

4This file is part of otaku-info. 

5 

6otaku-info is free software: you can redistribute it and/or modify 

7it under the terms of the GNU General Public License as published by 

8the Free Software Foundation, either version 3 of the License, or 

9(at your option) any later version. 

10 

11otaku-info is distributed in the hope that it will be useful, 

12but WITHOUT ANY WARRANTY; without even the implied warranty of 

13MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

14GNU General Public License for more details. 

15 

16You should have received a copy of the GNU General Public License 

17along with otaku-info. If not, see <http://www.gnu.org/licenses/>. 

18LICENSE""" 

19 

20from typing import List, TYPE_CHECKING 

21from jerrycan.base import db 

22from jerrycan.db.User import User 

23from jerrycan.db.ModelMixin import ModelMixin 

24from otaku_info.enums import ListService, MediaType 

25if TYPE_CHECKING: 25 ↛ 26line 25 didn't jump to line 26, because the condition on line 25 was never true

26 from otaku_info.db.MediaListItem import MediaListItem 

27 

28 

29class MediaList(ModelMixin, db.Model): 

30 """ 

31 Database model for user-specific media lists. 

32 """ 

33 

34 def __init__(self, *args, **kwargs): 

35 """ 

36 Initializes the Model 

37 :param args: The constructor arguments 

38 :param kwargs: The constructor keyword arguments 

39 """ 

40 super().__init__(*args, **kwargs) 

41 

42 __tablename__ = "media_lists" 

43 

44 user_id: int = db.Column( 

45 db.Integer, 

46 db.ForeignKey( 

47 "users.id", ondelete="CASCADE", onupdate="CASCADE" 

48 ), 

49 primary_key=True 

50 ) 

51 name: str = db.Column(db.Unicode(255), primary_key=True) 

52 service: ListService = db.Column(db.Enum(ListService), primary_key=True) 

53 media_type: MediaType = db.Column(db.Enum(MediaType), primary_key=True) 

54 

55 user: User = db.relationship( 

56 "User", 

57 backref=db.backref("media_lists", lazy=True, cascade="all, delete") 

58 ) 

59 list_items: List["MediaListItem"] = db.relationship( 

60 "MediaListItem", back_populates="media_list", cascade="all, delete" 

61 )