diff options
Diffstat (limited to 'lib/sqlalchemy/databases/firebird.py')
| -rw-r--r-- | lib/sqlalchemy/databases/firebird.py | 33 |
1 files changed, 30 insertions, 3 deletions
diff --git a/lib/sqlalchemy/databases/firebird.py b/lib/sqlalchemy/databases/firebird.py index c83a44e01..17232f5d4 100644 --- a/lib/sqlalchemy/databases/firebird.py +++ b/lib/sqlalchemy/databases/firebird.py @@ -262,7 +262,7 @@ class FBExecutionContext(default.DefaultExecutionContext): class FBDialect(default.DefaultDialect): """Firebird dialect""" - supports_sane_rowcount = False + supports_sane_rowcount = True supports_sane_multi_rowcount = False max_identifier_length = 31 preexecute_pk_sequences = True @@ -434,6 +434,22 @@ class FBDialect(default.DefaultDialect): WHERE rc.rdb$constraint_type=? AND rc.rdb$relation_name=? ORDER BY se.rdb$index_name, se.rdb$field_position """ + # Heuristic-query to determine the generator associated to a PK field + genqry = """ + SELECT trigdep.rdb$depended_on_name AS fgenerator + FROM rdb$dependencies tabdep + JOIN rdb$dependencies trigdep ON (tabdep.rdb$dependent_name=trigdep.rdb$dependent_name + AND trigdep.rdb$depended_on_type=14 + AND trigdep.rdb$dependent_type=2) + JOIN rdb$triggers trig ON (trig.rdb$trigger_name=tabdep.rdb$dependent_name) + WHERE tabdep.rdb$depended_on_name=? + AND tabdep.rdb$depended_on_type=0 + AND trig.rdb$trigger_type=1 + AND tabdep.rdb$field_name=? + AND (SELECT count(*) + FROM rdb$dependencies trigdep2 + WHERE trigdep2.rdb$dependent_name = trigdep.rdb$dependent_name) = 2 + """ tablename = self._denormalize_name(table.name) @@ -457,7 +473,7 @@ class FBDialect(default.DefaultDialect): args = [name] kw = {} - # get the data types and lengths + # get the data type coltype = ischema_names.get(row['ftype'].rstrip()) if coltype is None: util.warn("Did not recognize type '%s' of column '%s'" % @@ -476,10 +492,21 @@ class FBDialect(default.DefaultDialect): # does it have a default value? if row['fdefault'] is not None: # the value comes down as "DEFAULT 'value'" + assert row['fdefault'].startswith('DEFAULT ') defvalue = row['fdefault'][8:] args.append(schema.PassiveDefault(sql.text(defvalue))) - table.append_column(schema.Column(*args, **kw)) + col = schema.Column(*args, **kw) + if kw['primary_key']: + # if the PK is a single field, try to see if its linked to + # a sequence thru a trigger + if len(pkfields)==1: + genc = connection.execute(genqry, [tablename, row['fname']]) + genr = genc.fetchone() + if genr is not None: + col.sequence = schema.Sequence(self._normalize_name(genr['fgenerator'])) + + table.append_column(col) if not found_table: raise exceptions.NoSuchTableError(table.name) |