models.py
3.26 KB
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
from flask_admin.contrib.sqla import ModelView
from flaskr.core import generate_unique_id
from flask_sqlalchemy import SQLAlchemy
from flask_login import UserMixin, AnonymousUserMixin
from werkzeug.security import generate_password_hash, check_password_hash
from yaml import safe_load as yaml_load
import enum
# These are not the emission "models" in the scientific meaning of the word.
# They are the SQL Database Models.
# These are also named Entities, in other conventions (we're following flasks")
# If you're looking for the Emission Models (aka scaling laws),
# look in `flaskr/laws/`.
db = SQLAlchemy()
class StatusEnum(enum.Enum):
pending = 'pending'
working = 'working'
success = 'success'
failure = 'failure'
class Estimation(db.Model):
id = db.Column(db.Integer(), primary_key=True)
public_id = db.Column(
db.Unicode(),
default=lambda: generate_unique_id(),
unique=True
)
email = db.Column(db.Unicode(1024))
first_name = db.Column(db.Unicode(1024)) # Antoine
last_name = db.Column(db.Unicode(1024)) # Goutenoir
institution = db.Column(db.Unicode(1024)) # IRAP
status = db.Column(db.Enum(StatusEnum), default=StatusEnum.pending)
# City, Country
# One address per line
origin_addresses = db.Column(db.UnicodeText())
destination_addresses = db.Column(db.UnicodeText())
compute_optimal_destination = db.Column(db.Boolean())
output_yaml = db.Column(db.UnicodeText())
warnings = db.Column(db.UnicodeText())
errors = db.Column(db.UnicodeText())
def has_failed(self):
return self.status == StatusEnum.failure
def get_output_dict(self):
return yaml_load(self.output_yaml)
def has_many_to_many(self):
return 'cities' in self.get_output_dict()
class EstimationView(ModelView):
# Show only name and email columns in list view
column_list = (
'public_id',
'status',
'first_name',
'last_name',
'origin_addresses',
'destination_addresses',
'warnings',
'errors',
)
# Enable search functionality - it will search for terms in
# name and email fields
# column_searchable_list = ('name', 'email')
# Add filters for name and email columns
column_filters = ('first_name', 'last_name')
# USERS #######################################################################
class User(db.Model, UserMixin):
id = db.Column(db.Integer(), primary_key=True)
username = db.Column(db.String())
password = db.Column(db.String())
def __init__(self, username, password):
self.username = username
self.set_password(password)
def set_password(self, password):
self.password = generate_password_hash(password)
def check_password(self, value):
return check_password_hash(self.password, value)
@property
def is_authenticated(self):
if isinstance(self, AnonymousUserMixin):
return False
else:
return True
def is_active(self):
return True
def is_anonymous(self):
if isinstance(self, AnonymousUserMixin):
return True
else:
return False
def get_id(self):
return self.id
def __repr__(self):
return '<User %r>' % self.username