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