diff --git a/arango_ev_class.py b/arango_ev_class.py deleted file mode 100644 index bdd00f5..0000000 --- a/arango_ev_class.py +++ /dev/null @@ -1,70 +0,0 @@ -from arango import ArangoClient -from dotenv import load_dotenv -import os - -# Load environment variables from .env file -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 speeches from the 'ev_speeches' collection. - - Returns: - A list of all speeches in the 'ev_speeches' collection. - """ - return list(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) - - -if __name__ == "__main__": - arango = ArangoDB() - - # Example usage - speeches = arango.all_ev_speeches() - print(type(speeches)) - for speech in speeches: - print(speech['_key']) # Print the id of the speech - print(speech.keys()) # Print the keys of the speech - exit() \ No newline at end of file diff --git a/example_fetch_docs.py b/example_fetch_docs.py index 2dc44cf..8c63a9d 100644 --- a/example_fetch_docs.py +++ b/example_fetch_docs.py @@ -1,6 +1,6 @@ # Example of fetching documents and go though them one by one using Ollama: -from arango_ev_class import ArangoDB +from arango_class import ArangoDB import ollama