mirror of
https://github.com/fuckpiracyshield/data-storage.git
synced 2024-05-20 14:06:18 +02:00
28 lines
728 B
Python
28 lines
728 B
Python
from piracyshield_data_storage.database.arangodb.connection import DatabaseArangodbConnection
|
|
|
|
from arango.exceptions import AQLQueryExecuteError
|
|
|
|
class DatabaseArangodbDocument(DatabaseArangodbConnection):
|
|
|
|
def collection(self, collection):
|
|
try:
|
|
return self.instance.collection(collection)
|
|
|
|
except:
|
|
raise DatabaseArangodbCollectionNotFoundException()
|
|
|
|
def query(self, aql, **kwargs):
|
|
try:
|
|
return self.instance.aql.execute(aql, **kwargs)
|
|
|
|
except AQLQueryExecuteError:
|
|
raise DatabaseArangodbQueryException()
|
|
|
|
class DatabaseArangodbCollectionNotFoundException(Exception):
|
|
|
|
pass
|
|
|
|
class DatabaseArangodbQueryException(Exception):
|
|
|
|
pass
|