def files_location(app, db):
try:
from invenio_files_rest.models import Location
loc = Location()
loc.name = 'test'
loc.uri = Path(tempfile.gettempdir()).as_uri()
loc.default = True
db.session.add(loc)
db.session.commit()
except ImportError:
pass
def init_default_location():
"""
Add default Location, if not already present.
Used by Travis as well.
"""
if not Location.query.filter(Location.name == 'default').count():
loc = Location()
loc.name = 'default'
loc.default = True
loc.uri = '/virtualenv/files/'
db.session.add(loc)
db.session.commit()
else:
error("Default location already exists.")
def init_default_location():
"""
Add default Location, if not already present.
Used by Travis as well.
"""
if not Location.query.filter(Location.name == 'default').count():
loc = Location()
loc.name = 'default'
loc.default = True
loc.uri = '/virtualenv/files/'
db.session.add(loc)
db.session.commit()
else:
error("Default location already exists.")
