Update for Python 3.9

otter/otter.py

@@ -18,7 +18,7 @@ class Otter():
     reports for long-running or complex jobs where and iPython
     notebook would be an impractical way of presenting information.
     """
-    def __init__(self, filename, config_file=None, **kwargs):
+    def __init__(self, filename, theme_location=None, config_file=None, **kwargs):
         """
         An Otter report is created by this class.
 
@@ -50,15 +50,25 @@ def __init__(self, filename, config_file=None, **kwargs):
                 self.meta[option] = config.get('meta', option)
             for option in kwargs.items():
                 self.meta[option[0]] = option[1]
-
-        try:
-            theme = config.get("theme", "location")
-        except:
-            print("Cannot find theme in the config file. Using the default theme.")
+
+        if theme_location:
+            theme = theme_location
+
+        elif config.has_option("theme", "name"):
+            try:
+                import importlib
+                theme = importlib.import_module(config.get("theme", "name"))
+            except:
+                pass
+        else:
             try:
-                theme = resource_filename(__name__, "themes/default/")
+                theme = config.get("theme", "location")
             except:
-                print("No theme files found.")
+                print("Cannot find theme in the config file. Using the default theme.")
+                try:
+                    theme = resource_filename(__name__, "themes/default/")
+                except:
+                    print("No theme files found.")
 
         self.env = Environment(loader=FileSystemLoader(theme))
 

otter/plot.py

@@ -19,7 +19,7 @@ def __init__(self, figure, filename=None, dpi=300):
         if filename==None:
             figure.savefig(self.image, dpi=dpi)
 
-        self.bitstring = base64.encodestring(self.image.getvalue())
+        self.bitstring = base64.encodebytes(self.image.getvalue())
 
     def __repr__(self):
         html_str= r"""<img src="data:image/png;base64,{}" style="max-width: 100%;" class="img-responsive"/>""".format(self.bitstring.decode("utf-8"))

setup.py

@@ -31,7 +31,7 @@
     description="The simple HTML report generator for Python jobs.",
     long_description=readme + '\n\n' + history,
     author="Daniel Williams",
-    author_email='daniel.williams@gla.ac.uk',
+    author_email='daniel.williams@glasgow.ac.uk',
     url='https://github.com/transientlunatic/otter',
     packages=[
         'otter',