-
Notifications
You must be signed in to change notification settings - Fork 0
/
models.py
72 lines (57 loc) · 2.27 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
from flask_marshmallow import Marshmallow
import datetime
from utils import *
from config import *
class User(UserMixin, db.Model):
id = db.Column(db.Integer, primary_key=True)
email = db.Column(db.String(100), unique=True)
password = db.Column(db.String(100))
name = db.Column(db.String(100))
identification = db.Column(db.String(100))
address = db.Column(db.String(100))
contact = db.Column(db.String(100))
services = db.Column(db.String(100))
is_admin = db.Column(db.Boolean, nullable=False, default=False)
registered_on = db.Column(db.DateTime, nullable=False)
confirmed = db.Column(db.Boolean, nullable=False, default=False)
confirmed_on = db.Column(db.DateTime, nullable=True)
token_confirmations = db.Column(db.String(4))
observation = db.relationship('Observation', backref='user')
def __init__(self, email, password, name, confirmed, observation=None,
paid=False, is_admin=False, confirmed_on=None, token_confirmations=random_token()):
self.email = email
self.password = password
self.name = name
self.registered_on = datetime.datetime.now()
self.is_admin = is_admin
self.confirmed = confirmed
self.confirmed_on = confirmed_on
self.token_confirmations = token_confirmations
class UserSchema(ma.Schema):
class Meta:
fields = ( 'id',
'email',
'name',
'is_admin'
'identification',
'address',
'contact',
'services'
)
user_schema = UserSchema()
users_schema = UserSchema(many=True)
class Observation(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(70), unique=True)
description = db.Column(db.String(100))
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
def __init__(self, title, description, user_id):
self.title = title
self.description = description
self.user_id = user_id
db.create_all()
class ObservationSchema(ma.Schema):
class Meta:
fields = ('id', 'title', 'description', 'user_id', 'name')
observation_schema = ObservationSchema()
observations_schema = ObservationSchema(many=True)