OSGA/models.py at master · Miloceane/OSGA · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
################################
# OSGA - models.py  	       #
# Written by Charlotte Lafage  #
# (GitHub: Miloceane)          #
# For Minor Programmeren       #
# (Universiteit van Amsterdam) #
################################

from flask_sqlalchemy import SQLAlchemy
from sqlalchemy import Table, Column, Integer, ForeignKey
from sqlalchemy.orm import relationship
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.sql import func

db = SQLAlchemy()

class Shows(db.Model):
	__tablename__ = 'shows'
	id = db.Column(db.Integer, primary_key=True)
	name = db.Column(db.String(128))
	universe_id = db.Column(db.Integer, db.ForeignKey('shows_universes.id'))
	api_id = db.Column(db.String(128))
	is_series = db.Column(db.Boolean, default=True)
	universe = relationship("Universes", back_populates="shows")


class Universes(db.Model):
	__tablename__ = 'shows_universes'
	id = db.Column(db.Integer, primary_key=True)
	name = db.Column(db.String(128))
	shows = relationship(Shows)


class FavouritedShows(db.Model):
	__tablename__ = 'shows_favourited'
	id = db.Column(db.Integer, primary_key=True)
	user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
	show_id = db.Column(db.Integer, db.ForeignKey('shows.id'))


class BlacklistedShows(db.Model):
	__tablename__ = 'shows_blacklisted'
	id = db.Column(db.Integer, primary_key=True)
	user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
	show_id = db.Column(db.Integer, db.ForeignKey('shows.id'))

class CharactersMessages(db.Model):
	__tablename__ = 'characters_messages'
	id = db.Column(db.Integer, primary_key=True)
	date = db.Column(db.DateTime(timezone=True), server_default=func.now())
	content = db.Column(db.String(128))
	user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
	character_id = db.Column(db.Integer, db.ForeignKey('characters.id'))
	admin_id = db.Column(db.Integer, default=0) # Admin who validated the message
	pos_x =  db.Column(db.Integer, default=0)
	pos_y =  db.Column(db.Integer, default=0)
	character = relationship("Characters", back_populates="messages")
	user = relationship("Users", back_populates="messages")

class CharactersFlowers(db.Model):
	__tablename__ = 'characters_flowers'
	id = db.Column(db.Integer, primary_key=True, autoincrement=True)
	date = db.Column(db.DateTime(timezone=True), server_default=func.now())
	flowertype_id = db.Column(db.Integer, default=0)# , db.ForeignKey('flower_types.id'))  -> For some reason, complained about there being no constraints?
	user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
	character_id = db.Column(db.Integer, db.ForeignKey('characters.id'))
	pos_x =  db.Column(db.Integer)
	pos_y =  db.Column(db.Integer)
	character = relationship("Characters", back_populates="flowers")
	user =	relationship("Users", back_populates="flowers")

class Characters(db.Model):
	__tablename__ = 'characters'
	id = db.Column(db.Integer, primary_key=True)
	name = db.Column(db.String(128))
	universe_id = db.Column(db.Integer, db.ForeignKey('shows_universes.id'))
	show_id = db.Column(db.Integer, db.ForeignKey('shows.id'))
	death_date = db.Column(db.DateTime(timezone=True), server_default=func.now())
	death_season = db.Column(db.Integer)
	death_episode = db.Column(db.Integer)
	admin_id = db.Column(db.Integer, db.ForeignKey('users.id')) # Admin who declared death
	flower_count = db.Column(db.Integer, default=0)
	flowers = relationship(CharactersFlowers)
	messages = relationship(CharactersMessages)


class Users(db.Model):
	__tablename__ = 'users'
	id = db.Column(db.Integer, primary_key=True)
	name = db.Column(db.String(128))
	password = db.Column(db.String(128))
	password_salt = db.Column(db.String(128))
	email = db.Column(db.String(128))
	admin_level = db.Column(db.Integer, default=0)
	registration_date = db.Column(db.DateTime(timezone=True), server_default=func.now())
	display_fav = db.Column(db.Boolean, default=False)
	display_activity = db.Column(db.Boolean, default=False)
	flowers_left = db.Column(db.Integer, default=5)
	blocked = db.Column(db.Boolean, default=False)
	activated = db.Column(db.Boolean, default=False)
	activation_code = db.Column(db.String(128))
	activation_timelimit = db.Column(db.DateTime(timezone=True), server_default=func.now())
	remembered = db.Column(db.Boolean, default=False)

	flowers = relationship(CharactersFlowers)
	messages = relationship(CharactersMessages)

	def is_authenticated(self):
		return True

	def is_active(self):
		return self.activated

	def is_anonymous(self):
		return False

	def get_id(self):
		return(self.id)


class Suggestions(db.Model):
	__tablename__ = 'suggestions'
	id = db.Column(db.Integer, primary_key=True)
	user_id = db.Column(db.Integer)
	date = db.Column(db.DateTime(timezone=True), server_default=func.now())
	show = db.Column(db.String(128))
	content = db.Column(db.String(512))
	is_read = db.Column(db.Boolean, default=False)
	is_important = db.Column(db.Boolean, default=False)