py3: Update Shared.DC.ZRDB.{sqltest,sqlvar} monkey patches.
This supports both python2 and python3. For python3, this decode() properly arguments returned by string_literal() to have str(), otherwise the query string is an str() (DocumentTemplate) and the arguments are bytes(). In other places, bytes() is used directly but in this case this is not needed and would require monkey patching (at least) DocumentTemplate.
Showing
| ... | ... | @@ -14,84 +14,99 @@ |
| # dtml-sqlvar patch to convert None to NULL, and deal with DateTime | ||
| from Shared.DC.ZRDB.sqlvar import SQLVar | ||
| from Shared.DC.ZRDB.sqlvar import * | ||
| from Shared.DC.ZRDB import sqlvar | ||
| from string import atoi,atof | ||
| from DateTime import DateTime | ||
| def SQLVar_render(self, md): | ||
| args=self.args | ||
| t=args['type'] | ||
| try: | ||
| expr=self.expr | ||
| if type(expr) is str: v=md[expr] | ||
| else: v=expr(md) | ||
| except Exception: | ||
| if args.get('optional'): | ||
| if 1: # For easy diff with original (ZSQLMethods 3.14) | ||
| def render(self, md): | ||
| name = self.__name__ | ||
| args = self.args | ||
| t = args['type'] | ||
| try: | ||
| expr = self.expr | ||
| if isinstance(expr, StringTypes): | ||
| v = md[expr] | ||
| else: | ||
| v = expr(md) | ||
| except Exception: | ||
| if 'optional' in args and args['optional']: | ||
| return 'null' | ||
| if not isinstance(expr, StringTypes): | ||
| raise | ||
| raise ValueError('Missing input variable, <em>%s</em>' % name) | ||
| if v is None: | ||
| return 'null' | ||
|
||
| if type(expr) is not str: | ||
| raise | ||
| raise ValueError('Missing input variable, <em>%s</em>' % self.__name__) | ||
| if v is None and args.get('optional'): | ||
| return 'null' | ||
| if t == 'int': | ||
| try: | ||
| if isinstance(v, StringTypes): | ||
| if v[-1:] == 'L': | ||
| v = v[:-1] | ||
| int(v) | ||
| else: | ||
| v = str(int(v)) | ||
| except Exception: | ||
| if 'optional' in args and args['optional']: | ||
| return 'null' | ||
| err = 'Invalid integer value for <em>%s</em>' % name | ||
| raise ValueError(err) | ||
| elif t == 'float': | ||
| try: | ||
| if isinstance(v, StringTypes): | ||
| if v[-1:] == 'L': | ||
| v = v[:-1] | ||
| float(v) | ||
| else: | ||
| # ERP5 patch: We use repr that have better precision than str for | ||
| # floats (on python2 only) | ||
| v = repr(float(v)) | ||
| except Exception: | ||
| if 'optional' in args and args['optional']: | ||
| return 'null' | ||
| err = 'Invalid floating-point value for <em>%s</em>' % name | ||
| raise ValueError(err) | ||
| if t=='int': | ||
| try: | ||
| if type(v) is str: | ||
| if v[-1:]=='L': | ||
| v=v[:-1] | ||
| atoi(v) | ||
| return v | ||
| return str(int(v)) | ||
| except Exception: | ||
| t = 'integer' | ||
| elif t=='float': | ||
| try: | ||
| if type(v) is str: | ||
| if v[-1:]=='L': | ||
| v=v[:-1] | ||
| atof(v) | ||
| return v | ||
| # ERP5 patch, we use repr that have better precision than str for | ||
| # floats | ||
| return repr(float(v)) | ||
| except Exception: | ||
| t = 'floating-point' | ||
| 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' | ||
| elif t=='nb' and not v: | ||
| t = 'empty string' | ||
| else: | ||
| v = md.getitem('sql_quote__',0)( | ||
| v if isinstance(v, basestring) else str(v)) | ||
| #if find(v,"\'") >= 0: v=join(split(v,"\'"),"''") | ||
| #v="'%s'" % v | ||
| return v | ||
| 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' | ||
| elif t=='nb' and not v: | ||
| t = 'empty string' | ||
| if args.get('optional'): | ||
| return 'null' | ||
| raise ValueError('Invalid %s value for <em>%s</em>: %r' | ||
| % (t, self.__name__, v)) | ||
| else: | ||
| if not isinstance(v, (str, StringTypes)): | ||
| v = str(v) | ||
| if t == 'nb': | ||
| if 'optional' in args and args['optional']: | ||
| return 'null' | ||
| else: | ||
| err = 'Invalid empty string value for <em>%s</em>' % name | ||
| raise ValueError(err) | ||
| v = md.getitem('sql_quote__', 0)(v) | ||
| # if v.find("\'") >= 0: v="''".join(v.split("\'")) | ||
| # v="'%s'" % v | ||
| return v | ||
| # Patched by yo. datetime is added. | ||
| new_valid_types = 'int', 'float', 'string', 'nb', 'datetime' | ||
| new_valid_types += tuple(map('datetime(%s)'.__mod__, xrange(7))) | ||
| new_valid_types += tuple(map('datetime(%s)'.__mod__, range(7))) | ||
| try: | ||
| # BBB | ||
| from Shared.DC.ZRDB.sqlvar import valid_type | ||
| sqlvar.valid_type = new_valid_types.__contains__ | ||
| except ImportError: | ||
| sqlvar.valid_types = new_valid_types | ||
| SQLVar.render = SQLVar_render | ||
| SQLVar.__call__ = SQLVar_render | ||
| SQLVar.render = render | ||
| SQLVar.__call__ = render | ||
