def shutdown_reana_db_session(response_or_exc):
"""Close session on app teardown."""
from reana_db.database import Session as reana_db_session
from invenio_db import db as invenio_db
reana_db_session.remove()
invenio_db.session.remove()
return response_or_exc
def _import_users(admin_access_token, users_csv_file):
"""Import list of users to database.
:param admin_access_token: Admin access token.
:type admin_access_token: str
:param users_csv_file: CSV file object containing a list of users.
:type users_csv_file: _io.TextIOWrapper
"""
admin = User.query.filter_by(id_=ADMIN_USER_ID).one_or_none()
if admin_access_token != admin.access_token:
raise ValueError('Admin access token invalid.')
csv_reader = csv.reader(users_csv_file)
for row in csv_reader:
user = User(id_=row[0], email=row[1], access_token=row[2])
Session.add(user)
Session.commit()
Session.remove()
