mirror of https://github.com/kianby/stacosys
draft tinydb persistence
parent
b258194d6f
commit
aa122f3138
@ -0,0 +1,22 @@
|
|||||||
|
#!/usr/bin/python
|
||||||
|
# -*- coding: UTF-8 -*-
|
||||||
|
|
||||||
|
from tinydb import TinyDB
|
||||||
|
from tinydb.storages import MemoryStorage
|
||||||
|
|
||||||
|
from stacosys.conf import config
|
||||||
|
|
||||||
|
|
||||||
|
class Persistence:
|
||||||
|
def __init__(self):
|
||||||
|
db_file = config.get(config.DB_FILE)
|
||||||
|
if db_file:
|
||||||
|
self.db = TinyDB(db_file, sort_keys=True, indent=4, separators=(",", ": "))
|
||||||
|
else:
|
||||||
|
self.db = TinyDB(storage=MemoryStorage)
|
||||||
|
|
||||||
|
def get_db(self):
|
||||||
|
return self.db
|
||||||
|
|
||||||
|
def get_table_comments(self):
|
||||||
|
return self.db.table("comments")
|
@ -0,0 +1,20 @@
|
|||||||
|
#!/usr/bin/python
|
||||||
|
# -*- coding: UTF-8 -*-
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture
|
||||||
|
def persistence():
|
||||||
|
from stacosys.conf import config
|
||||||
|
|
||||||
|
config.params = {"main.db_file": None}
|
||||||
|
from stacosys.core import persistence
|
||||||
|
|
||||||
|
return persistence.Persistence()
|
||||||
|
|
||||||
|
|
||||||
|
def test_init_persistence(persistence):
|
||||||
|
assert persistence is not None
|
||||||
|
assert persistence.get_db() is not None
|
||||||
|
assert persistence.get_table_comments() is not None
|
Loading…
Reference in New Issue