ERP5Type.patches.sqltest: Add support for type="datetime".

product/ERP5Type/patches/sqltest.py

@@ -11,6 +11,7 @@
 #
 ##############################################################################
 from Shared.DC.ZRDB.sqltest import *
+from Shared.DC.ZRDB import sqltest
 
 list_type_list = list, tuple, set, frozenset, dict
 
@@ -61,6 +62,16 @@ if 1: # For easy diff with original
                 except ValueError:
                     raise ValueError, (
                         'Invalid floating-point value for <em>%s</em>' % name)
+            elif t.startswith('datetime'):
+                # For subsecond precision, use 'datetime(N)' MySQL type,
+                # where N is the number of digits after the decimal point.
+                n = 0 if t == 'datetime' else int(t[9])
+                try:
+                    v = (v if isinstance(v, DateTime) else DateTime(v)).toZone('UTC')
+                    return "'%s%s'" % (v.ISO(),
+                        ('.%06u' % (v.micros() % 1000000))[:1+n] if n else '')
+                except Exception:
+                    t = 'datetime'
 
             else:
                 if not isinstance(v, (str, unicode)):
@@ -94,3 +105,5 @@ if 1: # For easy diff with original
                 return "%s in (%s)" % (self.column, vs)
         return "%s %s %s" % (self.column, self.op, vs[0])
     SQLTest.render = SQLTest.__call__ = render
+
+sqltest.valid_type = (('int', 'float', 'string', 'nb', 'datetime') + tuple('datetime(%s)' % x for x in xrange(7))).__contains__