import pymongo
import os
from config import DATABASE, COLLECTION
from logger import get_logger

logger = get_logger()

class DBWrite:
    """
    Inserts processed news into MongoDB
    """
    def __init__(self):
        self.url = os.getenv('DB_URL')
        self.database = DATABASE
        self.collection = COLLECTION
        self.__client = None
        self.__error = 0

    def __connect(self):
        try:
            self.__client = pymongo.MongoClient(self.url)
            _ = self.__client.list_database_names()
        except Exception as conn_exception:
            self.__error = 1
            self.__close_connection()
            self.__client = None
            raise

    def __insert(self, documents):
        try:

            db = self.__client[self.database]
            coll = db[self.collection]
            coll.drop()
            coll.insert_many(documents=documents)
        except Exception as insert_err:
            self.__error = 1
            self.__close_connection()
            raise

    def __close_connection(self):
        if self.__client is not None:
            self.__client.close()
            self.__client = None

    def insert_news_into_db(self, documents: list):
        logger.warning('Entering insert_news_into_db()')
        if self.url is not None:
            if self.__error == 0:
                self.__connect()
            if self.__error == 0:
                self.__insert(documents=documents)
            if self.__error == 0:
                print("Insertion Successful")
            if self.__client is not None:
                self.__close_connection()
        logger.warning('Exiting insert_news_into_db()')