generated from uwidcit/flask-starter
-
Notifications
You must be signed in to change notification settings - Fork 1
/
models.py
83 lines (70 loc) · 3.22 KB
/
models.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
79
80
81
82
83
from flask_sqlalchemy import SQLAlchemy
from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, SubmitField
from wtforms.validators import InputRequired, EqualTo, Email
from wtforms.fields.html5 import EmailField
from werkzeug.security import generate_password_hash, check_password_hash
from flask_login import UserMixin
db = SQLAlchemy()
import datetime
class User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(80), unique=True, nullable=False)
email = db.Column(db.String(120), unique=True, nullable=False)
password = db.Column(db.String(120), nullable=False)
routines = db.relationship('Routines', backref='user', lazy=True)
def toDict(self):
return {
'id': self.id,
'username': self.username,
'email': self.email,
'password': self.password
}
def set_password(self, password):
self.password = generate_password_hash(password, method='sha256')
def check_password(self, password):
return check_password_hash(self.password, password)
class Exercise(db.Model):
id= db.Column(db.Integer, primary_key=True)
name= db.Column(db.String(100))
difficulty= db.Column(db.String(20))
description= db.Column(db.String(5000))
equipment_needed= db.Column(db.String(20))
equipment= db.Column(db.String(100))
primary_muscle= db.Column(db.String(100))
secondary_muscle= db.Column(db.String(100))
routines = db.relationship('Routines', backref='exercise', lazy=True)
def toDict(self):
return {
'id': self.id,
'name': self.name,
'difficulty': self.difficulty,
'description': self.description,
'equipment_needed': self.equipment_needed,
'equipment': self.equipment,
'primary_muscle': self.primary_muscle,
'secondary_muscle': self.secondary_muscle,
'routines': self.routines
}
class Routines(db.Model):
id= db.Column(db.Integer, primary_key=True)
name= db.Column(db.String(100), unique=True ,nullable=False) #name of routine
exerciseId= db.Column(db.Integer, db.ForeignKey('exercise.id'), nullable=False) #exercises saved in routine
userId= db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False) #keep track of user that created routine
def toDict(self):
return {
'id': self.id,
'name': self.name,
'exerciseId': self.exerciseId,
'userId': self.userId
}
class SignUp(FlaskForm):
username = StringField('username', validators=[InputRequired()])
email = EmailField('email', validators=[Email(), InputRequired()])
password = PasswordField('New Password', validators=[InputRequired(), EqualTo('confirm', message='Passwords must match')])
confirm = PasswordField('Repeat Password')
submit = SubmitField('Sign Up', render_kw={'class': 'btn waves-effect waves-light white-text'})
class LogIn(FlaskForm):
username = StringField('username', validators=[InputRequired()])
password = PasswordField('New Password', validators=[InputRequired()])
submit = SubmitField('Login', render_kw={'class': 'btn waves-effect waves-light white-text'})