def setUp(self): db.create_all() self.init_data = init_api_data(self.app) register_restaurants_blueprints(self.app) data = json.loads( self.client.post('/auth', data=json.dumps(dict( username=self.init_data['USER_ADMIN_EMAIL'], password=self.init_data['USER_ADMIN_PASSWORD']) ), content_type='application/json').data.decode('utf-8')) self.john_token = data['access_token'] data = json.loads( self.client.post('/auth', data=json.dumps(dict( username=self.init_data['USER_COMMON_EMAIL'], password=self.init_data['USER_COMMON_PASSWORD']) ), content_type='application/json').data.decode('utf-8')) self.susan_token = data['access_token'] data = json.loads( self.client.post('/auth', data=json.dumps(dict( username=self.init_data['USER_MANAGER_EMAIL'], password=self.init_data['USER_MANAGER_PASSWORD']) ), content_type='application/json').data.decode('utf-8')) self.karl_token = data['access_token']
def db(app): _db.app = app _db.create_all() yield _db # Code after yield executes on teardown _db.session.close() _db.drop_all()
def db(app): url = app.config['SQLALCHEMY_DATABASE_URI'] if not sqlalchemy_utils.database_exists(url): sqlalchemy_utils.create_database(url) db_drop_everything(_db) _db.create_all() yield _db _db.session.close() db_drop_everything(_db)
def create_app(config='config.py', db=db): app = Flask(__name__, instance_relative_config=True) CORS(app) if config: app.config.from_pyfile(config) else: app.config['SQLALCHEMY_DATABASE_URI'] = os.environ.get('DATABASE_URL') app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False app.secret_key = os.environ.get('SECRET_KEY') from api import models db.init_app(app) db.app = app db.create_all() from api.conversation_controller import conversation_api app.register_blueprint(conversation_api) return app
def inituser(): db.create_all() try: group = Group(name='admin', app_add=True, app_edit_all=True, app_edit_my=True, app_drop=True, user_add=True, user_drop=False) db.session.add(group) except: pass try: hash = sha256_crypt.encrypt('admin') user = User(email='admin@intra', password_hash=hash, group_id=1) db.session.add(user) except: pass db.session.commit()
def setUp(self): self.db_fd, self.db_filename = tempfile.mkstemp() app.config['SQLALCHEMY_DATABASE_URI'] = 'postgresql://localhost/monkeystest' app.config['TESTING'] = True app.testing = True app.config['WTF_CSRF_ENABLED'] = False self.app = app.test_client() connect_db(app) if self.logged_in: self.fake_login() with app.app_context(): db.drop_all() db.create_all() load_student() load_coach() load_class_schedule() load_class_instance() load_student_class_instance() load_student_class_schedule()
def create_app(pconfig=None, debug=False): app = Flask(__name__) app.debug = debug configobj = pconfig or \ os.environ.get('PYSTRO_SETTINGS') or \ defaultconfig app.config.from_object(configobj) app.app_context().push() with app.app_context(): db.init_app(app) db.drop_all() try: db.create_all() if not app.testing: init_api_data(app) except (OperationalError, ObjectDeletedError): db.session.rollback() db.session.remove() setup_admin_api_routes(app) setup_public_api_routes(app) JWT(app, authenticate, identity) register_restaurants_blueprints(app) return app
def setup_database(app): with app.app_context(): db.create_all() # db_mock_data.mock_data() db.session.commit()
new_item = ClassSchedule(name=item['name'], day_of_week=item['day_of_week'], time=item['time']) db.session.add(new_item) db.session.commit() def load_class_instance(): """Load class instance from seed_data into database.""" tablename = 'class_instance' data = jsonify_seed_data(tablename) for item in data[tablename]: new_item = ClassInstance(class_schedule_id=item['class_schedule_id'], date=item['date']) db.session.add(new_item) db.session.commit() if __name__ == "__main__": connect_db(app) db.drop_all() db.create_all() load_student() load_coach() load_class_schedule() load_class_instance() load_student_class_instance() load_student_class_schedule()
def setUp(self): self.client = test_app.test_client() db.create_all()
# email = "*****@*****.**" # id = "some_shibboleth_id_of_the_user" # return jsonify({ # "token": create_jwt(identity={ "id": id, "name": name, "email": email }) # }) # current user identity api.add_resource(IdentityResource, "/users/me") # deployments api.add_resource(Deployments, "/deployments") api.add_resource(Deployment, "/deployments/<int:deployment_id>") api.add_resource(DeploymentLog, "/deployments/<int:deployment_id>/log") api.add_resource(DeploymentUndeploy, "/deployments/<int:deployment_id>/undeploy") # templates api.add_resource(Templates, "/templates") api.add_resource(Template, "/templates/<int:template_id>") db.init_app(app) return app if __name__ == "__main__": app = get_app() if not os.path.exists(const.DATABASE_URI): db.create_all(app=app) app.run(debug=True, threaded=True)
def setUp(self): self.app_context = self.app.app_context() self.app_context.push() db.create_all()
def init_db(app): db.drop_all() db.create_all() yield db db.session.close() db.drop_all()