31 lines
831 B
Python
31 lines
831 B
Python
import os
|
|
import motor
|
|
import motor.motor_asyncio
|
|
import pymongo
|
|
|
|
|
|
MONGO_HOST = os.getenv('MONGO_HOST', 'localhost')
|
|
MONGO_PASSWORD = os.getenv('MONGO_PASSWORD', 'password')
|
|
|
|
CONNECTION_STRING = f'mongodb://mongo:{MONGO_PASSWORD}@{MONGO_HOST}:27017/'
|
|
|
|
|
|
database: 'motor.MotorDatabase' = motor.motor_asyncio.AsyncIOMotorClient(CONNECTION_STRING).configurator
|
|
|
|
def create_indexes():
|
|
client = pymongo.MongoClient(CONNECTION_STRING)
|
|
database = client.get_database('configurator')
|
|
database.get_collection('configs').create_index([
|
|
('stage', 1),
|
|
('project', 1),
|
|
('name', 1)
|
|
])
|
|
database.get_collection('experiments').create_index([
|
|
('stage', 1),
|
|
('project', 1),
|
|
('name', 1)
|
|
])
|
|
database.get_collection('staff').create_index([
|
|
('platform_id', 1),
|
|
])
|