def init_db():
data_path = computer_singleton().app_folder / 'data'
if not data_path.exists():
data_path.mkdir()
if settings.IS_LOCAL_SETUP:
Model.set_db_drivers([
FileDbDriver(PickleSerializer(), m,
Path(f'{data_path}/{m.__name__}')) for s, m in Models
])
Index.set_db_drivers([
FileIndexDbDriver(YamlSerializer(), m,
Path(f'{data_path}/{m.__name__}.yaml'))
for m in Indexes
])
else:
import redis
db = redis.Redis(host='localhost', port=6379, db=0)
Model.set_db_drivers([RedisDbDriver(s, m, db) for s, m in Models])
Index.set_db_drivers([RedisIndexDbDriver(m, db) for m in Indexes])
project.create_project(settings.FLOAT_PROJECT_TOKEN, 'float project')
project.create_project(settings.SAMPLES_PROJECT_TOKEN, 'samples project')
def test_setup():
db = redis.Redis(host='localhost', port=6379, db=0)
Model.set_db_drivers([
RedisDbDriver(PickleSerializer(), PickleModel, db),
RedisDbDriver(JsonSerializer(), JsonModel, db),
RedisDbDriver(YamlSerializer(), YamlModel, db),
])
def test_setup():
db = redis.Redis(host='localhost', port=6379, db=0)
Model.set_db_drivers([
RedisDbDriver(PickleSerializer(), User, db),
RedisDbDriver(YamlSerializer(), Project, db)
])
Index.set_db_drivers([RedisIndexDbDriver(UsernameIndex, db)])
def test_setup():
Model.set_db_drivers([
FileDbDriver(PickleSerializer(), User, Path('./data/user')),
FileDbDriver(YamlSerializer(), Project, Path('./data/project'))
])
Index.set_db_drivers([
FileIndexDbDriver(JsonSerializer(), UsernameIndex,
Path('./data/UserNameIndex.yaml'))
])
AnalysisManager = getattr(import_module(settings.ANALYSES_MODULE, package='app'), "AnalysisManager")
Models = [(YamlSerializer(), User), (YamlSerializer(), Project), (JsonSerializer(), Status),
(JsonSerializer(), RunStatus), (JsonSerializer(), Session), (JsonSerializer(), Run),
(JsonSerializer(), Computer)] + [(s(), m) for s, m, p in AnalysisManager.get_db_models()]
Indexes = [ProjectIndex, UserIndex, SessionIndex, RunIndex, ComputerIndex] + [m for s, m, p in
AnalysisManager.get_db_indexes()]
DATA_PATH = settings.DATA_PATH
db = redis.Redis(host='localhost', port=6379, db=0)
if settings.IS_LOCAL_SETUP:
Model.set_db_drivers([FileDbDriver(JsonSerializer(), m, Path(f'{DATA_PATH}/{m.__name__}')) for s, m in Models])
else:
Model.set_db_drivers([RedisDbDriver(s, m, db) for s, m in Models])
if settings.IS_LOCAL_SETUP:
Index.set_db_drivers(
[FileIndexDbDriver(YamlSerializer(), m, Path(f'{DATA_PATH}/{m.__name__}.yaml')) for m in Indexes])
else:
Index.set_db_drivers([RedisIndexDbDriver(m, db) for m in Indexes])
create_project(settings.FLOAT_PROJECT_TOKEN, 'float project')
create_project(settings.SAMPLES_PROJECT_TOKEN, 'samples project')
project.clean_project(settings.FLOAT_PROJECT_TOKEN)
from .user import User, UserIndex
from .status import Status, RunStatus
from .session import Session, SessionIndex
from .run import Run, RunIndex, RunPreferences
from .series_collection import SeriesCollection
from .. import settings
DATA_PATH = settings.DATA_PATH
Model.set_db_drivers([
FileDbDriver(YamlSerializer(), User, Path(f'{DATA_PATH}/User')),
FileDbDriver(YamlSerializer(), Project, Path(f'{DATA_PATH}/Project')),
FileDbDriver(JsonSerializer(), Status, Path(f'{DATA_PATH}/Status')),
FileDbDriver(JsonSerializer(), RunStatus, Path(f'{DATA_PATH}/RunStatus')),
FileDbDriver(JsonSerializer(), Session, Path(f'{DATA_PATH}/Session')),
FileDbDriver(JsonSerializer(), Run, Path(f'{DATA_PATH}/Run')),
FileDbDriver(PickleSerializer(), SeriesCollection,
Path(f'{DATA_PATH}/SeriesCollection')),
FileDbDriver(JsonSerializer(), RunPreferences,
Path(f'{DATA_PATH}/RunPreferences')),
])
Index.set_db_drivers([
FileIndexDbDriver(YamlSerializer(), ProjectIndex,
Path(f'{DATA_PATH}/ProjectIndex.yaml')),
FileIndexDbDriver(YamlSerializer(), UserIndex,
Path(f'{DATA_PATH}/UserIndex.yaml')),
FileIndexDbDriver(YamlSerializer(), SessionIndex,
Path(f'{DATA_PATH}/SessionIndex.yaml')),
FileIndexDbDriver(YamlSerializer(), RunIndex,
Path(f'{DATA_PATH}/RunIndex.yaml')),
from .run import Run, RunIndex
from .computer import Computer, ComputerIndex
from ..analyses import AnalysisManager
from .. import settings
DATA_PATH = settings.DATA_PATH
db = redis.Redis(host='localhost', port=6379, db=0)
Model.set_db_drivers(
[RedisDbDriver(s(), m, db)
for s, m, p in AnalysisManager.get_db_models()] + [
RedisDbDriver(YamlSerializer(), User, db),
RedisDbDriver(YamlSerializer(), Project, db),
RedisDbDriver(JsonSerializer(), Status, db),
RedisDbDriver(JsonSerializer(), RunStatus, db),
RedisDbDriver(JsonSerializer(), Session, db),
RedisDbDriver(JsonSerializer(), Run, db),
RedisDbDriver(JsonSerializer(), Computer, db),
])
Index.set_db_drivers([
RedisIndexDbDriver(m, db) for s, m, p in AnalysisManager.get_db_indexes()
] + [
RedisIndexDbDriver(ProjectIndex, db),
RedisIndexDbDriver(UserIndex, db),
RedisIndexDbDriver(SessionIndex, db),
RedisIndexDbDriver(RunIndex, db),
RedisIndexDbDriver(ComputerIndex, db),
])
def test_setup():
Model.set_db_drivers([
FileDbDriver(PickleSerializer(), PickleModel, Path('./data/pickle')),
FileDbDriver(JsonSerializer(), JsonModel, Path('./data/json')),
FileDbDriver(YamlSerializer(), YamlModel, Path('./data/yaml')),
])