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

Fix serializers of unified_jobs & ad_hoc_commands to avoid special exceptions

Browse Source
This commit is contained in:
Guoqiang Zhang
2018-07-01 15:33:54 -04:00
parent 6f63c7ba38
commit 5a4451ddd4
2 changed files with 28 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
awx/api/serializers.py
View File

@@ -375,7 +375,10 @@ class BaseSerializer(serializers.ModelSerializer):
isinstance(obj, Project)): isinstance(obj, Project)):
continue continue
fkval = getattr(obj, fk, None) try:
fkval = getattr(obj, fk, None)
except ObjectDoesNotExist:
continue
if fkval is None: if fkval is None:
continue continue
if fkval == obj: if fkval == obj:
@@ -3430,10 +3433,10 @@ class AdHocCommandSerializer(UnifiedJobSerializer):
def get_related(self, obj): def get_related(self, obj):
res = super(AdHocCommandSerializer, self).get_related(obj) res = super(AdHocCommandSerializer, self).get_related(obj)
if obj.inventory: if obj.inventory_id:
res['inventory'] = self.reverse('api:inventory_detail', kwargs={'pk': obj.inventory.pk}) res['inventory'] = self.reverse('api:inventory_detail', kwargs={'pk': obj.inventory_id})
if obj.credential: if obj.credential_id:
res['credential'] = self.reverse('api:credential_detail', kwargs={'pk': obj.credential.pk}) res['credential'] = self.reverse('api:credential_detail', kwargs={'pk': obj.credential_id})
res.update(dict( res.update(dict(
events = self.reverse('api:ad_hoc_command_ad_hoc_command_events_list', kwargs={'pk': obj.pk}), events = self.reverse('api:ad_hoc_command_ad_hoc_command_events_list', kwargs={'pk': obj.pk}),
activity_stream = self.reverse('api:ad_hoc_command_activity_stream_list', kwargs={'pk': obj.pk}), activity_stream = self.reverse('api:ad_hoc_command_activity_stream_list', kwargs={'pk': obj.pk}),
@@ -3445,9 +3448,9 @@ class AdHocCommandSerializer(UnifiedJobSerializer):
def to_representation(self, obj): def to_representation(self, obj):
ret = super(AdHocCommandSerializer, self).to_representation(obj) ret = super(AdHocCommandSerializer, self).to_representation(obj)
if 'inventory' in ret and not obj.inventory: if 'inventory' in ret and not obj.inventory_id:
ret['inventory'] = None ret['inventory'] = None
if 'credential' in ret and not obj.credential: if 'credential' in ret and not obj.credential_id:
ret['credential'] = None ret['credential'] = None
# For the UI, only module_name is returned for name, instead of the # For the UI, only module_name is returned for name, instead of the
# longer module name + module_args format. # longer module name + module_args format.

33
awx/main/access.py
View File

@@ -391,21 +391,24 @@ class BaseAccess(object):
return user_capabilities return user_capabilities
def get_method_capability(self, method, obj, parent_obj): def get_method_capability(self, method, obj, parent_obj):
if method in ['change']: # 3 args try:
return self.can_change(obj, {}) if method in ['change']: # 3 args
elif method in ['delete', 'run_ad_hoc_commands', 'copy']: return self.can_change(obj, {})
access_method = getattr(self, "can_%s" % method) elif method in ['delete', 'run_ad_hoc_commands', 'copy']:
return access_method(obj) access_method = getattr(self, "can_%s" % method)
elif method in ['start']: return access_method(obj)
return self.can_start(obj, validate_license=False) elif method in ['start']:
elif method in ['attach', 'unattach']: # parent/sub-object call return self.can_start(obj, validate_license=False)
access_method = getattr(self, "can_%s" % method) elif method in ['attach', 'unattach']: # parent/sub-object call
if type(parent_obj) == Team: access_method = getattr(self, "can_%s" % method)
relationship = 'parents' if type(parent_obj) == Team:
parent_obj = parent_obj.member_role relationship = 'parents'
else: parent_obj = parent_obj.member_role
relationship = 'members' else:
return access_method(obj, parent_obj, relationship, skip_sub_obj_read_check=True, data={}) relationship = 'members'
return access_method(obj, parent_obj, relationship, skip_sub_obj_read_check=True, data={})
except (ParseError, ObjectDoesNotExist):
return False
return False return False