def init_dev():
"""Init data with database."""
db.drop_all()
db.create_all()
click.echo('[ok] init databases')
# init actions
init_user()
def create_app(config_name):
app = Flask(config_name)
app.config.from_object(config[config_name])
config[config_name].init_app(app)
app_admin.init_app(app)
babel.init_app(app)
celery.init_app(app)
bootstrap.init_app(app)
db.init_app(app)
with app.app_context():
print current_app.name
db.create_all()
login_manager.init_app(app)
app_admin.base_template = 'layout.html'
from main import main as main_blueprint
from admin import admin as admin_blueprint
from auth import auth as auth_blueprint
app.register_blueprint(main_blueprint)
app.register_blueprint(auth_blueprint, url_prefix='/auth')
import os.path
_dir = os.path.dirname(os.path.abspath(__file__))
app.template_folder = os.path.join(_dir, 'templates')
app.static_folder = os.path.join(_dir, 'static')
return app
def setUp(self):
app.config.update(TESTING=True,
SQLALCHEMY_DATABASE_URI='sqlite:///:memory:')
self.context = app.test_request_context()
self.context.push()
self.client = app.test_client()
self.runner = app.test_cli_runner()
db.create_all()
def forge(drop):
if drop:
click.echo('正在初始化数据库...')
db.drop_all()
db.create_all()
click.echo('数据库初始化完毕！')
click.echo('正在生成分类...')
fake.fake_categories()
click.echo('正在生成条目...')
fake.fake_items()
click.echo('正在生成记录...')
fake.fake_records()
click.echo('正在生成标签...')
fake.fake_tags()
click.echo('数据生成完毕！')
def resetdb():
"""
Drops all tables in our database, installs postgis extension if
necessary and recreates all tables
"""
from app.exts import db
# drop all tables
db.drop_all()
# install postgis if not already done
sql = "SELECT count(extname) FROM pg_extension WHERE extname = 'postgis'"
if (db.engine.execute(sql).rowcount == 0): db.engine.execute("CREATE EXTENSION postgis;")
# create all tables
db.create_all()
def resetdb():
db.drop_all()
db.create_all()
def setUp(self):
self.user = UserFactory()
db.create_all()
def setUp(self):
db.create_all()
def register_extensions(app):
db.init_app(app)
from app import models
db.create_all(app=app)
def init_db():
click.echo('正在初始化数据库...')
db.drop_all()
db.create_all()
click.echo('数据库初始化完毕！')
# 管理员登录日志
class Adminlog(db.Model):
__tablename__ = "adminlog"
id = db.Column(db.Integer, primary_key=True) # 编号
admin_id = db.Column(db.Integer, db.ForeignKey('admin.id')) # 所属管理员编号
admin = db.relationship('Admin', backref='adminlogs') #管理员登录日志外键关系关联
ip = db.Column(db.String(100)) # 登录IP
addtime = db.Column(db.DateTime, index=True, default=datetime.now) # 登录时间
def __repr__(self):
return "<Adminlog %r>" % self.id
# 操作日志
class Oplog(db.Model):
__tablename__ = "oplog"
id = db.Column(db.Integer, primary_key=True) # 编号
admin_id = db.Column(db.Integer, db.ForeignKey('admin.id')) # 所属管理员编号
admin = db.relationship('Admin', backref='oplogs') # 操作日志外键关系关联
ip = db.Column(db.String(100)) # 登录IP
reason = db.Column(db.String(600)) # 操作原因
addtime = db.Column(db.DateTime, index=True, default=datetime.now) # 登录时间
def __repr__(self):
return "<Oplog %r>" % self.id
if __name__ == '__main__':
db.create_all()
def setUp(self):
self.app = create_app('testing')
self.app_context = self.app.app_context()
self.app_context.push()
db.create_all()
def createall(): db.create_all()
def setUp(self):
self.questions = [QuestionFactory() for _ in range(5)]
db.create_all()
