-
Notifications
You must be signed in to change notification settings - Fork 0
/
users.py
74 lines (62 loc) · 2.54 KB
/
users.py
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
from database import DB
from movies import Movies
class Users:
def __init__(self, id, username, mail, password, birthday):
self.id = id
self.username = username
self.mail = mail
self.password = password
self.birthday = birthday
def create(self):
with DB() as db:
values = (self.username, self.mail, self.password, self.birthday)
row = db.execute('INSERT INTO Users (username, mail, password, birthday) VALUES (?, ?, ?, ?)', values)
return self
def delete(self):
with DB() as db:
db.execute('DELETE FROM Users WHERE id = ?', (self.id,))
def userAssign(self, movieId):
with DB() as db:
values = (movieId, self.id)
row = db.execute('INSERT INTO Third (movieId, userId) VALUES(?, ?)', values)
return self
def userAllMovies(self):
with DB() as db:
values = (self.id,)
rows = db.execute('SELECT Movies.id, Movies.title, Movies.description, Movies.directorId, Movies.ageLimit, Movies.date FROM Movies INNER JOIN Third ON Movies.id = Third.movieId WHERE Third.userId = ?'
, values).fetchall()
print(rows)
return [Movies(*row) for row in rows]
def findDirectorMovies(self):
with DB() as db:
values = (self.id,)
rows = db.execute('SELECT * FROM Movies WHERE directorId = ?', values).fetchall()
return [Movies(*row) for row in rows]
@staticmethod
def loadUserId(Id):
with DB() as db:
values = (Id,)
row = db.execute('SELECT * FROM Users WHERE id = ?', values).fetchone()
if not row:
return None
return Users(*row)
@staticmethod
def loadUserMail(mail):
with DB() as db:
values = (mail,)
row = db.execute('SELECT * FROM Users WHERE mail = ?', values).fetchone()
if not row:
return None
return Users(*row)
@staticmethod
def loadUserMailAndPass(mail, password):
with DB() as db:
values = (mail, password,)
row = db.execute('SELECT * FROM Users WHERE mail = ? AND password = ?', values).fetchone()
if not row:
return None
return Users(*row)
def userUnassign(self, movieId):
with DB() as db:
values = (movieId, self.id)
row = db.execute('DELETE FROM Third WHERE movieId = ? AND userId = ?', values)