util.py

import mysql.connector
from pymongo import MongoClient
from dotenv import load_dotenv, find_dotenv
import os

# Function to load environment variables
def load_env_vars():
    load_dotenv(find_dotenv())

# Function to initialize MySQL database connection
def init_database(user, password, host, port, database):
    connection = mysql.connector.connect(
        user=user,
        password=password,
        host=host,
        port=port,
        database=database
    )
    return connection

# Function to initialize MongoDB database connection
def init_mongo_database(user, password, host, port, database):
    connection_string = f"mongodb://{user}:{password}@{host}:{port}/{database}"
    client = MongoClient(connection_string)
    return client[database]