def test_smallkeylist():
db_ = TinyDB(storage=KeystoreMiddleware(MemoryStorage, ['a', 'b', 'c', 'd']))
tb_ = db_.table()
ks_tb_ = db_._storage.read()['keystore']
tb_.insert({'myid1':1, 'myid2':{'myid3':3}, 'myid4':4})
assert len(tb_) == 1
assert len(ks_tb_) == 4
assert 'myid1' in ks_tb_.keys()
assert 'myid2' in ks_tb_.keys()
assert 'myid3' in ks_tb_.keys()
assert 'myid4' in ks_tb_.keys()
assert 'myid5' not in ks_tb_.keys()
assert 'myid6' not in ks_tb_.keys()
assert ks_tb_['myid1'] in ['a', 'b', 'c', 'd']
assert ks_tb_['myid2'] in ['a', 'b', 'c', 'd']
assert ks_tb_['myid3'] in ['a', 'b', 'c', 'd']
assert ks_tb_['myid4'] in ['a', 'b', 'c', 'd']
old_ks_tb_ = ks_tb_.copy()
tb_.insert({'myid5':{'myid6':2}})
assert len(tb_) == 2
assert len(ks_tb_) == 4
assert 'myid5' not in ks_tb_.keys()
assert 'myid6' not in ks_tb_.keys()
assert ks_tb_['myid1'] == old_ks_tb_['myid1']
assert ks_tb_['myid2'] == old_ks_tb_['myid2']
assert ks_tb_['myid3'] == old_ks_tb_['myid3']
assert ks_tb_['myid4'] == old_ks_tb_['myid4']
def test_json_readwrite(tmpdir):
"""
Regression test for issue #1
"""
path = str(tmpdir.join('test.db'))
# Create TinyDB instance
db = TinyDB(path, storage=JSONStorage)
item = {'name': 'A very long entry'}
item2 = {'name': 'A short one'}
get = lambda s: db.get(where('name') == s)
db.insert(item)
assert get('A very long entry') == item
db.remove(where('name') == 'A very long entry')
assert get('A very long entry') is None
db.insert(item2)
assert get('A short one') == item2
db.remove(where('name') == 'A short one')
assert get('A short one') is None
def __init__(self,
storage_type=constants.DB_STORAGE_TYPE,
abs_path=Path(constants.DB_REL_DIR,
constants.DB_FILE_NAME).absolute()):
self.db = TinyDB(abs_path,
sort_keys=True,
indent=4,
storage=storage_type,
create_dirs=True)
self.db_path = Path(abs_path)
def test_deepreplace():
db_ = TinyDB(storage=KeystoreMiddleware(MemoryStorage, ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']))
tb_ = db_.table()
ks_tb_ = db_._storage.read()['keystore']
tb_.insert({'myid1':1})
assert len(tb_) == 1
assert len(ks_tb_) == 1
assert 'myid1' in ks_tb_.keys()
assert ks_tb_['myid1'] == 'a'
old_ks_tb_ = ks_tb_.copy()
tb_.insert({'myid1':{'myid2':2}})
assert len(tb_) == 2
assert len(ks_tb_) == 2
assert 'myid2' in ks_tb_.keys()
assert ks_tb_['myid2'] == 'b'
assert ks_tb_['myid1'] == old_ks_tb_['myid1']
old_ks_tb_ = ks_tb_.copy()
tb_.insert({'myid1':{'myid2':{'myid3':3}}})
assert len(tb_) == 3
assert len(ks_tb_) == 3
assert 'myid3' in ks_tb_.keys()
assert ks_tb_['myid3'] == 'c'
assert ks_tb_['myid1'] == old_ks_tb_['myid1']
assert ks_tb_['myid2'] == old_ks_tb_['myid2']
old_ks_tb_ = ks_tb_.copy()
tb_.insert({'myid1':{'myid2':{'myid3':{'myid4':4}}}})
assert len(tb_) == 4
assert len(ks_tb_) == 4
assert 'myid4' in ks_tb_.keys()
assert ks_tb_['myid4'] == 'd'
assert ks_tb_['myid1'] == old_ks_tb_['myid1']
assert ks_tb_['myid2'] == old_ks_tb_['myid2']
assert ks_tb_['myid3'] == old_ks_tb_['myid3']
old_ks_tb_ = ks_tb_.copy()
tb_.insert({'myid1':{'myid2':{'myid3':{'myid4':{'myid5':5}}}}})
assert len(tb_) == 5
assert len(ks_tb_) == 5
assert 'myid5' in ks_tb_.keys()
assert ks_tb_['myid5'] == 'e'
assert ks_tb_['myid1'] == old_ks_tb_['myid1']
assert ks_tb_['myid2'] == old_ks_tb_['myid2']
assert ks_tb_['myid3'] == old_ks_tb_['myid3']
assert ks_tb_['myid4'] == old_ks_tb_['myid4']
class GameDatabase:
def __init__(self,
storage_type=constants.DB_STORAGE_TYPE,
abs_path=Path(constants.DB_REL_DIR,
constants.DB_FILE_NAME).absolute()):
self.db = TinyDB(abs_path,
sort_keys=True,
indent=4,
storage=storage_type,
create_dirs=True)
self.db_path = Path(abs_path)
#
# try:
# if self.db_path.absolute().is_file():
# with open(self.db_path, "r") as read_file:
# self.loaded_json_db = json.load(read_file)
# else:
# self.db_path.parent.mkdir(parents=True, exist_ok=True)
# except OSError as e:
# if e.errno != errno.EEXIST:
# raise
# def saveDatabaseAsJSON(self, json_database):
# self.db.table()
# try:
# with self.db_path.absolute().open("w") as write_file:
# # json.dump(json_database, write_file)
# return
# except OSError as e:
# if e.errno != errno.EEXIST:
# raise
def getGamePath(self, game_name):
return self.db.search(Query()['Game'] == game_name
and Query()['System'] == 'Windows')
def getLibraryPath(self, lib_name):
return self.db.search(Query()['Library'] == lib_name)
def addLibraryPath(self, lib_name, lib_abbrev, lib_path):
self.db.table("Libraries").insert({
"LibService": lib_name,
"Abbrev": lib_abbrev,
"Path": lib_path
})
def rebuildDatabase(self, table_name):
for game in self.db.search(Query()['Game']):
print()
def test_onelevelreplace():
db_ = TinyDB(storage=KeystoreMiddleware(MemoryStorage, ['a', 'b', 'c', 'd']))
tb_ = db_.table()
ks_tb_ = db_._storage.read()['keystore']
tb_.insert({'myid1':1})
assert len(tb_) == 1
assert len(ks_tb_) == 1
assert 'myid1' in ks_tb_.keys()
assert ks_tb_['myid1'] == 'a'
tb_.insert({'myid2':5})
assert len(tb_) == 2
assert len(ks_tb_) == 2
assert 'myid2' in ks_tb_.keys()
assert ks_tb_['myid2'] == 'b'
"""
def __init__(self, a_collection: Table):
super().__init__(a_collection)
def insert(self, item):
self._a_collection.insert(item)
def contains(self, id_) -> bool:
return self._a_collection.contains(
Query().itemListInfoExt.itId == int(id_))
class _BidsCollection(BaseCollection, ABC):
"""
bids.insert_many(bids) downloaded list of bids for particular allegro item
"""
def __init__(self, a_collection: Table):
super().__init__(a_collection)
def insert(self, bids):
self._a_collection.insert_multiple(bids)
env = ensure_env()
with env.prefixed("TALPA_ALLEGRO_TINYDB_"):
DB_FILE = env("DB_FILE")
tiny_db = TinyDB(DB_FILE)
def test_keylistinit_duplicateselimated():
db_ = TinyDB(storage=KeystoreMiddleware(MemoryStorage, ['a', 'b', 'c', 'b', 'c', 'd', 'e', 'd', 'b']))
assert db_._storage.keylist == ['a', 'b', 'c', 'd', 'e']
def test_keylistinit_unprovided():
db_ = TinyDB(storage=KeystoreMiddleware(MemoryStorage))
assert db_._storage.keylist == []
def test_keylistinit_empty():
db_ = TinyDB(storage=KeystoreMiddleware(MemoryStorage, []))
assert db_._storage.keylist == []
def test_keylistinit_withelements():
db_ = TinyDB(storage=KeystoreMiddleware(MemoryStorage, ['a', 'b', 'c']))
assert db_._storage.keylist == ['a', 'b', 'c']
def test_convertfromnonmiddleware():
db_ = TinyDB('test.json')
db_.purge_tables()
tb_ = db_.table('testtable')
populate_db_simple(tb_)
tb_.insert({'myid1':1})
tb_.insert({'myid1':2, 'myid2':{'myid3':3}})
tb_.insert({'myid1':3, 'myid2':{'myid3':{'myid4':4}}})
assert len(tb_) == 6
assert tb_.count(where('char') == 'a') == 1
assert tb_.count(where('char') == 'b') == 1
assert tb_.count(where('char') == 'c') == 1
assert tb_.get(where('char') == 'a')['char'] == 'a'
assert tb_.get(where('char') == 'b')['char'] == 'b'
assert tb_.get(where('char') == 'c')['char'] == 'c'
assert tb_.get(where('myid1') == 1)['myid1'] == 1
assert tb_.get(where('myid1') == 2)['myid1'] == 2
assert tb_.get(where('myid1') == 2)['myid2']['myid3'] == 3
assert tb_.get(where('myid1') == 3)['myid1'] == 3
assert tb_.get(where('myid1') == 3)['myid2']['myid3']['myid4'] == 4
db_.close()
db_ = TinyDB('test.json', storage=KeystoreMiddleware(key_list=['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']))
tb_ = db_.table('testtable')
assert len(tb_) == 6
assert tb_.count(where('char') == 'a') == 1
assert tb_.count(where('char') == 'b') == 1
assert tb_.count(where('char') == 'c') == 1
assert tb_.get(where('char') == 'a')['char'] == 'a'
assert tb_.get(where('char') == 'b')['char'] == 'b'
assert tb_.get(where('char') == 'c')['char'] == 'c'
assert tb_.get(where('myid1') == 1)['myid1'] == 1
assert tb_.get(where('myid1') == 2)['myid1'] == 2
assert tb_.get(where('myid1') == 2)['myid2']['myid3'] == 3
assert tb_.get(where('myid1') == 3)['myid1'] == 3
assert tb_.get(where('myid1') == 3)['myid2']['myid3']['myid4'] == 4
db_.close()
os.remove('test.json')
def test_mutipletable_add():
db_ = TinyDB(storage=KeystoreMiddleware(MemoryStorage, ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j']))
tb_ = [db_.table('tb1'), db_.table('tb2'), db_.table('tb3'), db_.table('tb4')]
ks_tb_ = db_._storage.read()['keystore']
assert set(db_.tables()) == set(['_default', 'keystore', 'tb1', 'tb2', 'tb3', 'tb4'])
assert len(ks_tb_) == 0
assert len(tb_[0]) == 0
assert len(tb_[1]) == 0
assert len(tb_[2]) == 0
assert len(tb_[3]) == 0
tb_[0].insert({'myid1':1})
assert len(ks_tb_) == 1
assert len(tb_[0]) == 1
assert len(tb_[1]) == 0
assert len(tb_[2]) == 0
assert len(tb_[3]) == 0
assert 'myid1' in ks_tb_.keys()
assert ks_tb_['myid1'] == 'a'
old_ks_tb_ = ks_tb_.copy()
tb_[1].insert({'myid1':{'myid2':'someothervalue'}})
assert len(ks_tb_) == 2
assert len(tb_[0]) == 1
assert len(tb_[1]) == 1
assert len(tb_[2]) == 0
assert len(tb_[3]) == 0
assert ks_tb_['myid1'] == old_ks_tb_['myid1']
assert 'myid2' in ks_tb_.keys()
assert ks_tb_['myid2'] == 'b'
old_ks_tb_ = ks_tb_.copy()
tb_[2].insert({'myid3':{'myid4':'c'}})
assert len(ks_tb_) == 4
assert len(tb_[0]) == 1
assert len(tb_[1]) == 1
assert len(tb_[2]) == 1
assert len(tb_[3]) == 0
assert ks_tb_['myid1'] == old_ks_tb_['myid1']
assert ks_tb_['myid2'] == old_ks_tb_['myid2']
assert 'myid3' in ks_tb_.keys()
assert 'myid4' in ks_tb_.keys()
assert ks_tb_['myid3'] in ['c', 'd']
assert ks_tb_['myid4'] in ['c', 'd']
old_ks_tb_ = ks_tb_.copy()
tb_[3].insert({'myid2':{'myid5':5}})
assert len(ks_tb_) == 5
assert len(tb_[0]) == 1
assert len(tb_[1]) == 1
assert len(tb_[2]) == 1
assert len(tb_[3]) == 1
assert ks_tb_['myid1'] == old_ks_tb_['myid1']
assert ks_tb_['myid2'] == old_ks_tb_['myid2']
assert ks_tb_['myid3'] == old_ks_tb_['myid3']
assert ks_tb_['myid4'] == old_ks_tb_['myid4']
assert 'myid5' in ks_tb_.keys()
assert ks_tb_['myid5'] == 'e'