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.
68 lines
2.0 KiB
68 lines
2.0 KiB
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) |
|
|