def dbcreate():
db.create_all()
if not os.path.exists(SQLALCHEMY_MIGRATE_REPO):
api.create(SQLALCHEMY_MIGRATE_REPO, 'database repository')
api.version_control(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO)
else:
api.version_control(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO, api.version(SQLALCHEMY_MIGRATE_REPO))
def setUp(self):
app.config['TESTING'] = True
app.config['WTF-CSRF_DATABASE_URI'] = False
app.config['SQLALCHEMY_DATABASE_RUI'] = 'sqlite:///' + os.path.join(
basedir, 'test.db')
self.app = app.test_client()
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()
seed_data(db)
self.client = self.app.test_client()
def init_db(test_client):
# Create the database and the database table
db.create_all()
yield db
db.session.remove()
db.drop_all()
def setUp(self):
"""Set up a blank temp database before each test"""
basedir = os.path.abspath(os.path.dirname(__file__))
app.config['TESTING'] = True
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///' + \
os.path.join(basedir, TEST_DB)
self.app = app.test_client()
db.create_all()
def setUp(self):
db.create_all()
user = User()
user.username = '******'
user.password = '******'
db.session.add(user)
db.session.commit()
def dbcreate():
db.create_all()
if not os.path.exists(SQLALCHEMY_MIGRATE_REPO):
api.create(SQLALCHEMY_MIGRATE_REPO, 'database repository')
api.version_control(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO)
else:
api.version_control(SQLALCHEMY_DATABASE_URI, SQLALCHEMY_MIGRATE_REPO,
api.version(SQLALCHEMY_MIGRATE_REPO))
def setUp(self):
app.config['TESTING'] = True
app.config['DEBUG'] = False
app.config['DATABASE_URL'] = 'sqlite:///' + \
os.path.join(app.config['BASE_PATH'], TEST_DB)
self.app = app.test_client()
db.drop_all()
db.create_all()
self.assertEqual(app.debug, False)
def createdb():
if path.exists(db_name):
remove(db_name)
print(db_name + " removed")
db.create_all()
query = input("Want to add an admin? (Y/N): ")
if query.lower() == "y":
import add_user
def seed_db():
db.drop_all()
db.create_all()
Tag.generate_data()
Frontpage_post.generate_data()
AdminUser.generate_data()
Startup.generate_data()
Job_applicant.generate_data()
Job_position.generate_data()
print("populated databse")
def db(app, request):
'''
Provide the transactional fixtures with access to the database via a Flask-SQLAlchemy
database connection.
'''
#db = SQLAlchemy(app=app)
from flaskr import models
with app.app_context():
_db.drop_all()
_db.create_all()
return db
def app():
app = create_app({
'TESTING': True,
'WTF_CSRF_ENABLED': False,
'SECRET_KEY': 'dev',
'SQLALCHEMY_DATABASE_URI': \
'postgresql:///portfoliotest',
# 'SQLALCHEMY_ECHO': True,
'SQLALCHEMY_TRACK_MODIFICATIONS': False
})
with app.app_context():
db.drop_all()
db.create_all()
db.session.commit()
yield app
def app():
app = create_app(TestConfig)
with app.app_context():
db.drop_all()
db.create_all()
user1 = User(username='******', password=generate_password_hash('test'))
user2 = User(username='******',
password=generate_password_hash('test2'))
db.session.add_all([user1, user2])
db.session.commit()
post = Post(title='test title',
body='test',
created='2018-01-01 00:00:00',
author_id=1)
db.session.add(post)
db.session.commit()
yield app
def admin(username, password):
"""Create user."""
db.create_all()
user = User.query.first()
if user is not None:
click.echo('Updating user...')
user.username = username
user.setPassword(password) # 设置密码
else:
click.echo('Creating user...')
user = User(username=username, name='Admin')
user.setPassword(password) # 设置密码
db.session.add(user)
db.session.commit() # 提交数据库会话
click.echo('Done.')
def seed_command():
db.init_app(current_app)
db.drop_all()
db.create_all()
with open(data_filename) as data:
products_data = json.load(data)
products = [to_product_model(p) for p in products_data]
for product in products:
db.session.add(product)
for photo in product.photos:
db.session.add(photo)
db.session.commit()
print("seed Done!")
def create_db():
db.drop_all()
db.create_all()
boards = [
BoardsModel('technology', 't'),
BoardsModel('off-topic', 'o'),
BoardsModel('news', 'n'),
BoardsModel('images', 'i'),
BoardsModel('client-test', 'c')
]
for board in boards:
db.session.add(board)
try:
db.session.commit()
print("changes commited to database successfully")
except:
print("error commiting changes to database")
def setUpClass(cls):
try:
cls.client = webdriver.Chrome(
executable_path=cls.webdriver_exe_path,
chrome_options=cls.options)
except:
pass
if cls.client: # Only execute these tests if our web driver connects to a browser.
cls.test_app = create_app(
'testing'
) # Create our app within the app factory, using 'testing' config.
# Here, we configure our application context and our request context.
cls.app_context = cls.test_app.app_context(
) # Creates an app context.
cls.request_context = cls.test_app.test_request_context()
cls.app_context.push()
cls.request_context.push()
# Anything between the app_context.push() and the app_context.pop() method will be considered to run...
# ...within a 'app_context()' with statement.
db.create_all() # Creates our database.
# Configures our app to run in a new thread.
cls.server_thread = threading.Thread(name="TestInstance",
target=cls.test_app.run)
cls.server_thread.start() # Runs our test_app on a new thread.
cls.client.get(cls.root_url) # Takes us to the root URL.
print(f"The currently set root url is: {cls.root_url}")
time.sleep(
1
) # Give the server a second to ensure it is up, before running any tests prematurely.
def app():
"""Create and configure a new app instance for each test."""
# create the app with common test config
app = create_app({
'TESTING': True,
'SQLALCHEMY_DATABASE_URI': 'sqlite:///:memory:',
})
# create the database and load test data
with app.app_context():
db.create_all()
user1 = User(username='******', password=generate_password_hash('test'))
user2 = User(username='******',
password=generate_password_hash('other'))
post = Post(title='test title',
body=('test\nbody'),
author_id=1,
created=datetime(2018, 1, 1))
db.session.add_all([user1, user2, post])
db.session.commit()
yield app
def create_db():
db.create_all()
def init():
print("==============================")
db.create_all()
def create_db():
db.drop_all()
db.create_all()
db.session.commit()
def dbInit():
''' 数据库初始化 '''
from flaskr import db
db.create_all()
def setUp(self):
db.create_all()
db.session.add(Entries('user_name', 'title', 'text', False))
db.session.commmit()
# flaskr 是專案裡的資料夾名稱
from flaskr import app
from flaskr import db
from flask import render_template, flash, redirect, url_for, request
from flaskr.model import UserRegister
from flaskr.form import FormRegister, FormLogin
from flask_login import login_user, current_user, login_required, logout_user
# db一樣要初始化
db.init_app(app)
with app.app_context():
db.create_all()
@app.route('/')
@login_required
def index():
return render_template('base.html')
@app.route('/register', methods=['GET', 'POST'])
def register():
form = FormRegister()
if form.validate_on_submit():
user = UserRegister(username=form.username.data,
email=form.email.data,
password=form.password.data)
db.session.add(user)
db.session.commit()
return '註冊成功！'
def setUp(self):
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite://'
db.create_all()
def init_db():
db.create_all()
def init():
db.create_all()
def init_db():
"""创建初始数据"""
from flaskr import db
print("\n\n\n\n---------删除表-----------")
db.drop_all()
print("\n\n\n\n---------创建表-----------")
db.create_all()
# 创建用户
print("\n\n\n\n---------创建用户-----------")
from flaskr.model import User
gabe = User(name='gabe')
jack = User(name='jack')
rose = User(name='rose')
# db.session.add(gabe)
# db.session.add(jack)
# db.session.add(rose)
db.session.add_all([gabe, jack, rose])
db.session.commit()
# 创建角色
print("\n\n\n\n---------创建角色-----------")
from flaskr.model import Role
adminRole = Role(name='admin')
userRole = Role(name='user')
guestRole = Role(name='guest')
db.session.add_all([adminRole, userRole, guestRole])
db.session.commit()
# 创建团队
print("\n\n\n\n---------创建团队-----------")
from flaskr.model import Group
rayjarGroup = Group(name="rayjar", leader=gabe)
titanicGroup = Group(name="titanic", leader=jack)
db.session.add_all([rayjarGroup, titanicGroup])
db.session.commit()
# 创建菜单
print("\n\n\n\n---------创建菜单-----------")
from flaskr.model import Menu
welcomeMenu = Menu(name="welcome", path="/welcome", icon="smile", sort=0)
dashboardMenu = Menu(name="dashboard",
path="/dashboard",
icon="dashboard",
sort=1)
db.session.add_all([welcomeMenu, welcomeMenu])
db.session.commit()
usersMenu = Menu(name="users",
path="/dashboard/users",
sort=0,
parent=dashboardMenu)
authsMenu = Menu(name="auths",
path="/dashboard/auths",
sort=0,
parent=dashboardMenu)
rolesMenu = Menu(name="roles",
path="/dashboard/roles",
sort=0,
parent=dashboardMenu)
menusMenu = Menu(name="groups",
path="/dashboard/groups",
sort=0,
parent=dashboardMenu)
groupsMenu = Menu(name="menus",
path="/dashboard/menus",
sort=0,
parent=dashboardMenu)
usergroupsMenu = Menu(name="usergroups",
path="/dashboard/usergroups",
sort=0,
parent=dashboardMenu)
db.session.add_all([
usersMenu, authsMenu, rolesMenu, menusMenu, groupsMenu, usergroupsMenu
])
db.session.commit()
# 关联角色菜单
print("\n\n\n\n---------关联角色菜单-----------")
adminRole.menus = [
welcomeMenu, dashboardMenu, usersMenu, authsMenu, rolesMenu, menusMenu,
groupsMenu, usergroupsMenu
]
userRole.menus = [welcomeMenu, dashboardMenu, usersMenu, menusMenu]
db.session.commit()
# 关联用户角色
print("\n\n\n\n---------关联用户角色-----------")
gabe.roles = [adminRole, userRole, guestRole]
jack.roles = [userRole, guestRole]
rose.roles = [guestRole]
db.session.commit()
# 关联用户团队
print("\n\n\n\n---------关联用户团队-----------")
# gabe.groups = [rayjarGroup]
# jack.groups = [rayjarGroup,titanicGroup]
# rose.groups = [titanicGroup]
rayjarGroup.menbers = [gabe, jack]
titanicGroup.menbers = [jack, rose]
db.session.commit()
# 创建帐号
print("\n\n\n\n---------创建帐号-----------")
from flaskr.model import Auth
authData = {
"user":
gabe,
"authType":
"account",
"authName":
gabe.name,
"authCode":
generate_password_hash("123456"),
# "password": "******",
"verifyTime":
datetime.datetime.utcnow(),
"expireTime":
datetime.datetime.utcnow() + datetime.timedelta(seconds=60 * 60 * 24)
}
auth = Auth(**authData)
db.session.add(auth)
db.session.commit()
def init():
db.create_all()
def init_db():
db.create_all()
from flaskr import create_app, db
# Press the green button in the gutter to run the script.
if __name__ == '__main__':
app = create_app()
app.app_context().push()
db.create_all(app=app)
app.run(host='0.0.0.0', port=5000, debug=True)
# See PyCharm help at https://www.jetbrains.com/help/pycharm/
def init():
db.create_all()
#このファイルを作成後、python shellで以下の操作し、DBを初期化する。
#from flaskr.models import init
#init()
def setUp(self): # 各テスト毎に呼ばれる
db.create_all() # テーブル全作成
def init():
db.drop_all()
db.create_all()
def setUp(self):
self.db_fd, app.DATABASE = tempfile.mkstemp()
self.app = app.test_client()
db.create_all()
