From 81c5837b90322630524a62eacec8cf90ac85ab7a Mon Sep 17 00:00:00 2001
From: Daniel <d.hornung@indiscale.com>
Date: Fri, 9 Feb 2024 09:46:39 +0100
Subject: [PATCH] MAINT: Cleaned up the code a bit more.
---
src/linkahead/utils/get_entity.py | 4 ++--
unittests/test_utils.py | 3 +--
2 files changed, 3 insertions(+), 4 deletions(-)
diff --git a/src/linkahead/utils/get_entity.py b/src/linkahead/utils/get_entity.py
index 9fee4582..282f7c86 100644
--- a/src/linkahead/utils/get_entity.py
+++ b/src/linkahead/utils/get_entity.py
@@ -24,7 +24,7 @@
from typing import Union
from ..common.models import Entity, execute_query
-from .escape import escape_quoted_text
+from .escape import escape_squoted_text
def get_entity_by_name(name: str) -> Entity:
@@ -32,7 +32,7 @@ def get_entity_by_name(name: str) -> Entity:
Submits the query "FIND ENTITY WITH name='{name}'".
"""
- name = escape_quoted_text(name)
+ name = escape_squoted_text(name)
return execute_query(f"FIND ENTITY WITH name='{name}'", unique=True)
diff --git a/unittests/test_utils.py b/unittests/test_utils.py
index 9cfa50f7..e7495a91 100644
--- a/unittests/test_utils.py
+++ b/unittests/test_utils.py
@@ -25,8 +25,7 @@
from __future__ import unicode_literals
from linkahead.common.utils import xml2str
-from linkahead.utils.escape import (escape_dquoted_text, escape_quoted_text,
- escape_squoted_text)
+from linkahead.utils.escape import (escape_dquoted_text, escape_squoted_text)
from lxml.etree import Element
--
GitLab