External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-03-07 11:41:08 -03:30
Code Issues Packages Projects Releases Wiki Activity

Fixed ImplicitRoleField and ImplicitResourceField's from being too lazy

Browse Source 
This ensures that the role and resource fields get created and bound
automatically without having to explicitly access them a first time.
This commit is contained in:
Akita Noek
2016-01-29 18:22:08 -05:00
parent 1035a6737e
commit c6b2e509fd
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
awx/main/fields.py
View File

@@ -2,6 +2,7 @@
# All Rights Reserved. # All Rights Reserved.
# Django # Django
from django.db.models.signals import post_save, post_init
from django.db import models from django.db import models
from django.db.models.fields.related import SingleRelatedObjectDescriptor from django.db.models.fields.related import SingleRelatedObjectDescriptor
from django.db.models.fields.related import ReverseSingleRelatedObjectDescriptor from django.db.models.fields.related import ReverseSingleRelatedObjectDescriptor
@@ -91,7 +92,11 @@ class ImplicitResourceField(models.ForeignKey):
def contribute_to_class(self, cls, name): def contribute_to_class(self, cls, name):
super(ImplicitResourceField, self).contribute_to_class(cls, name) super(ImplicitResourceField, self).contribute_to_class(cls, name)
setattr(cls, self.name, ResourceFieldDescriptor(self.parent_resource, self)) setattr(cls, self.name, ResourceFieldDescriptor(self.parent_resource, self))
post_save.connect(self._save, cls, True)
def _save(self, instance, *args, **kwargs):
# Ensure that our field gets initialized after our first save
getattr(instance, self.name)
class ImplicitRoleDescriptor(ReverseSingleRelatedObjectDescriptor): class ImplicitRoleDescriptor(ReverseSingleRelatedObjectDescriptor):
@@ -185,4 +190,8 @@ class ImplicitRoleField(models.ForeignKey):
self self
) )
) )
post_save.connect(self._save, cls, True)
def _save(self, instance, *args, **kwargs):
# Ensure that our field gets initialized after our first save
getattr(instance, self.name)