@@ -67,6 +67,12 @@ def batch_upload(
67
67
total_nodes_created = 0
68
68
total_relationships_created = 0
69
69
total_properties_set = 0
70
+ neo4j_creds = (cdata .neo4j_uri , cdata .neo4j_user , cdata .neo4j_password )
71
+ neo4j_database = cdata .neo4j_database
72
+
73
+ # If overwrite enabled, clear db
74
+ if cdata .overwrite is True :
75
+ reset (neo4j_creds , neo4j_database )
70
76
71
77
# Get list of tuples containing queries and accompanying params for driver execution
72
78
query_params = specification_queries (gdata .nodes , cdata )
@@ -75,10 +81,10 @@ def batch_upload(
75
81
for qp in query_params :
76
82
# Run queries and retrieve summary of upload
77
83
summary = upload_query (
78
- creds = ( cdata . neo4j_uri , cdata . neo4j_user , cdata . neo4j_password ) ,
84
+ creds = neo4j_creds ,
79
85
query = qp [0 ],
80
86
params = qp [1 ],
81
- database = cdata . neo4j_database
87
+ database = neo4j_database
82
88
)
83
89
84
90
# Sample summary result
@@ -126,7 +132,7 @@ def upload(
126
132
should_overwrite : bool = False ,
127
133
database_name : str = 'neo4j' ,
128
134
max_batch_size : int = 500 ,
129
- )-> ( float , int , int , int ) :
135
+ )-> UploadResult :
130
136
"""
131
137
Uploads a dictionary of simple node and relationship records to a target Neo4j instance specified in the arguments.
132
138
0 commit comments