import re
from arango import ArangoClient
from dotenv import load_dotenv
import os
import env_manager

load_dotenv() # Install with pip install python-dotenv
class ArangoDB:
    def __init__(self):
        """
        Initializes an instance of the ArangoEVClass.

        Args:
            db_name (str): The name of the database.
            username (str): The username for authentication.
            password (str): The password for authentication.
        """
        password = os.getenv("PASSWORD_ARANGO")
        self.client = ArangoClient(hosts='https://arango.lasseedfast.se')
        self.db = self.client.db('ev_dataharvest', username='dataharvest', password=password)


    def all_ev_speeches(self):
            """
            Retrieves all EV speeches from the 'ev_speeches' collection.

            Returns:
                A cursor object containing all EV speeches.
            """
            return self.db.collection('ev_speeches').all()

    def update_ev_document(self, document):
        """
        Updates an EV document in the 'ev_speeches' collection.

        Args:
            document: The document to be updated.

        Returns:
            None
        """
        self.db.collection('ev_speeches').update(document, merge=False)

    def get_document_by_id(self, document_id):
            """
            Retrieves a document from the 'ev_speeches' collection by its ID.

            Args:
                document_id (str): The ID of the document to retrieve.

            Returns:
                dict: The retrieved document.

            """
            if '/' in document_id:
                document_id = document_id.split('/')[-1]
            return self.db.collection('ev_speeches').get(document_id)

    def fix_key(self, _key):

        return re.sub(r'[^A-Za-z0-9_\-\.@()+=;$!*\'%:]', '_', _key)


if __name__ == "__main__":
    arango = ArangoDB()
    test = [i for i in arango.db.aql.execute('FOR doc IN sci_articles LIMIT 1 RETURN doc._key')]
    print(test)