from datetime import datetime
from pymongo.mongo_client import MongoClient
from pymongo.server_api import ServerApi
from . import settings

client: MongoClient = MongoClient(settings.DB_URI, server_api=ServerApi('1'))
class __DB:
    personalities = client.ChatDB.personalityCores
DB: __DB = __DB()


def get (asistente):
    found = DB.personalities.find_one({"name":asistente})
    return found["prompt"].replace(
        "{date}", datetime.now().strftime("%Y-%m-%D")
    )