Procházet zdrojové kódy

pygrass: Add VARCHAR column support on Table

git-svn-id: https://svn.osgeo.org/grass/grass/trunk@65076 15284696-431f-4ddb-bdfa-cd5b030d7da7
Pietro Zambelli před 10 roky
rodič
revize
418ca0356c
1 změnil soubory, kde provedl 4 přidání a 2 odebrání
  1. 4 2
      lib/python/pygrass/vector/table.py

+ 4 - 2
lib/python/pygrass/vector/table.py

@@ -405,8 +405,10 @@ class Columns(object):
             :type col_type: str
             """
             valid_type = ('DOUBLE PRECISION', 'DOUBLE', 'INT', 'INTEGER',
-                          'DATE')
-            if 'VARCHAR' in col_type or col_type.upper() not in valid_type:
+                          'DATE', 'VARCHAR')
+            col = col_type.upper()
+            valid = [col.startswith(tp) for tp in valid_type]
+            if not any(valid):
                 str_err = ("Type: %r is not supported."
                            "\nSupported types are: %s")
                 raise TypeError(str_err % (col_type, ", ".join(valid_type)))