| Jeremy Mowery | a65b356 | 2015-10-01 19:19:38 -0700 | [diff] [blame] | 1 | from django.contrib import admin |
| 2 | |
| 3 | from helloworldservice.models import * |
| 4 | from django import forms |
| 5 | from django.utils.safestring import mark_safe |
| 6 | from django.contrib.auth.admin import UserAdmin |
| 7 | from django.contrib.admin.widgets import FilteredSelectMultiple |
| 8 | from django.contrib.auth.forms import ReadOnlyPasswordHashField |
| 9 | from django.contrib.auth.signals import user_logged_in |
| 10 | from django.utils import timezone |
| 11 | from django.contrib.contenttypes import generic |
| 12 | from suit.widgets import LinkedSelect |
| 13 | from core.admin import ServiceAppAdmin,SliceInline,ServiceAttrAsTabInline, ReadOnlyAwareAdmin, XOSTabularInline, ServicePrivilegeInline, TenantRootTenantInline, TenantRootPrivilegeInline |
| 14 | from core.middleware import get_request |
| 15 | |
| 16 | from functools import update_wrapper |
| 17 | from django.contrib.admin.views.main import ChangeList |
| 18 | from django.core.urlresolvers import reverse |
| 19 | from django.contrib.admin.utils import quote |
| 20 | |
| 21 | class HelloWorldServiceAdmin(ReadOnlyAwareAdmin): |
| 22 | model = HelloWorldService |
| 23 | verbose_name = "Hello World Service" |
| 24 | verbose_name_plural = "Hello World Service" |
| 25 | list_display = ("backend_status_icon", "name", "enabled") |
| 26 | list_display_links = ('backend_status_icon', 'name', ) |
| 27 | fieldsets = [(None, {'fields': ['backend_status_text', 'name','enabled','versionNumber', 'description',"view_url","icon_url" ], 'classes':['suit-tab suit-tab-general']})] |
| 28 | readonly_fields = ('backend_status_text', ) |
| 29 | inlines = [SliceInline,ServiceAttrAsTabInline,ServicePrivilegeInline] |
| 30 | |
| 31 | extracontext_registered_admins = True |
| 32 | |
| 33 | user_readonly_fields = ["name", "enabled", "versionNumber", "description"] |
| 34 | |
| 35 | suit_form_tabs =(('general', 'Hello World Service Details'), |
| 36 | ('administration', 'Administration'), |
| 37 | ('slices','Slices'), |
| 38 | ('serviceattrs','Additional Attributes'), |
| 39 | ('serviceprivileges','Privileges'), |
| 40 | ) |
| 41 | |
| 42 | suit_form_includes = (('helloworldserviceadmin.html', 'top', 'administration'), |
| 43 | ) |
| 44 | def queryset(self, request): |
| 45 | return HelloWorldService.get_service_objects_by_user(request.user) |
| 46 | |
| 47 | class HelloWorldTenantForm(forms.ModelForm): |
| 48 | creator = forms.ModelChoiceField(queryset=User.objects.all()) |
| 49 | display_message = forms.CharField(required=False) |
| 50 | def __init__(self,*args,**kwargs): |
| 51 | super (HelloWorldTenantForm,self ).__init__(*args,**kwargs) |
| 52 | self.fields['kind'].widget.attrs['readonly'] = True |
| 53 | self.fields['provider_service'].queryset = HelloWorldService.get_service_objects().all() |
| 54 | if self.instance: |
| 55 | # fields for the attributes |
| 56 | self.fields['creator'].initial = self.instance.creator |
| 57 | self.fields['display_message'].initial = self.instance.display_message |
| 58 | if (not self.instance) or (not self.instance.pk): |
| 59 | # default fields for an 'add' form |
| 60 | self.fields['kind'].initial = HELLO_WORLD_KIND |
| 61 | self.fields['creator'].initial = get_request().user |
| 62 | if HelloWorldService.get_service_objects().exists(): |
| 63 | self.fields["provider_service"].initial = HelloWorldService.get_service_objects().all()[0] |
| 64 | |
| 65 | |
| 66 | def save(self, commit=True): |
| 67 | self.instance.creator = self.cleaned_data.get("creator") |
| 68 | self.instance.display_message = self.cleaned_data.get("display_message") |
| 69 | return super(HelloWorldTenantForm, self).save(commit=commit) |
| 70 | |
| 71 | class Meta: |
| 72 | model = HelloWorldTenant |
| 73 | |
| 74 | class HelloWorldTenantAdmin(ReadOnlyAwareAdmin): |
| 75 | list_display = ('backend_status_icon', 'id', ) |
| 76 | list_display_links = ('backend_status_icon', 'id') |
| 77 | fieldsets = [ (None, {'fields': ['backend_status_text', 'kind', 'provider_service', 'service_specific_attribute', |
| 78 | 'instance', 'creator', 'display_message'], |
| 79 | 'classes':['suit-tab suit-tab-general']})] |
| 80 | readonly_fields = ('backend_status_text', 'instance', 'service_specific_attribute', ) |
| 81 | form = HelloWorldTenantForm |
| 82 | |
| 83 | suit_form_tabs = (('general','Details'),) |
| 84 | |
| 85 | def queryset(self, request): |
| 86 | return HelloWorldTenant.get_tenant_objects_by_user(request.user) |
| 87 | |
| 88 | admin.site.register(HelloWorldService, HelloWorldServiceAdmin) |
| 89 | admin.site.register(HelloWorldTenant, HelloWorldTenantAdmin) |