Commit af650c8c authored by Arnaud Fontaine's avatar Arnaud Fontaine Committed by Aurel

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.
parent 119829b1
...@@ -13,57 +13,64 @@ ...@@ -13,57 +13,64 @@
from Shared.DC.ZRDB.sqltest import * from Shared.DC.ZRDB.sqltest import *
from Shared.DC.ZRDB import sqltest from Shared.DC.ZRDB import sqltest
from DateTime import DateTime from DateTime import DateTime
from types import StringType
list_type_list = list, tuple, set, frozenset, dict list_type_list = list, tuple, set, frozenset, dict
if 1: # For easy diff with original if 1: # For easy diff with original (ZSQLMethods 3.14)
def render(self, md): def render(self, md):
name=self.__name__ name = self.__name__
t=self.type t = self.type
args=self.args args = self.args
try: try:
expr=self.expr expr = self.expr
if type(expr) is type(''): if isinstance(expr, StringTypes):
v=md[expr] v = md[expr]
else: else:
v=expr(md) v = expr(md)
except (KeyError, NameError): except (KeyError, NameError):
if 'optional' in args and args['optional']: if 'optional' in args and args['optional']:
return '' return ''
raise ValueError('Missing input variable, <em>%s</em>' % name) raise ValueError('Missing input variable, <em>%s</em>' % name)
# PATCH: use isinstance instead of type comparison, to allow
# subclassing.
if isinstance(v, list_type_list): if isinstance(v, list_type_list):
if len(v) > 1 and not self.multiple: if len(v) > 1 and not self.multiple:
raise ValueError( msg = 'multiple values are not allowed for <em>%s</em>' % name
'multiple values are not allowed for <em>%s</em>' raise ValueError(msg)
% name) else:
else: v=[v] v = [v]
vs=[] vs = []
for v in v: for v in v:
if not v and type(v) is StringType and t != 'string': continue if not v and isinstance(v, StringTypes) and t != 'string':
if t=='int': continue
if t == 'int':
try: try:
if type(v) is StringType: if isinstance(v, StringTypes):
if v[-1:]=='L': if six.PY3 and isinstance(v, bytes):
v=v[:-1] v = v.decode(self.encoding or 'UTF-8')
if v[-1:] == 'L':
v = v[:-1]
int(v) int(v)
else: v=str(int(v)) else:
except ValueError: v = str(int(v))
raise ValueError( except (TypeError, ValueError):
'Invalid integer value for <em>%s</em>' % name) msg = 'Invalid integer value for <em>%s</em>' % name
elif t=='float': raise ValueError(msg)
if not v and type(v) is StringType: continue elif t == 'float':
if not v and isinstance(v, StringTypes):
continue
try: try:
if type(v) is StringType: float(v) if six.PY3 and isinstance(v, bytes):
else: v=str(float(v)) v = v.decode(self.encoding or 'UTF-8')
except ValueError: if isinstance(v, StringTypes):
raise ValueError( float(v)
'Invalid floating-point value for <em>%s</em>' % name) else:
v = str(float(v))
except (TypeError, ValueError):
msg = 'Invalid floating-point value for <em>%s</em>' % name
raise ValueError(msg)
elif t.startswith('datetime'): elif t.startswith('datetime'):
# For subsecond precision, use 'datetime(N)' MySQL type, # For subsecond precision, use 'datetime(N)' MySQL type,
# where N is the number of digits after the decimal point. # where N is the number of digits after the decimal point.
...@@ -73,38 +80,45 @@ if 1: # For easy diff with original ...@@ -73,38 +80,45 @@ if 1: # For easy diff with original
('.%06u' % (v.micros() % 1000000))[:1+n] if n else '') ('.%06u' % (v.micros() % 1000000))[:1+n] if n else '')
else: else:
if not isinstance(v, (str, unicode)): if not isinstance(v, StringTypes):
v = str(v) v = str(v)
v=md.getitem('sql_quote__',0)(v) if isinstance(v, six.binary_type):
#if find(v,"\'") >= 0: v=join(split(v,"\'"),"''") v = v.decode('utf-8')
#v="'%s'" % v # The call to sql_quote__ can return something that is not
# a native string anymore!
v = md.getitem('sql_quote__', 0)(v)
if isinstance(v, six.binary_type):
v = v.decode('utf-8')
# if v.find("\'") >= 0: v="''".(v.split("\'"))
# v="'%s'" % v
vs.append(v) vs.append(v)
if not vs and t=='nb': if not vs and t == 'nb':
if 'optional' in args and args['optional']: if 'optional' in args and args['optional']:
return '' return ''
else: else:
raise ValueError( err = 'Invalid empty string value for <em>%s</em>' % name
'Invalid empty string value for <em>%s</em>' % name) raise ValueError(err)
if not vs: if not vs:
if self.optional: return '' if self.optional:
raise ValueError( return ''
'No input was provided for <em>%s</em>' % name) raise ValueError('No input was provided for <em>%s</em>' % name)
if len(vs) > 1: if len(vs) > 1:
vs = ', '.join(map(str, vs)) vs = ', '.join(map(str, vs))
if self.op == '<>': if self.op == '<>':
## Do the equivalent of 'not-equal' for a list, # Do the equivalent of 'not-equal' for a list,
## "a not in (b,c)" # "a not in (b,c)"
return "%s not in (%s)" % (self.column, vs) return '%s not in (%s)' % (self.column, vs)
else: else:
## "a in (b,c)" # "a in (b,c)"
return "%s in (%s)" % (self.column, vs) return '%s in (%s)' % (self.column, vs)
return "%s %s %s" % (self.column, self.op, vs[0]) return '%s %s %s' % (self.column, self.op, vs[0])
SQLTest.render = SQLTest.__call__ = render SQLTest.render = SQLTest.__call__ = render
new_valid_types = (('int', 'float', 'string', 'nb', 'datetime') + tuple('datetime(%s)' % x for x in xrange(7))) from builtins import range
new_valid_types = (('int', 'float', 'string', 'nb', 'datetime') + tuple('datetime(%s)' % x for x in range(7)))
try: try:
# BBB # BBB
......
...@@ -14,84 +14,99 @@ ...@@ -14,84 +14,99 @@
# dtml-sqlvar patch to convert None to NULL, and deal with DateTime # 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 Shared.DC.ZRDB import sqlvar
from string import atoi,atof
from DateTime import DateTime from DateTime import DateTime
def SQLVar_render(self, md): if 1: # For easy diff with original (ZSQLMethods 3.14)
args=self.args def render(self, md):
t=args['type'] name = self.__name__
try: args = self.args
expr=self.expr t = args['type']
if type(expr) is str: v=md[expr] try:
else: v=expr(md) expr = self.expr
except Exception: if isinstance(expr, StringTypes):
if args.get('optional'): 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' 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'): if t == 'int':
return 'null' 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': elif t.startswith('datetime'):
try: # For subsecond precision, use 'datetime(N)' MySQL type,
if type(v) is str: # where N is the number of digits after the decimal point.
if v[-1:]=='L': n = 0 if t == 'datetime' else int(t[9])
v=v[:-1] try:
atoi(v) v = (v if isinstance(v, DateTime) else DateTime(v)).toZone('UTC')
return v return "'%s%s'" % (v.ISO(),
return str(int(v)) ('.%06u' % (v.micros() % 1000000))[:1+n] if n else '')
except Exception: except Exception:
t = 'integer' t = 'datetime'
elif t=='float':
try: elif t=='nb' and not v:
if type(v) is str: t = 'empty string'
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
if args.get('optional'): else:
return 'null' if not isinstance(v, (str, StringTypes)):
raise ValueError('Invalid %s value for <em>%s</em>: %r' v = str(v)
% (t, self.__name__, 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. # Patched by yo. datetime is added.
new_valid_types = 'int', 'float', 'string', 'nb', 'datetime' 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: try:
# BBB # BBB
from Shared.DC.ZRDB.sqlvar import valid_type from Shared.DC.ZRDB.sqlvar import valid_type
sqlvar.valid_type = new_valid_types.__contains__ sqlvar.valid_type = new_valid_types.__contains__
except ImportError: except ImportError:
sqlvar.valid_types = new_valid_types sqlvar.valid_types = new_valid_types
SQLVar.render = SQLVar_render SQLVar.render = render
SQLVar.__call__ = SQLVar_render SQLVar.__call__ = render
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment