from app import db

ROLE_USER = 0
ROLE_ADMIN = 1

class User(db.Model):
    id = db.Column(db.Integer, primary_key = True)
    username = db.Column(db.String(64), index = True, unique = True)
    email = db.Column(db.String(120), index = True, unique = True)
    password = db.Column(db.String(120), index = True)
    role = db.Column(db.SmallInteger, default=ROLE_USER)

    def is_authenticated():
        return True

    def is_active():
        return True

    def is_anonymous():
        return False

    def get_id(self):
        return str(self.id)

    def __init__(self, id, username, email, password, role):
        self.id = id
        self.username = username
        self.email = email
        self.password = password
        self.role = role

    def __repr__(self):
        return '<User %r>' % (self.username)

class LoginForm(Form):
    email = TextField('Email Address', [validators.Length(min=6, max=35)])
    password = PasswordField('Your Password', [
        validators.Required(),
    ])
    remember_me = BooleanField('Remember me', default=False)

from flask import Flask, request, render_template, flash, redirect, url_for, session, g
from sqlalchemy import Table, Column, Integer, String, MetaData, ForeignKey, create_engine
from sqlalchemy.orm import mapper,sessionmaker
import os
from flask_sqlalchemy import SQLAlchemy
from flask_login import LoginManager
from config import basedir
from flask_wtf import Form
from wtforms import TextField, BooleanField, PasswordField, TextAreaField, validators

app = Flask(__name__)
app.config.from_object('config')
db = SQLAlchemy(app)
lm = LoginManager()
lm.init_app(app)
lm.login_view = 'login'
from app import views, models