Coverage for bundesliga_tippspiel/db/SeasonEvent.py: 100%

1"""LICENSE 

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

3 

4This file is part of bundesliga-tippspiel. 

5 

6bundesliga-tippspiel 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 

11bundesliga-tippspiel 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 bundesliga-tippspiel. If not, see <http://www.gnu.org/licenses/>. 

18LICENSE""" 

19 

20from enum import Enum 

21from jerrycan.base import db 

22from jerrycan.db.ModelMixin import ModelMixin 

23 

24 

25class SeasonEventType(Enum): 

26 """ 

27 Enumeration that describes all the possible season event types 

28 """ 

29 PRE_SEASON_MAIL = "pre_season_mail" 

30 MID_SEASON_REMINDER = "mid_season_reminder" 

31 POST_SEASON_WRAPUP = "post_season_wrapup" 

32 

33 

34class SeasonEvent(ModelMixin, db.Model): 

35 """ 

36 Model that describes the 'season_events' SQL table 

37 """ 

38 

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

40 """ 

41 Initializes the Model 

42 :param args: The constructor arguments 

43 :param kwargs: The constructor keyword arguments 

44 """ 

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

46 

47 __tablename__ = "season_events" 

48 

49 season: int = db.Column(db.Integer, primary_key=True) 

50 league: int = db.Column(db.String(255), primary_key=True) 

51 event_type: SeasonEventType = db.Column( 

52 db.Enum(SeasonEventType), 

53 primary_key=True 

54 ) 

55 

56 executed: bool = db.Column(db.Boolean, nullable=False, default=False)