Merge pull request #3 from SpanishST/clean_statement

Bump version to 0.2.1 and update license to GPL-3.0-or-later; enhance…

Author: SpanishST

pyproject.toml

@@ -4,18 +4,17 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "hueclientrest"
-version = "0.2.0"
+version = "0.2.1"
 authors = [
     {name = "Ramon LOPEZ", email = "ramon.lopez@outlook.fr"},
 ]
 description = "A Python REST client for interacting with Hadoop Hue's REST API"
 readme = "README.md"
-license = {text = "GPL-3.0"}
+license = "GPL-3.0-or-later"
 requires-python = ">=3.9"
 classifiers = [
     "Development Status :: 3 - Alpha",
     "Intended Audience :: Developers",
-    "License :: OSI Approved :: MIT License",
     "Operating System :: OS Independent",
     "Programming Language :: Python :: 3",
     "Programming Language :: Python :: 3.9",

src/hueclientrest/core.py

@@ -1,4 +1,5 @@
 import os
+import re
 import time
 import csv
 import json
@@ -47,7 +48,7 @@ def login(self):
 
         self.session.headers.update({"Authorization": f"Bearer {self.token}"})
 
-    def execute(self, statement: str, dialect: str = "hive") -> str:
+    def execute(self, statement: str, dialect: str = "hive", clean_statement: bool = True) -> str:
         """
         Execute query and get operation ID (history_uuid)
         Raises:
@@ -60,7 +61,14 @@ def execute(self, statement: str, dialect: str = "hive") -> str:
         """
 
         url = f"{self.host}/api/v1/editor/execute/{dialect}"
-        resp = self.session.post(url, data={"statement": statement})
+
+        clean_statement = statement.strip()
+        
+        if clean_statement:
+            clean_statement = re.sub(r'[\x00-\x1F]+', ' ', clean_statement)
+            clean_statement = re.sub(r'\s+', ' ', clean_statement)
+
+        resp = self.session.post(url, data={"statement": clean_statement})
         resp.raise_for_status()
         js = resp.json()