Source code for src.dackar.knowledge_graph.graph_utils

import os
import re

[docs] def search_phrase(text, phrase): """Search phrase in text Args: text (str): text string phrase (str): phase string Returns: bool: True if phase in text else False """ # Compile a regular expression pattern for the specific phrase pattern = re.compile(re.escape(phrase)) # Search for the pattern in the text match = pattern.search(text) # Check if a match was found if match: return True else: return False
[docs] def set_neo4j_import_folder(config_file_path, import_folder_path): """Set neo4j import folder Args: config_file_path (str): location for Neo4j config file import_folder_path (str): location for user provided import folder """ # Ensure the import directory exists if not os.path.exists(import_folder_path): os.makedirs(import_folder_path) with open(config_file_path, 'r') as file: lines = file.readlines() with open(config_file_path, 'w') as file: # found_dbms = False found_server = False for line in lines: # if line.startswith('dbms.directories.import='): # file.write(f'dbms.directories.import={import_folder_path}\n') # found_dbms = True if line.startswith('server.directories.import='): file.write(f'server.directories.import={import_folder_path}\n') found_server = True else: file.write(line) # if not found_dbms: # file.write('\n') # file.write(f'dbms.directories.import={import_folder_path}\n') if not found_server: file.write('\n') file.write(f'server.directories.import={import_folder_path}\n')