diff options
Diffstat (limited to 'wqflask/secure_server.py')
-rw-r--r-- | wqflask/secure_server.py | 67 |
1 files changed, 67 insertions, 0 deletions
diff --git a/wqflask/secure_server.py b/wqflask/secure_server.py new file mode 100644 index 00000000..5448dbeb --- /dev/null +++ b/wqflask/secure_server.py @@ -0,0 +1,67 @@ +from __future__ import print_function, division, absolute_import + +from wqflask import app + +from flask import Flask, render_template +from flask.ext.sqlalchemy import SQLAlchemy +from flask.ext.security import Security, SQLAlchemyUserDatastore, UserMixin, RoleMixin + + +# Create database connection object +db = SQLAlchemy(app) + +# Define models +roles_users = db.Table('roles_users', + db.Column('user_id', db.Integer(), db.ForeignKey('user.id')), + db.Column('role_id', db.Integer(), db.ForeignKey('role.id'))) + +class Role(db.Model, RoleMixin): + id = db.Column(db.Integer(), primary_key=True) + name = db.Column(db.String(80), unique=True) + description = db.Column(db.String(255)) + +class User(db.Model, UserMixin): + id = db.Column(db.Integer(), primary_key=True) + email = db.Column(db.String(255), unique=True) + password = db.Column(db.String(255)) + active = db.Column(db.Boolean()) + confirmed_at = db.Column(db.DateTime()) + + last_login_at = db.Column(db.DateTime()) + current_login_at = db.Column(db.DateTime()) + last_login_ip_= db.Column(db.String(39)) + current_login_ip = db.Column(db.String(39)) + login_count = db.Column(db.Integer()) + + roles = db.relationship('Role', secondary=roles_users, + backref=db.backref('users', lazy='dynamic')) + +# Setup Flask-Security +user_datastore = SQLAlchemyUserDatastore(db, User, Role) +security = Security(app, user_datastore) + +# Create a user to test with +##@app.before_first_request +#def create_user(): +# db.create_all() +# user_datastore.create_user(email='matt@example.com', password='notebook') +# db.session.commit() + + +import logging +file_handler = logging.FileHandler(app.config['LOGFILE']) +file_handler.setLevel(logging.DEBUG) +app.logger.addHandler(file_handler) + +import logging_tree +logging_tree.printout() + +#print("app.config is:", app.config) + +if __name__ == '__main__': + #create_user() + app.run(host='0.0.0.0', + port=app.config['SERVER_PORT'], + use_debugger=False, + threaded=True, + use_reloader=True) |