"""
This module defines common view code to be used by multiple gnuviechadmin apps.

"""
from __future__ import absolute_import, unicode_literals

from django.shortcuts import get_object_or_404
from hostingpackages.models import CustomerHostingPackage


class HostingPackageAndCustomerMixin(object):
    """
    Mixin for views that gets the hosting package instance from the URL
    keyword argument 'package'.

    """
    hosting_package_kwarg = 'package'
    """Keyword argument used to find the hosting package in the URL."""

    def get_hosting_package(self):
        return get_object_or_404(
            CustomerHostingPackage,
            pk=int(self.kwargs[self.hosting_package_kwarg]))

    def get_customer_object(self):
        return self.get_hosting_package().customer