flask-login-example/models.py
Andros Fenollosa f056300d27 First commit
2017-05-17 20:16:30 +02:00

43 lines
1.2 KiB
Python

from os import getenv
from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask_script import Manager
from flask_migrate import Migrate, MigrateCommand
import crypt
import uuid
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = getenv('SQLALCHEMY_DATABASE_URI')
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
db = SQLAlchemy(app)
migrate = Migrate(app, db)
manager = Manager(app)
manager.add_command('db', MigrateCommand)
class User(db.Model):
__tablename__ = 'users'
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(30), nullable=False)
email = db.Column(db.String(254), nullable=False, unique=True)
password = db.Column(db.String(106), nullable=False, unique=False)
is_active = db.Column(db.Boolean)
token = db.Column(db.String(32), nullable=False, unique=False)
def __init__(self, username, email, password):
self.username = username
self.email = email
self.password = crypt.crypt(password, crypt.mksalt(crypt.METHOD_SHA512))
self.is_active = False
self.token = str(uuid.uuid4()).replace('-', '')
def __repr__(self):
return '<User %r>' % self.username
if __name__ == '__main__':
manager.run()