Commit df935d4c authored by Tao feng's avatar Tao feng

Fix sample_data_loader script

parent ad08cdcc
...@@ -22,6 +22,8 @@ DB_FILE = '/tmp/test.db' ...@@ -22,6 +22,8 @@ DB_FILE = '/tmp/test.db'
SQLITE_CONN_STRING = 'sqlite:////tmp/test.db' SQLITE_CONN_STRING = 'sqlite:////tmp/test.db'
Base = declarative_base() Base = declarative_base()
# replace localhost with docker host ip
# todo: get the ip from input argument
NEO4J_ENDPOINT = 'bolt://localhost:7687' NEO4J_ENDPOINT = 'bolt://localhost:7687'
neo4j_endpoint = NEO4J_ENDPOINT neo4j_endpoint = NEO4J_ENDPOINT
...@@ -136,6 +138,8 @@ def create_sample_job(table_name, model_name): ...@@ -136,6 +138,8 @@ def create_sample_job(table_name, model_name):
neo4j_user, neo4j_user,
'publisher.neo4j.{}'.format(neo4j_csv_publisher.NEO4J_PASSWORD): 'publisher.neo4j.{}'.format(neo4j_csv_publisher.NEO4J_PASSWORD):
neo4j_password, neo4j_password,
'publisher.neo4j.{}'.format(neo4j_csv_publisher.JOB_PUBLISH_TAG):
'unique_tag', # should use unique tag here like {ds}
}) })
job = DefaultJob(conf=job_config, job = DefaultJob(conf=job_config,
task=task, task=task,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment