import os
from dotenv import load_dotenv
import psycopg2
import mysql.connector as sql

load_dotenv()

class DB:
    def __init__(self):
        self.string = os.getenv("Connection_String")
        self.collection = os.getenv("Collection_Name")
        self.host = os.getenv("Host")
        self.port = os.getenv("Port")
        self.user = os.getenv("User")
        self.password = os.getenv("Password")
        self.db = os.getenv("DB_Name")
    
    def connect_PS(self):
        self.connection_PS = psycopg2.connect(self.string)
        self.cursor_PS = self.connection_PS.cursor()
    
    def connect_MS(self):
        self.connection_MS = sql.connect(
            host = self.host,
            port = self.port,
            user = self.user,
            password = self.password,
            database = self.db
        )
        self.cursor_MS = self.connection_MS.cursor()

    def insert(self, data):
        insert_query = "INSERT INTO user(n,email_id,d,t,pasword,score) VALUES (%s,%s,%s,%s,%s,%s)"
        self.cursor_MS.execute(insert_query, data)


    def clear(self):
        self.cursor_PS.execute(f"DELETE FROM {self.collection}")
        self.connection_PS.commit()
    
    def close_PS(self):
        self.cursor_PS.close()
        self.connection_PS.close()
    
    def close_MS(self):
        self.cursor_MS.close()
        self.connection_MS.close()