fixed missing cgi.escape() when running under python3.8

nccgroup · Jan 27, 2020 · e72d304 · e72d304
1 parent bc24ab7
commit e72d304
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/houseStatic.py b/houseStatic.py
@@ -31,10 +31,14 @@
 # import IPython
 import jinja2
 import os
-import cgi
 from flask import render_template, request
 import argparse
 
+if sys.version_info >= (3, 8):
+    import html as cgi
+else:
+    import cgi
+
 Error = colored.fg("red") + colored.attr("bold")
 Info = colored.fg("green") + colored.attr("bold")
 MightBeImportant = colored.fg("blue") + colored.attr("bold")
@@ -46,4 +50,4 @@
 class dynamicHookOption(Enum):
     only_new = "new"
     only_existing = "existing"
-    both = "all"
+    both = "all"