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 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()
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()