fix pg numeric conversion regression

chembl · Feb 3, 2022 · 06e2de0 · 06e2de0
1 parent 0d815db
commit 06e2de0
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/cbl_migrator/conv.py b/cbl_migrator/conv.py
@@ -42,13 +42,14 @@ def ora2pg(col):
     Used in ChEMBL dump generation
     """
     if isinstance(col.type, Numeric):
-        if not col.type.precision or col.type.precision > 4:
-            col.type = col.type.adapt(BigInteger)
-        else:
-            if col.type.precision <= 2:
-                col.type = col.type.adapt(SmallInteger)
-            elif 2 < col.type.precision <= 4:
-                col.type = col.type.adapt(Integer)
+        if col.type.scale == 0:
+            if not col.type.precision or col.type.precision > 4:
+                col.type = col.type.adapt(BigInteger)
+            else:
+                if col.type.precision <= 2:
+                    col.type = col.type.adapt(SmallInteger)
+                elif 2 < col.type.precision <= 4:
+                    col.type = col.type.adapt(Integer)
     return col