apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-17 09:11:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add database override to bolt output, default is neo4j

Browse source
This commit is contained in:
Cyber MacGeddon 2024-11-28 19:18:03 +00:00
parent b2f7b34529
commit c1ec3d6abb
4 changed files with 166 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
trustgraph-flow/trustgraph/query/triples/neo4j/service.py
View file

@ -21,6 +21,7 @@ default_subscriber = module
default_graph_host = 'bolt://neo4j:7687'
default_username = 'neo4j'
default_password = 'password'
default_database = 'neo4j'
class Processor(ConsumerProducer):
@ -31,7 +32,8 @@ class Processor(ConsumerProducer):
subscriber = params.get("subscriber", default_subscriber)
graph_host = params.get("graph_host", default_graph_host)
username = params.get("username", default_username)
password = params.get("passowrd", default_password)
password = params.get("password", default_password)
database = params.get("database", default_database)
super(Processor, self).__init__(
**params | {
@ -44,7 +46,7 @@ class Processor(ConsumerProducer):
}
)
self.db = "neo4j"
self.db = database
self.io = GraphDatabase.driver(graph_host, auth=(username, password))
@ -342,6 +344,12 @@ class Processor(ConsumerProducer):
help=f'Neo4j password (default: {default_password})'
)
parser.add_argument(
'--database',
default=default_database,
help=f'Neo4j database (default: {default_database})'
)
def run():
Processor.start(module, __doc__)

12
trustgraph-flow/trustgraph/storage/triples/neo4j/write.py
View file

@ -24,6 +24,7 @@ default_subscriber = module
default_graph_host = 'bolt://neo4j:7687'
default_username = 'neo4j'
default_password = 'password'
default_database = 'neo4j'
class Processor(Consumer):
@ -33,7 +34,8 @@ class Processor(Consumer):
subscriber = params.get("subscriber", default_subscriber)
graph_host = params.get("graph_host", default_graph_host)
username = params.get("username", default_username)
password = params.get("passowrd", default_password)
password = params.get("password", default_password)
database = params.get("database", default_database)
super(Processor, self).__init__(
**params | {
@ -44,7 +46,7 @@ class Processor(Consumer):
}
)
self.db = "neo4j"
self.db = database
self.io = GraphDatabase.driver(graph_host, auth=(username, password))
@ -152,6 +154,12 @@ class Processor(Consumer):
help=f'Neo4j password (default: {default_password})'
)
parser.add_argument(
'--database',
default=default_database,
help=f'Neo4j database (default: {default_database})'
)
def run():
Processor.start(module, __doc__)