def open(uri, serializer_name='pickle'):
''' open collection by URI,
if collection does not exist kvlite will try to create it
serializer_name: see details in SERIALIZERS section
returns MysqlCollection or SqliteCollection object in case of successful
opening or creation new collection
'''
# TODO use `None` for serializer to store messages in plain text, suitable for strings, integers, etc
manager = CollectionManager(uri)
params = manager.parse_uri(uri)
if params['collection'] not in manager.collections():
manager.create(params['collection'])
collection = manager.collection_class(manager.connection,
params['collection'],
SERIALIZERS[serializer_name])
if collection.meta is None:
collection.meta = {
'name': params['collection'],
'serializer': serializer_name,
'kvlite-version': kvlite.__version__,
}
return collection
def remove(uri):
''' remove collection by URI
'''
manager = CollectionManager(uri)
params = manager.parse_uri(uri)
if params['collection'] in manager.collections():
manager.remove(params['collection'])
def test_sqlite_manager(self):
URI = 'sqlite://tests/db/testdb.sqlite'
collection_name = 'kvlite_test'
manager = CollectionManager(URI)
if collection_name in manager.collections():
manager.remove(collection_name)
self.assertNotIn(collection_name, manager.collections())
manager.create(collection_name)
self.assertIn(collection_name, manager.collections())
manager.remove(collection_name)
self.assertNotIn(collection_name, manager.collections())
def test_mysql_manager(self):
URI = 'mysql://*****:*****@localhost/kvlite_test'
collection_name = 'kvlite_test'
manager = CollectionManager(URI)
if collection_name in manager.collections():
manager.remove(collection_name)
self.assertNotIn(collection_name, manager.collections())
manager.create(collection_name)
self.assertIn(collection_name, manager.collections())
manager.remove(collection_name)
self.assertNotIn(collection_name, manager.collections())
def test_sqlite_manager(self):
URI = 'sqlite://tests/db/testdb.sqlite'
collection_name = 'kvlite_test'
manager = CollectionManager(URI)
if collection_name in manager.collections():
manager.remove(collection_name)
self.assertNotIn(collection_name, manager.collections())
manager.create(collection_name)
self.assertIn(collection_name, manager.collections())
manager.remove(collection_name)
self.assertNotIn(collection_name, manager.collections())
def test_mysql_manager(self):
URI = 'mysql://*****:*****@localhost/kvlite_test'
collection_name = 'kvlite_test'
manager = CollectionManager(URI)
if collection_name in manager.collections():
manager.remove(collection_name)
self.assertNotIn(collection_name, manager.collections())
manager.create(collection_name)
self.assertIn(collection_name, manager.collections())
manager.remove(collection_name)
self.assertNotIn(collection_name, manager.collections())