Changeset View
Changeset View
Standalone View
Standalone View
swh/indexer/storage/api/server.py
| # Copyright (C) 2015-2019 The Software Heritage developers | # Copyright (C) 2015-2019 The Software Heritage developers | ||||
| # See the AUTHORS file at the top-level directory of this distribution | # See the AUTHORS file at the top-level directory of this distribution | ||||
| # License: GNU General Public License version 3, or any later version | # License: GNU General Public License version 3, or any later version | ||||
| # See top-level LICENSE file for more information | # See top-level LICENSE file for more information | ||||
| import os | import os | ||||
| import logging | import logging | ||||
| from swh.core import config | from swh.core import config | ||||
| from swh.core.api import (RPCServerApp, error_handler, | from swh.core.api import (RPCServerApp, error_handler, | ||||
| encode_data_server as encode_data) | encode_data_server as encode_data) | ||||
| from swh.indexer.storage import ( | from swh.indexer.storage import ( | ||||
| get_indexer_storage, INDEXER_CFG_KEY | get_indexer_storage, INDEXER_CFG_KEY | ||||
| ) | ) | ||||
| from swh.indexer.storage.exc import IndexerStorageArgumentException | |||||
| from swh.indexer.storage.interface import IndexerStorageInterface | from swh.indexer.storage.interface import IndexerStorageInterface | ||||
| def get_storage(): | def get_storage(): | ||||
| global storage | global storage | ||||
| if not storage: | if not storage: | ||||
| storage = get_indexer_storage(**app.config[INDEXER_CFG_KEY]) | storage = get_indexer_storage(**app.config[INDEXER_CFG_KEY]) | ||||
| return storage | return storage | ||||
| app = RPCServerApp(__name__, | app = RPCServerApp(__name__, | ||||
| backend_class=IndexerStorageInterface, | backend_class=IndexerStorageInterface, | ||||
| backend_factory=get_storage) | backend_factory=get_storage) | ||||
| storage = None | storage = None | ||||
| @app.errorhandler(Exception) | @app.errorhandler(Exception) | ||||
| def my_error_handler(exception): | def my_error_handler(exception): | ||||
| return error_handler(exception, encode_data) | return error_handler(exception, encode_data) | ||||
| @app.errorhandler(IndexerStorageArgumentException) | |||||
| def argument_error_handler(exception): | |||||
| return error_handler(exception, encode_data, status_code=400) | |||||
| @app.route('/') | @app.route('/') | ||||
| def index(): | def index(): | ||||
| return 'SWH Indexer Storage API server' | return 'SWH Indexer Storage API server' | ||||
| api_cfg = None | api_cfg = None | ||||
| ▲ Show 20 Lines • Show All 63 Lines • Show Last 20 Lines | |||||