diff options
author | Daniele Varrazzo <daniele.varrazzo@gmail.com> | 2017-01-01 06:26:54 +0100 |
---|---|---|
committer | Daniele Varrazzo <daniele.varrazzo@gmail.com> | 2017-01-01 06:26:54 +0100 |
commit | ad2643266fc3868b30cdff4ba70ad606f555d00d (patch) | |
tree | d06c38572f95ffb9255307bc7a9ba03970ea22c6 /lib/sql.py | |
parent | 8c020ca47a50fb021f3af6a0813687ed9daa85d2 (diff) | |
download | psycopg2-ad2643266fc3868b30cdff4ba70ad606f555d00d.tar.gz |
Fixed sql.compose with no args and added tests
Diffstat (limited to 'lib/sql.py')
-rw-r--r-- | lib/sql.py | 8 |
1 files changed, 4 insertions, 4 deletions
@@ -204,7 +204,7 @@ re_compose = re.compile(""" """, re.VERBOSE) -def compose(sql, args=None): +def compose(sql, args=()): phs = list(re_compose.finditer(sql)) # check placeholders consistent @@ -240,8 +240,8 @@ def compose(sql, args=None): return _compose_map(sql, phs, args) else: - if not isinstance(args, collections.Sequence) and args: - raise TypeError( + if isinstance(args, collections.Sequence) and args: + raise ValueError( "the sql string expects no value, got %s instead" % len(args)) # If args are a mapping, no placeholder is an acceptable case @@ -267,7 +267,7 @@ def _compose_seq(sql, phs, args): if phs: rv.append(SQL(sql[phs[-1].end():])) else: - rv.append(sql) + rv.append(SQL(sql)) return Composed(rv) |