FEDOT.Web/init_db.py at master · aimclub/FEDOT.Web

Repository navigation

import os

import pymongo

from dotenv import load_dotenv

from app import load_datasets_from_file_system

from app.singletons.db_service import DBServiceSingleton

from init.init_cases import create_default_cases

from init.init_history import create_default_history

from init.init_pipelines import create_default_pipelines

if __name__ == "__main__":

load_dotenv("mongo_conn_string.env")

env = os.getenv('MONGO_CONN_STRING')

print("Connection string:", env)

client = pymongo.MongoClient(env)

print("Database names:", client.list_database_names())

db = client.get_default_database()

print("Default database name:", db.name)

load_datasets_from_file_system()

DBServiceSingleton(db)

create_default_cases()

create_default_pipelines()

create_default_history([5, 3, 3])