diff options
| author | Mike Bayer <mike_mp@zzzcomputing.com> | 2010-12-09 20:20:23 -0500 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2010-12-09 20:20:23 -0500 |
| commit | 67259bfef1068b46ea72f04718de5bb514093dd2 (patch) | |
| tree | f36a04dc57d2ec962c49f93b53e0bb0d0d6be2e7 /lib/sqlalchemy | |
| parent | 5622cfd5ecd5efe5e8c737be6b2ba0100aaf86b0 (diff) | |
| download | sqlalchemy-67259bfef1068b46ea72f04718de5bb514093dd2.tar.gz | |
- callcount
- more inlining
Diffstat (limited to 'lib/sqlalchemy')
| -rw-r--r-- | lib/sqlalchemy/orm/mapper.py | 16 | ||||
| -rw-r--r-- | lib/sqlalchemy/orm/properties.py | 2 |
2 files changed, 10 insertions, 8 deletions
diff --git a/lib/sqlalchemy/orm/mapper.py b/lib/sqlalchemy/orm/mapper.py index 54d6a53e2..05e904a0d 100644 --- a/lib/sqlalchemy/orm/mapper.py +++ b/lib/sqlalchemy/orm/mapper.py @@ -1833,13 +1833,15 @@ class Mapper(object): if primary_key is not None: # set primary key attributes - for i, col in enumerate(mapper._pks_by_table[table]): - if mapper._get_state_attr_by_column( - state, state_dict, col) \ - is None and len(primary_key) > i: - mapper._set_state_attr_by_column( - state, state_dict, col, - primary_key[i]) + for pk, col in zip(primary_key, mapper._pks_by_table[table]): + # TODO: make sure this inlined code is OK + # with composites + prop = mapper._columntoproperty[col] + if state_dict.get(prop.key) is None: + # TODO: would rather say: + # state_dict[prop.key] = pk + # here, one test fails + prop._setattr(state, state_dict, pk, col) mapper._postfetch(uowtransaction, table, state, state_dict, c, diff --git a/lib/sqlalchemy/orm/properties.py b/lib/sqlalchemy/orm/properties.py index c2f88f9ed..776ecaf99 100644 --- a/lib/sqlalchemy/orm/properties.py +++ b/lib/sqlalchemy/orm/properties.py @@ -124,7 +124,7 @@ class ColumnProperty(StrategizedProperty): get_committed_value(state, dict_, passive=passive) def _setattr(self, state, dict_, value, column): - state.get_impl(self.key).set(state, dict_, value, None) + state.manager[self.key].impl.set(state, dict_, value, None) def merge(self, session, source_state, source_dict, dest_state, dest_dict, load, _recursive): |