database.py: FileNotFoundError is py3-only, cope

jolivain · Jun 24, 2018 · 59f3ec9 · 59f3ec9
1 parent f8e3112
commit 59f3ec9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/wormhole_mailbox_server/database.py b/src/wormhole_mailbox_server/database.py
@@ -14,6 +14,10 @@ def get_schema(name, version):
     return schema_bytes.decode("utf-8")
 
 def get_upgrader(name, new_version):
+    try:
+        FileNotFoundError # py3
+    except NameError:
+        FileNotFoundError = EnvironmentError # py2
     try:
         schema_bytes = resource_string("wormhole_mailbox_server",
                                        "db-schemas/upgrade-%s-to-v%d.sql" %