From 526f0876242eeab836fab78487d76c2adb45b7f1 Mon Sep 17 00:00:00 2001 From: surister Date: Thu, 11 Apr 2024 11:38:34 +0200 Subject: [PATCH] Re-write `remove_ending_semicolon` --- connectorx-python/connectorx/__init__.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/connectorx-python/connectorx/__init__.py b/connectorx-python/connectorx/__init__.py index b58590e80..074a29e00 100644 --- a/connectorx-python/connectorx/__init__.py +++ b/connectorx-python/connectorx/__init__.py @@ -16,6 +16,7 @@ from importlib_metadata import version __version__ = version(__name__) + except: pass @@ -388,8 +389,15 @@ def reconstruct_pandas(df_infos: Dict[str, Any]): def remove_ending_semicolon(query: str) -> str: - if query[-1] == ';': - query= list(query) - query.pop(-1) - query = "".join(query) + """ + Removes the semicolon if the query ends with it. + + Parameters + ========== + query + SQL query + + """ + if query.endswith(';'): + query = query[:-1] return query