runabot/RUNA/Database/Mongo/__init__.py
tmddn3070 6693186d40 FEAT: Inital commit
This Commit Contained Rebrand Patches
2024-03-17 09:42:36 +09:00

19 lines
596 B
Python

import os
from motor.motor_asyncio import AsyncIOMotorClient
from dotenv import load_dotenv, find_dotenv
load_dotenv(find_dotenv())
class MongoDBClient:
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.client = \
AsyncIOMotorClient(os.getenv("MONGODB_HOST"), int(os.getenv("MONGODB_PORT")),
username=os.getenv("MONGODB_ID"),
password=os.getenv("MONGODB_PASSWORD"))[os.getenv("MONGODB_DATABASE")]
async def connect(self):
return self.client