from login import db, app
from flask_security import Security, SQLAlchemyUserDatastore, login_required, UserMixin, RoleMixin


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 User(db.Model, UserMixin):
    id = db.Column(db.Integer, primary_key=True)
    email = db.Column(db.String, unique=True)
    username = db.Column(db.String(25), unique=True)
    password = db.Column(db.String(30))
    active = db.Column(db.Boolean)
    confirmed_at = db.Column(db.DateTime())
    last_login_at = db.Column(db.String)
    current_login_at = db.Column(db.String)
    last_login_ip = db.Column(db.String)
    current_login_ip = db.Column(db.String)
    login_count = db.Column(db.Integer)
    roles = db.relationship('Role', secondary=roles_users,
                            backref=db.backref('users', lazy='dynamic'))


class Role(db.Model, RoleMixin):
    id = db.Column(db.Integer(), primary_key=True)
    name = db.Column(db.String(80), unique=True)
    desc = db.Column(db.String(255))

# setting up flask security
user_datastore = SQLAlchemyUserDatastore(db, User, Role)