mwolff44 / pyfreebilling

    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()

    def get_context_data(self, **kwargs):

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

    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()

class CdrReportCustView(LoginRequiredMixin, FormMixin, ListView):