File size: 613 Bytes
e84d389
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from pymongo import MongoClient
from ..config import settings

class MongoDB:
    def __init__(self):
        self.client = MongoClient(settings.CONNECTION_STRING)
        self.db = self.client[settings.DATABASE_NAME]
        self.users = self.db["users"]
        self.videos = self.db[settings.COLLECTION_NAME]
        # Indexes
        self.users.create_index("username", unique=True)
        self.users.create_index("email", unique=True)
        self.videos.create_index("video_id", unique=True)
        self.videos.create_index("user_id")

    def close(self):
        self.client.close()

mongodb = MongoDB()