be9fe9fe75
Summary:
Parsing query with string values wasn't correct. It didn't matter
with which type of quote mark string value started. It simply took
any possible quote mark which wasn't escaped as possible beginning
or end for string value.
Following example didn't work:
{name: "Chris Anderson: Technology's long tail"}
Detected string value was "Chris Anderson: Technology' and because
of that query broke. Now, string is detected by memorizing which
quote mark was used to start parsing string value and is required
to end string value with that quote mark.
Reviewers: teon.banek, msantl
Reviewed By: teon.banek
Subscribers: pullbot
Differential Revision: https://phabricator.memgraph.io/D1029
76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
import database
|
|
import os
|
|
from behave import *
|
|
|
|
|
|
def clear_graph(context):
|
|
database.query("MATCH (n) DETACH DELETE n", context)
|
|
if context.exception != None:
|
|
context.exception = None
|
|
database.query("MATCH (n) DETACH DELETE n", context)
|
|
|
|
|
|
@given('an empty graph')
|
|
def empty_graph_step(context):
|
|
clear_graph(context)
|
|
context.graph_properties.set_beginning_parameters()
|
|
|
|
|
|
@given('any graph')
|
|
def any_graph_step(context):
|
|
clear_graph(context)
|
|
context.graph_properties.set_beginning_parameters()
|
|
|
|
|
|
@given('graph "{name}"')
|
|
def graph_name(context, name):
|
|
create_graph(name, context)
|
|
|
|
|
|
def create_graph(name, context):
|
|
"""
|
|
Function deletes everything from database and creates a new
|
|
graph. Graph file name is an argument of function. Function
|
|
executes queries written in a .cypher file separated by ';'
|
|
and sets graph properties to beginning values.
|
|
"""
|
|
clear_graph(context)
|
|
path = find_graph_path(name, context.config.root)
|
|
|
|
q_marks = ["'", '"', '`']
|
|
|
|
with open(path, 'r') as f:
|
|
content = f.read().replace('\n', ' ')
|
|
single_query = ''
|
|
quote = None
|
|
i = 0
|
|
while i < len(content):
|
|
ch = content[i]
|
|
if ch == '\\' and i != len(content) - 1 and content[i + 1] in q_marks:
|
|
single_query += ch + content[i + 1]
|
|
i += 2
|
|
else:
|
|
single_query += ch
|
|
if quote == ch:
|
|
quote = None
|
|
elif ch in q_marks and quote is None:
|
|
quote = ch
|
|
if ch == ';' and quote is None:
|
|
database.query(single_query, context)
|
|
single_query = ''
|
|
i += 1
|
|
if single_query.strip() != '':
|
|
database.query(single_query, context)
|
|
context.graph_properties.set_beginning_parameters()
|
|
|
|
|
|
def find_graph_path(name, path):
|
|
"""
|
|
Function returns path to .cypher file with given name in
|
|
given folder or subfolders. Argument path is path to a given
|
|
folder.
|
|
"""
|
|
for root, dirs, files in os.walk(path):
|
|
if name + '.cypher' in files:
|
|
return root + '/' + name + '.cypher'
|