Coverage for members/admin.py : 68%

from django.contrib import admin 

from django import forms 

from django.forms import ModelForm, CharField 

from .models import ( 

ArchivedEmail, BalanceTransactionLog, EmailAddress, 

EmailToMember, Erfa, Member, Subscriber, PremiumAddressLabel 

) 

@admin.register(EmailAddress) 

class EmailAdmin(admin.ModelAdmin): 

readonly_fields = ('gpg_error',) 

class EmailInline(admin.StackedInline): 

readonly_fields = ('gpg_error',) 

extra = 0 

model = EmailAddress 

def formfield_for_dbfield(self, db_field, **kwargs): 

if db_field.attname == 'gpg_key_id': 

kwargs['widget'] = forms.TextInput( 

attrs={'oninput': "this.value = this.value.replace(/\s+/g, '');", 'size': 40}) 

return super(EmailInline, self).formfield_for_dbfield(db_field, **kwargs) 

class MemberAdminForm(ModelForm): 

transaction_message = CharField(required=False) 

class Meta: 

model = Member 

fields = '__all__' 

def clean(self): 

if 'account_balance' in self.changed_data and 'transaction_message' not in self.changed_data: 

self._errors['transaction_message'] = self.error_class( 

['This is a required field when changing the account balance.']) 

return self.cleaned_data 

@admin.register(Member) 

class MemberAdmin(admin.ModelAdmin): 

form = MemberAdminForm 

inlines = [ 

EmailInline, 

] 

list_display = ('chaos_number', 'get_name', 'get_address', 'is_member', 'is_active', 'fee_paid_until', 

'account_balance', 'get_emails_string', 'comment') 

search_fields = ('chaos_number', 'first_name', 'last_name', 'address_1', 'address_2', 'address_3', 

'emailaddress__email_address', '=address_country', 'comment') 

readonly_fields = ["membership_type", 

"chaos_number", 

"membership_start", 

"fee_last_paid", 

"fee_paid_until", 

"transfer_token", 

"initial_password", 

"username"] 

def has_delete_permission(self, request, obj=None): 

return False 

def change_view(self, request, object_id, form_url='', extra_context=None): 

self.fields = ['chaos_number', 'membership_type', 'is_active', 'membership_reduced', 'erfa', 'first_name', 

'last_name', 'address_1', 'address_2', 'address_3', 'address_country', 'address_unknown', 

'transfer_token', 'username', 'initial_password', 'account_balance', 'transaction_message', 

'fee_override', 'notification_consent', 'wants_datenschleuder', 'comment', 'membership_start', 

'membership_end', 'fee_last_paid', 'fee_paid_until'] 

if request.GET.__contains__('doppel'): 

self.readonly_fields = ["chaos_number"] 

else: 

self.readonly_fields = ["membership_type", 

"chaos_number", 

"membership_start", 

"fee_last_paid", 

"fee_paid_until", 

"transfer_token", 

"initial_password", 

"username"] 

return super().change_view(request, object_id, form_url, extra_context) 

def add_view(self, request, form_url='', extra_context=None): 

self.fields = ['membership_type', 'is_active', 'membership_reduced', 'erfa', 'first_name', 

'last_name', 'address_1', 'address_2', 'address_3', 'address_country', 'address_unknown', 

'account_balance', 'transaction_message', 'fee_override', 'notification_consent', 

'wants_datenschleuder', 'comment', 'membership_start', ] 

return super().add_view(request, form_url, extra_context) 

def save_model(self, request, obj, form, change): 

if 'account_balance' in form.changed_data: 

increase_by = obj.account_balance - obj._original_state['account_balance'] 

obj.log_increased_balance(increased_by=increase_by, 

reason=BalanceTransactionLog.MANUAL_BOOKING, 

comment=form.cleaned_data['transaction_message']) 

super(MemberAdmin, self).save_model(request, obj, form, change) 

@admin.register(Subscriber) 

class SubscriberAdmin(admin.ModelAdmin): 

search_fields = ('chaos_number', 'first_name', 'last_name', 'address_1', 'address_2', 'address_3') 

@admin.register(Erfa) 

class ErfaAdmin(admin.ModelAdmin): 

pass 

@admin.register(EmailToMember) 

class EmailToMemberAdmin(admin.ModelAdmin): 

# list_display = ('member', 'subject', 'body', 'created', 'ready_to_send', 'email_type', ') 

def change_view(self, request, object_id, form_url='', extra_context=None): 

self.readonly_fields = ['rendered_preview'] 

return super().change_view(request, object_id, form_url, extra_context) 

@admin.register(ArchivedEmail) 

class EmailArchiveAdmin(admin.ModelAdmin): 

def has_add_permission(self, request, obj=None): 

return False 

def has_delete_permission(self, request, obj=None): 

return False 

@admin.register(BalanceTransactionLog) 

class BalanceTransactionLogAdmin(admin.ModelAdmin): 

readonly_fields = ('created_on', 'member', 'changed_value', 'new_value') 

@admin.register(PremiumAddressLabel) 

class PremiumAddressLabel(admin.ModelAdmin): 

pass