def setUp(self): self.app = Flask(__name__) self.app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///:memory:' self.app.config['USER_LIST'] = ['Test Admin'] db.init_app(self.app) d = TempDirectory() d.makedir('dataset') d.write('dataset/files/testfile.jpg', 'abc') d.write('dataset/testfile2.jpg', 'abc') self.directory = d self.app.config['DATASET_ROOT'] = d.path with self.app.app_context(), TempDirectory() as d: db.create_all() lib.models.buildDB() dataset = lib.models.Datasets() dataset.name = 'Test' dataset.path = "dataset" db.session.add(dataset) db.session.commit()
def setUp(self): data.app.config['TESTING'] = True data.app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///:memory:' data.app.config['USER_LIST'] = ['Test Admin'] db.init_app(data.app) d = TempDirectory() d.makedir('dataset') d.write('dataset/files/testfile.jpg', 'abc') d.write('dataset/testfile2.jpg', 'abc') self.directory = d data.app.config['DATASET_ROOT'] = d.path with data.app.app_context(), TempDirectory() as d: db.create_all() lib.models.buildDB() license = lib.models.License.query.first() provider = lib.models.Provider.query.first() dataset = lib.models.Datasets() dataset.name = 'Test' dataset.license = [license] dataset.provider = [provider] dataset.path = "dataset" db.session.add(dataset) db.session.commit() self.app = data.app.test_client()
models.Datasets.datasetID == datasetID ).first() return(render_template('confirm.html', dataset=dataset)) @app.route('/del/<int:datasetID>') def delDataset(datasetID): if session.get('logged_in'): try: dataset = models.Datasets.query.filter( models.Datasets.datasetID == datasetID ).first() db.session.delete(dataset) db.session.commit() flash('Deleted!') return redirect(url_for('index')) except Exception as e: return(render_template('error.html', message=e)) @app.after_request def add_ua_compat(response): response.headers['X-UA-Compatible'] = 'IE=Edge' return response if __name__ == "__main__": with app.app_context(): db.create_all() models.buildDB() app.run(host='0.0.0.0', port=8000)
dataset = models.Datasets.query.filter( models.Datasets.datasetID == datasetID).first() return (render_template('confirm.html', dataset=dataset)) @app.route('/del/<int:datasetID>') def delDataset(datasetID): if session.get('logged_in'): try: dataset = models.Datasets.query.filter( models.Datasets.datasetID == datasetID).first() db.session.delete(dataset) db.session.commit() flash('Deleted!') return redirect(url_for('index')) except Exception as e: return (render_template('error.html', message=e)) @app.after_request def add_ua_compat(response): response.headers['X-UA-Compatible'] = 'IE=Edge' return response if __name__ == "__main__": with app.app_context(): db.create_all() models.buildDB() app.run(host='0.0.0.0', port=8000)