You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
70 lines
2.0 KiB
70 lines
2.0 KiB
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() |