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.")