revert(ml-waf): ⏪

Remove `ml-waf`  from logger

Author: Anyesh

.env

@@ -1,3 +1,2 @@
 USAGE_LOGGERS_DISABLE=False
 USAGE_LOGGERS_URL="http://localhost:4001/message"
-WAF_ENABLED=True

.isort.cfg

@@ -2,4 +2,4 @@
 line_length = 88
 multi_line_output = 3
 include_trailing_comma = True
-known_third_party = aiohttp,django,joblib,requests,setuptools,werkzeug
+known_third_party = aiohttp,django,requests,setuptools,werkzeug

requirements_dev.txt

@@ -12,4 +12,3 @@ Werkzeug==1.0.1
 django==3.1.12
 aiohttp==3.7.4
 tox==3.24.3
-joblib

setup.py

@@ -10,7 +10,6 @@ def read_file(name):
 
 middleware = util.convert_path("usagelogger/middleware")
 utils = util.convert_path("usagelogger/utils")
-artifacts = util.convert_path("usagelogger/artifacts")
 
 setup(
     name="usagelogger",
@@ -37,17 +36,11 @@ def read_file(name):
         "usagelogger": "usagelogger",
         "usagelogger.middleware": middleware,
         "usagelogger.utils": utils,
-        "usagelogger.artifacts": artifacts,
     },
-    packages=[
-        "usagelogger",
-        "usagelogger.middleware",
-        "usagelogger.utils",
-        "usagelogger.artifacts",
-    ],
+    packages=["usagelogger", "usagelogger.middleware", "usagelogger.utils"],
     # packages=find_packages(exclude=["tests"]),
     python_requires=">=3.7, <4",
-    install_requires=["requests>=2", "joblib>=1.0.1"],
+    install_requires=["requests>=2"],
     include_package_data=True,
     tests_require=["pytest"],
     project_urls={

usagelogger/_waf.py

@@ -17,8 +17,6 @@
 
 enclosure_queue: Queue = Queue()
 
-WAF_ENABLED = bool(os.getenv("WAF_ENABLED", False))
-
 
 class BaseLogger:
     """Basic usage logger to embed or extend."""
@@ -69,12 +67,6 @@ def __init__(
         self._submit_failures_lock = threading.Lock()
         self._submit_successes = 0
         self._submit_successes_lock = threading.Lock()
-        if WAF_ENABLED:
-            from ._waf import WAF
-
-            self.waf = WAF.load_model()
-        else:
-            self.waf = None
 
     def disable(self):
         self._enabled = False
@@ -113,12 +105,6 @@ def __internal_submission(self, q: Queue):
 
             while not q.empty():
                 payload = q.get()
-                # ML WAF
-                if WAF_ENABLED:
-                    proba = self.waf.get_threat_probabilities(
-                        query=payload["msg"][1][1]
-                    )
-                    payload["msg"].append(["threat_score", proba])
                 payload["msg"] = json.dumps(payload["msg"], separators=(",", ":"))
                 if not payload["skip_compression"]:
                     body = payload["msg"]