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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
from __future__ import print_function, division, absolute_import
import uuid
import datetime
import simplejson as json
from flask import request
from flask.ext.sqlalchemy import SQLAlchemy
#from flask.ext.security import Security, SQLAlchemyUserDatastore, UserMixin, RoleMixin
#from flask_security.forms import TextField
#from flask_security.forms import RegisterForm
from wqflask import app
from sqlalchemy import Column, Integer, String, Table, ForeignKey, Unicode, Boolean, DateTime, Text
from sqlalchemy.orm import relationship, backref
from wqflask.database import Base, init_db
# Create database connection object
#db = SQLAlchemy(app)
# Is this right? -Sam
#from sqlalchemy.ext.declarative import declarative_base
#Base = declarative_base()
#@classmethod
#def get(cls, key):
# """Convenience get method using the primary key
#
# If record doesn't exist, returns None
#
# Allows the following: User.get('121')
#
# """
# print("in get cls is:", cls)
# print(" key is {} : {}".format(type(key), key))
# query = Model.query(cls)
# print("query is: ", query)
# record = query.get(key)
# return record
#
#
#print("Model is:", vars(Model))
#Model.get = get
# Define models
#roles_users = Table('roles_users',
# Column('user_id', Integer(), ForeignKey('user.the_id')),
# Column('role_id', Integer(), ForeignKey('role.the_id')))
class Role(Base):
__tablename__ = "role"
id = Column(Unicode(36), primary_key=True, default=lambda: unicode(uuid.uuid4()))
name = Column(Unicode(80), unique=True, nullable=False)
description = Column(Unicode(255))
class User(Base):
__tablename__ = "user"
id = Column(Unicode(36), primary_key=True, default=lambda: unicode(uuid.uuid4()))
email_address = Column(Unicode(50), unique=True, nullable=False)
# Todo: Turn on strict mode for Mysql
password = Column(Text, nullable=False)
full_name = Column(Unicode(50))
organization = Column(Unicode(50))
active = Column(Boolean(), nullable=False, default=True)
registration_info = Column(Text) # json detailing when they were registered, etc.
confirmed = Column(Text) # json detailing when they confirmed, etc.
logins = relationship("Login",
order_by="desc(Login.timestamp)",
lazy='dynamic' # Necessary for filter in login_count
)
@property
def login_count(self):
return self.logins.filter_by(successful=True).count()
#return self.query.filter
#return len(self.logins)
#return 8
#return len(self.logins.query.filter(User.logins.has(successful=True)))
@property
def confirmed_at(self):
if self.confirmed:
confirmed_info = json.loads(self.confirmed)
return confirmed_info['timestamp']
else:
return None
@property
def most_recent_login(self):
try:
return self.logins[0]
except IndexError:
return None
#last_login_at = Column(DateTime())
#current_login_at = Column(DateTime())
#last_login_ip = Column(Unicode(39))
#current_login_ip = Column(Unicode(39))
#login_count = Column(Integer())
#roles = relationship('Role', secondary=roles_users,
# backref=backref('users', lazy='dynamic'))
class Login(Base):
__tablename__ = "login"
id = Column(Unicode(36), primary_key=True, default=lambda: unicode(uuid.uuid4()))
user = Column(Unicode(36), ForeignKey('user.id'))
timestamp = Column(DateTime(), default=lambda: datetime.datetime.utcnow())
ip_address = Column(Unicode(39))
successful = Column(Boolean(), nullable=False) # False if wrong password was entered
session_id = Column(Text) # Set only if successfully logged in, otherwise should be blank
def __init__(self, user):
self.user = user.id
self.ip_address = request.remote_addr
# Setup Flask-Security
#user_datastore = SQLAlchemyUserDatastore(db, User, Role)
#class ExtendedRegisterForm(RegisterForm):
# name = TextField('name')
# #print("name is:", name['_name'], vars(name))
# organization = TextField('organization')
#
#security = Security(app, user_datastore, register_form=ExtendedRegisterForm)
#user_datastore.create_role(name="Genentech", description="Genentech Beta Project(testing)")
|
