/
UserModel.py
79 lines (64 loc) · 2.15 KB
/
UserModel.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
75
76
77
78
# src/models/UserModel.py
from marshmallow import fields, Schema
import datetime
from . import db, bcrypt
from .BlogpostModel import BlogpostSchema
class UserModel(db.Model):
"""
User Model
"""
# table name
__tablename__ = 'users'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(128), nullable=False)
email = db.Column(db.String(128), unique=True, nullable=False)
password = db.Column(db.String(128), nullable=False)
created_at = db.Column(db.DateTime)
modified_at = db.Column(db.DateTime)
blogposts = db.relationship('BlogpostModel', backref='users', lazy=True)
# class constructor
def __init__(self, data):
"""
Class constructor
"""
self.name = data.get('name')
self.email = data.get('email')
self.password = self.__generate_hash(data.get('password'))
self.created_at = datetime.datetime.utcnow()
self.modified_at = datetime.datetime.utcnow()
def save(self):
db.session.add(self)
db.session.commit()
def update(self, data):
for key, item in data.items():
if key == 'password':
self.password = self.__generate_hash(value)
setattr(self, key, item)
self.modified_at = datetime.datetime.utcnow()
db.session.commit()
def delete(self):
db.session.delete(self)
db.session.commit()
@staticmethod
def get_all_users():
return UserModel.query.all()
@staticmethod
def get_one_user(id):
return UserModel.query.get(id)
@staticmethod
def get_user_by_email(value):
return UserModel.query.filter_by(email=value).first()
def __generate_hash(self, password):
return bcrypt.generate_password_hash(password, rounds=10).decode("utf-8")
def check_hash(self, password):
return bcrypt.check_password_hash(self.password, password)
def __repr(self):
return '<id {}>'.format(self.id)
class UserSchema(Schema):
id = fields.Int(dump_only=True)
name = fields.Str(required=True)
email = fields.Email(required=True)
password = fields.Str(required=True, load_only=True)
created_at = fields.DateTime(dump_only=True)
modified_at = fields.DateTime(dump_only=True)
blogposts = fields.Nested(BlogpostSchema, many=True)