Update examples

GordonZhang2024 · May 1, 2024 · cd99555 · cd99555
1 parent 957aa15
commit cd99555
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 7 deletions.
diff --git a/examples/example.ipynb b/examples/example.ipynb
@@ -2,15 +2,15 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 2,
-   "id": "5f41eab9-0fc3-43eb-add1-18f9ad56dc14",
+   "execution_count": 1,
+   "id": "ca6cd635-e99a-499d-bca9-53e60c87fe4c",
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "&ampltscript&ampgtalert(&#34&#59Bad.&#34&#59)&#59&amplt/script&ampgt\n"
+      "&lt;script&gt;alert(&#34;Bad.&#34;)&#59;&lt;&#47;script&gt;\n"
      ]
     }
    ],

diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "xss-shield"
-version = "0.0.3"
+version = "0.1.1"
 authors = [
   { name="Gordon Zhang", email="jp20171211@163.com" },
 ]

diff --git a/src/xss_shield/__init__.py b/src/xss_shield/__init__.py
@@ -1,11 +1,10 @@
 def escape(s: str) -> str:
-    s = s.replace('<', '&lt;')\
+    s = s.replace(';', '&#59;')\
+         .replace('<', '&lt;')\
          .replace('>', '&gt;')\
-         .replace('&', '&amp;')\
          .replace(' ', '-')\
          .replace("'", "&#39;")\
          .replace('"', "&#34;")\
-         .replace(';', '&#59;')\
          .replace('/', '&#47;')\
 
     return s