Clean up in connect library. PyPI fixes.

ableinc · Aug 30, 2020 · 172ee2b · 172ee2b
1 parent e095e35
commit 172ee2b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/mongrations/ClassType.py b/mongrations/ClassType.py
@@ -1,6 +1,9 @@
 from pymongo.database import Database
 from pymysql.connections import Connection
-from psycopg2.extensions import cursor
+try:
+    from psycopg2.extensions import cursor
+except ImportError:
+    cursor = None
 from os import environ
 
 db_type = {

diff --git a/mongrations/connect.py b/mongrations/connect.py
@@ -1,9 +1,9 @@
 from os import environ
+from pymongo import MongoClient
+import motor.motor_asyncio as motor
 try:
     import pymysql.cursors
     import psycopg2
-    from pymongo import MongoClient
-    import motor.motor_asyncio as motor
 except ImportError:
     pass
 

diff --git a/mongrations/version.py b/mongrations/version.py
@@ -1 +1 @@
-__version__ = '1.0.1'
+__version__ = '1.0.2'