diff --git a/api/rookeries/sites/views.py b/api/rookeries/sites/views.py index 9d4fbb1..7b1a655 100644 --- a/api/rookeries/sites/views.py +++ b/api/rookeries/sites/views.py @@ -34,8 +34,7 @@ def app_config(): site = None db_session = database.SQLAlchemy.session try: - # TODO: Filter by provided APP ID, provided in the header. - site = db_session.query(models.Site).first() + site = db_session.query(models.Site).filter_by(api_key=api_key).first() except Exception as err: logger.error('Error retrieving site because of: %s', err) flask.abort(http.HTTPStatus.INTERNAL_SERVER_ERROR) @@ -59,8 +58,7 @@ def menu_config(): site = None db_session = database.SQLAlchemy.session try: - # TODO: Filter by provided APP ID, provided in the header. - site = db_session.query(models.Site).first() + site = db_session.query(models.Site).filter_by(api_key=api_key).first() except Exception as err: logger.error('Error retrieving site because of: %s', err) flask.abort(http.HTTPStatus.INTERNAL_SERVER_ERROR)