4fd5b1ebc4
Reviewers: mferencevic, ipaljak, mculinovic Reviewed By: mculinovic Subscribers: teon.banek, msantl, pullbot Differential Revision: https://phabricator.memgraph.io/D1395
79 lines
2.2 KiB
Python
79 lines
2.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import database
|
|
import os
|
|
from behave import given
|
|
|
|
|
|
def clear_graph(context):
|
|
database.query("MATCH (n) DETACH DELETE n", context)
|
|
if context.exception is not 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'
|