Arango class to interact with the ev_speeches collection

arango_ev_class.py

@@ -0,0 +1,52 @@
+from arango import ArangoClient
+
+class ArangoDB:
+    class ArangoEVClass:
+        def __init__(self, db_name, username, password):
+            """
+            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.
+            """
+            self.client = ArangoClient(hosts='https://arango.lasseedfast.se')
+            self.db = self.client.db(db_name, username=username, 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)