Skip to content

Commit 1e4b570

Browse files
committed
Fix Python script running Cypher queries
1 parent b240abe commit 1e4b570

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

cypher/bi.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -32,11 +32,12 @@ def convert_to_date(timestamp):
3232
driver = GraphDatabase.driver("bolt://localhost:7687")
3333

3434
with driver.session() as session:
35-
for query_id in ["1", "2", "3", "4", "5", "6", "7", "8", "9", "10", "11", "12", "13", "14a", "14b", "15", "16", "17", "18", "19", "20"]:
36-
query_file = open(f'queries/bi-{query_id}.cypher', 'r')
35+
for query_variant in ["1", "2", "3", "4", "5", "6", "7", "8", "9", "10", "11", "12", "13", "14a", "14b", "15", "16", "17", "18", "19", "20"]:
36+
query_num = re.sub("[^0-9]", "", query_variant)
37+
query_file = open(f'queries/bi-{query_num}.cypher', 'r')
3738
query_spec = query_file.read()
3839

39-
parameters_csv = csv.DictReader(open(f'parameters/bi-{query_id}.txt'), delimiter='|')
40+
parameters_csv = csv.DictReader(open(f'parameters/bi-{query_variant}.txt'), delimiter='|')
4041

4142
for query_parameters in parameters_csv:
4243
# convert fields based on type designators
@@ -47,6 +48,6 @@ def convert_to_date(timestamp):
4748
# drop type designators
4849
type_pattern = re.compile(':.*')
4950
query_parameters = {type_pattern.sub('', k): v for k, v in query_parameters.items()}
50-
run_query(session, query_id, query_spec, query_parameters)
51+
run_query(session, query_variant, query_spec, query_parameters)
5152

5253
driver.close()

0 commit comments

Comments
 (0)