CrashAdmin

Complexity

Total Complexity

16

Size/Duplication

Total Lines	57
Duplicated Lines	0 %

Importance

Changes	1
Bugs	1	Features	0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	archive_field()	0	2	1
A	regenerate_stacktrace()	0	3	2
A	cpu_architecture_field()	0	2	2
A	summary_field()	0	5	2
B	sentry_link_field()	0	13	6
A	get_form()	0	7	3

# coding: utf8

"""
This software is licensed under the Apache 2 license, quoted below.

Copyright 2014 Crystalnix Limited

Licensed under the Apache License, Version 2.0 (the "License"); you may not
use this file except in compliance with the License. You may obtain a copy of
the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
License for the specific language governing permissions and limitations under
the License.
"""

from django.contrib import admin
from django.core.exceptions import ObjectDoesNotExist
from django.conf import settings
from django.utils.html import format_html

from celery import signature

from crash.forms import SymbolsAdminForm, CrashFrom
from crash.models import Crash, CrashDescription, Symbols
from crash.forms import TextInputForm

SENTRY_DOMAIN = getattr(settings, 'SENTRY_STACKTRACE_DOMAIN', None)
SENTRY_ORG_SLUG = getattr(settings, 'SENTRY_STACKTRACE_ORG_SLUG', None)
SENTRY_PROJ_SLUG = getattr(settings, 'SENTRY_STACKTRACE_PROJ_SLUG', None)
CRASH_TRACKER = getattr(settings, 'CRASH_TRACKER', None)

class BooleanFilter(admin.SimpleListFilter):
    title = None
    parameter_name = None

    def lookups(self, request, model_admin):
        return (
            ('yes', 'Yes'),
            ('no', 'No'),
        )

    def queryset(self, request, queryset):
        d = {self.parameter_name: ''}
        if self.value() == 'yes':
            return queryset.exclude(**d)
        if self.value() == 'no':
            return queryset.filter(**d)


class TextInputFilter(admin.filters.FieldListFilter):
    template = 'admin/textinput_filter.html'

    def __init__(self, field, request, params, model, model_admin, field_path):
        self.lookup_kwarg_equal = '%s' % field_path
        super(TextInputFilter, self).__init__(
            field, request, params, model, model_admin, field_path)
        self.form = self.get_form()

    def choices(self, cl):
        return []

    def expected_parameters(self):
        return [self.lookup_kwarg_equal]

    def get_form(self):
        return TextInputForm(data=self.used_parameters,
                             field_name=self.field_path)

    def queryset(self, request, queryset):
        if self.form.is_valid():
            filter_params = dict(filter(lambda x: bool(x[1]),
                                        self.form.cleaned_data.items()))
            return queryset.filter(**filter_params)
        else:
            return queryset


class CrashArchiveFilter(BooleanFilter):
    title = 'Instrumental file'
    parameter_name = 'archive'


@admin.register(CrashDescription)
class CrashDescriptionAdmin(admin.ModelAdmin):
    readonly_fields = ('created', 'modified')
    list_display = ('created', 'modified', 'summary')
    list_display_links = ('created', 'modified', 'summary')


class CrashDescriptionInline(admin.StackedInline):
    model = CrashDescription


@admin.register(Crash)
class CrashAdmin(admin.ModelAdmin):
    list_display = ('id', 'created', 'modified', 'archive_field', 'signature', 'appid', 'userid', 'summary_field', 'os', 'build_number', 'channel', 'cpu_architecture_field',)
    list_select_related = ['crash_description']
    list_display_links = ('id', 'created', 'modified', 'signature', 'appid', 'userid', 'cpu_architecture_field',)
    list_filter = (('id', TextInputFilter,), 'created', CrashArchiveFilter, 'os', 'build_number', 'channel')
    search_fields = ('appid', 'userid', 'archive',)
    form = CrashFrom
    readonly_fields = ['sentry_link_field', 'os', 'build_number', 'channel',]
    exclude = ('groupid', 'eventid', )
    actions = ('regenerate_stacktrace',)
    inlines = [CrashDescriptionInline]

    def archive_field(self, obj):
        return bool(obj.archive)
    archive_field.short_description = 'Instrumental file'

    def cpu_architecture_field(self, obj):
        return obj.stacktrace_json.get('system_info', {}).get('cpu_arch', '') if obj.stacktrace_json else ''
    cpu_architecture_field.short_description = "CPU Architecture"

    def sentry_link_field(self, obj):
        if not SENTRY_DOMAIN or not SENTRY_ORG_SLUG or not SENTRY_PROJ_SLUG:
            return "Sentry variables are not set"
        if not obj.groupid or not obj.eventid:
            return "No sentry link"
        link = "http://{}/{}/{}/group/{}/events/{}/".format(
            SENTRY_DOMAIN,
            SENTRY_ORG_SLUG,
            SENTRY_PROJ_SLUG,
            obj.groupid,
            obj.eventid
        )
        return format_html("<a href='{link}'>{link}</a>".format(link=link))

    sentry_link_field.short_description = "Sentry link"
    sentry_link_field.allow_tags = True

    def summary_field(self, obj):
        try:
            return obj.crash_description.summary
        except ObjectDoesNotExist:
            return None
    summary_field.short_description = 'Summary'

    def regenerate_stacktrace(self, request, queryset):
        for i in queryset:
            signature("tasks.processing_crash_dump", args=(i.pk,)).apply_async(queue='default')
    regenerate_stacktrace.short_description = 'Regenerate stacktrace'

    def get_form(self, request, obj=None, **kwargs):
        if CRASH_TRACKER != 'Sentry':
            try:
                self.readonly_fields.remove('sentry_link_field')
            except ValueError:
                pass
        return super(CrashAdmin, self).get_form(request, obj, **kwargs)


@admin.register(Symbols)
class SymbolsAdmin(admin.ModelAdmin):
    readonly_fields = ('created', 'modified', )
    list_display = ('created', 'modified', 'debug_file', 'debug_id',)
    list_display_links = ('created', 'modified', 'debug_file', 'debug_id',)
    form = SymbolsAdminForm