Only admin can view organization cost (#413)

VNOI-Admin · Nov 18, 2024 · 10869e1 · 10869e1
1 parent d8626d1
commit 10869e1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 10 deletions.
diff --git a/judge/views/organization.py b/judge/views/organization.py
@@ -567,18 +567,14 @@ def get_filter(self):
         return _filter & Q(organizations=self.organization)
 
 
-class MonthlyCreditUsageOrganization(TitleMixin, PublicOrganizationMixin, ListView):
-    model = OrganizationMonthlyUsage
+class MonthlyCreditUsageOrganization(LoginRequiredMixin, TitleMixin, AdminOrganizationMixin, ListView):
     template_name = 'organization/usage.html'
-    context_object_name = 'usages'
-
-    def get_queryset(self):
-        return OrganizationMonthlyUsage.objects.filter(organization=self.organization)\
-            .order_by('time').values('time', 'consumed_credit')
 
     def get_context_data(self, **kwargs):
         context = super().get_context_data(**kwargs)
         context['title'] = self.organization.name
+        context['usages'] = OrganizationMonthlyUsage.objects.filter(organization=self.organization)\
+            .order_by('time').values('time', 'consumed_credit')
 
         usages = context['usages']
         days = [usage['time'].isoformat() for usage in usages] + [_('Current month')]

diff --git a/templates/organization/home.html b/templates/organization/home.html
@@ -104,10 +104,10 @@ <h3>{{ _('Controls') }} <i class="fa fa-question-circle"></i></h3>
                         {% endif %}
                     {% endif %}
                     <br>
-                    <div>
-                        <a href="{{ url('organization_monthly_usage', organization.slug) }}">{{ _('Organization cost') }}</a>
-                    </div>
                     {% if can_edit %}
+                        <div>
+                            <a href="{{ url('organization_monthly_usage', organization.slug) }}">{{ _('Organization cost') }}</a>
+                        </div>
                         {% if not organization.is_open %}
                             <div>
                                 <a href="{{ url('organization_requests_pending', organization.slug) }}">