External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-05-13 04:17:36 -02:30
Code Issues Packages Projects Releases Wiki Activity

Refactor m2m_changed so reverse works

Browse Source
This commit is contained in:
Wayne Witzel III
2016-03-11 15:18:24 -05:00
parent 31a461956a
commit 45483fb057
1 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
awx/main/fields.py
View File

@@ -199,16 +199,17 @@ class ImplicitRoleField(models.ForeignKey):
if '.' in field_attr: if '.' in field_attr:
raise Exception('Referencing deep roles through ManyToMany fields is unsupported.') raise Exception('Referencing deep roles through ManyToMany fields is unsupported.')
reverse = type(field) is ReverseManyRelatedObjectsDescriptor if type(field) is ReverseManyRelatedObjectsDescriptor:
if reverse: sender = field.through
m2m_changed.connect(self.m2m_update(field_attr, reverse), field.through, weak=False)
else: else:
m2m_changed.connect(self.m2m_update(field_attr, reverse), field.related.through, weak=False) sender = field.related.through
def m2m_update(self, field_attr, _reverse): m2m_changed.connect(self.m2m_update(field_attr), sender, weak=False)
def _m2m_update(instance, action, model, pk_set, **kwargs):
def m2m_update(self, field_attr):
def _m2m_update(instance, action, model, pk_set, reverse, **kwargs):
if action == 'post_add' or action == 'pre_remove': if action == 'post_add' or action == 'pre_remove':
if _reverse: if reverse:
for pk in pk_set: for pk in pk_set:
obj = model.objects.get(pk=pk) obj = model.objects.get(pk=pk)
if action == 'post_add': if action == 'post_add':