No OneTemporary
Actions

Size

1 KB

Subscribers

None

View Options

	diff --git a/swh/storage/__init__.py b/swh/storage/__init__.py
	index 63ecdaa..4c306f9 100644
	--- a/swh/storage/__init__.py
	+++ b/swh/storage/__init__.py
	@@ -1,40 +1,43 @@
	# Copyright (C) 2015-2016 The Software Heritage developers
	# See the AUTHORS file at the top-level directory of this distribution
	# License: GNU General Public License version 3, or any later version
	# See top-level LICENSE file for more information

	from . import storage

	Storage = storage.Storage


	class HashCollision(Exception):
	pass


	def get_storage(cls, args):
	"""
	Get a storage object of class `storage_class` with arguments
	`storage_args`.

	Args:
	storage (dict): dictionary with keys:
	- - cls (str): storage's class, either 'local' or 'remote'
	+ - cls (str): storage's class, either 'local', 'remote',
	+ or 'memory'
	- args (dict): dictionary with keys

	Returns:
	an instance of swh.storage.Storage (either local or remote)

	Raises:
	ValueError if passed an unknown storage class.

	"""

	if cls == 'remote':
	from .api.client import RemoteStorage as Storage
	elif cls == 'local':
	from .storage import Storage
	+ elif cls == 'memory':
	+ from .in_memory import Storage
	else:
	raise ValueError('Unknown storage class `%s`' % cls)

	return Storage(**args)