Fix reprounzip-qt on Python 3

VIDA-NYU · Jul 2, 2021 · 644fb46 · 644fb46
1 parent d9851b5
commit 644fb46
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/reprounzip-qt/reprounzip_qt/qt_terminal.py b/reprounzip-qt/reprounzip_qt/qt_terminal.py
@@ -4,13 +4,18 @@
 
 from __future__ import division, print_function, unicode_literals
 
-import cgi
 import locale
 import logging
 from qtpy import QtCore, QtWidgets
 import sys
 
 
+if sys.version >= (3,):
+    from html import escape
+else:
+    from cgi import escape
+
+
 logger = logging.getLogger('reprounzip_qt')
 
 
@@ -82,7 +87,7 @@ def __init__(self, cmdline, env, input_enabled=False,
 </style>
 ''')
         self.text.append('<span style="color: blue;">%s</span>' %
-                         cgi.escape(' '.join(cmdline)))
+                         escape(' '.join(cmdline)))
 
     def _enter(self):
         cmd = self.input.text()
@@ -93,13 +98,13 @@ def _read_stdout(self):
         out = self.process.readAllStandardOutput()
         out = bytes(out).decode(locale.getpreferredencoding() or 'UTF-8',
                                 'replace')
-        self.text.append('<span>%s</span>' % cgi.escape(out))
+        self.text.append('<span>%s</span>' % escape(out))
 
     def _read_stderr(self):
         out = self.process.readAllStandardError()
         out = bytes(out).decode(locale.getpreferredencoding() or 'UTF-8',
                                 'replace')
-        self.text.append('<span class="err">%s</span>' % cgi.escape(out))
+        self.text.append('<span class="err">%s</span>' % escape(out))
 
     def _finished(self, code, status):
         good = False