mwolff44 / pyfreebilling

        return qs.none()

    def get_context_data(self, **kwargs):

        context = super(ListRatesCustView, self).get_context_data(**kwargs)

        try:

            self.usercompany = Person.objects.get(user=self.request.user)

            self.company = get_object_or_404(Company,

                                             name=self.usercompany.company)

            context['ratecards'] = CustomerRateCards.objects.filter(

                company=self.company.pk)\

                .filter(ratecard__enabled=True)\

                .order_by('priority')

            # import pdb; pdb.set_trace()

            return context

        except Person.DoesNotExist:

            messages.error(self.request,

                           _(u"""This user is not linked to a customer !"""))

        return context

class StatsCustView(LoginRequiredMixin, TemplateView):

    paginate_by = 10

    model = CustomerDirectory

    def get_queryset(self):

        qs = super(SipAccountCustView, self).get_queryset()

        try:

            self.usercompany = Person.objects.get(user=self.request.user)

            self.company = get_object_or_404(Company,

                                             name=self.usercompany.company)

            return CustomerDirectory.objects.filter(company=self.company.pk)\

                                            .order_by('id')

        except Person.DoesNotExist:

            messages.error(self.request,

                           _(u"""This user is not linked to a customer !"""))

        return qs.none()

    context_object_name = 'CustomerBalance'

    paginate_by = 10

    def get_queryset(self):

        qs = super(BalanceHistoryCustView, self).get_queryset()

        try:

            self.usercompany = Person.objects.get(user=self.request.user)

            self.company = get_object_or_404(Company, name=self.usercompany.company)

            return CompanyBalanceHistory.objects.filter(company=self.company.pk)\

                                                .filter(operation_type='customer')\

                                                .order_by('-date_modified')

        except Person.DoesNotExist:

            messages.error(self.request, _(u"""This user is not linked to a customer !"""))

        return qs.none()