Differential D2630 Diff 9529 swh/storage/api/server.py

Changeset View

Standalone View

swh/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.storage import get_storage as get_swhstorage			from swh.storage import get_storage as get_swhstorage
	from swh.core.api import (RPCServerApp,			from swh.core.api import (RPCServerApp,
	error_handler,			error_handler,
	encode_data_server as encode_data)			encode_data_server as encode_data)

	from ..interface import StorageInterface			from ..interface import StorageInterface
	from ..metrics import timed			from ..metrics import timed
				from ..exc import StorageArgumentException


	def get_storage():			def get_storage():
	global storage			global storage
	if not storage:			if not storage:
	storage = get_swhstorage(**app.config['storage'])			storage = get_swhstorage(**app.config['storage'])

	return storage			return storage


	app = RPCServerApp(__name__,			app = RPCServerApp(__name__,
	backend_class=StorageInterface,			backend_class=StorageInterface,
	backend_factory=get_storage)			backend_factory=get_storage)
	storage = None			storage = None


				@app.errorhandler(StorageArgumentException)
				def argument_error_handler(exception):
				return error_handler(exception, encode_data, status_code=400)


	@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.route('/')			@app.route('/')
	@timed			@timed
	def index():			def index():
	▲ Show 20 Lines • Show All 78 Lines • Show Last 20 Lines