update quota view

pyros_astroguita
1 parent 673ff098
Showing 2 changed files with 54 additions and 5 deletions Show diff stats
src/core/pyros_django/scp_mgmt/templates/scp_mgmt/quota_sp.html
src/core/pyros_django/scp_mgmt/views.py
@@ -91,9 +91,9 @@
             <td>{{institute.quota.d_nextq}}</td>
             <td>{{institute.quota.d_nextx}}</td>
         </tr>
-        {% for sp in institute.scientific_programs.all %}
+        {% for sp in sp_periods_per_institute|get_item:institute.name %}
             <tr class="sp">
-                <td>Scientific program: {{sp.name}} </td>
+                <td>Scientific program: {{sp.scientific_program.name}} </td>
                 <td>{{sp.quota_f}} </td>
                 <td> <b>{{sp.quota.d_total}}</b> </td>
                 <td>{{sp.quota.d_totalq}}</td>
@@ -119,6 +119,35 @@
                 <td>{{sp.quota.d_nextq}}</td>
                 <td>{{sp.quota.d_nextx}}</td>
             </tr>       
+                {% for sequence in sequences|get_item:sp.scientific_program.name %}
+                <tr class="sequence">
+                    <td> Sequence: {{sequence.name}} </td>
+                    <td> </td>
+                    <td> <b>{{sequence.quota.d_total}}</b> </td>
+                    <td>{{sequence.quota.d_totalq}}</td>
+                    <td>{{sequence.quota.d_totalx}}</td>
+            
+                    <td> <b>{{sequence.quota.d_previous}}</b> </td>
+                    <td>{{sequence.quota.d_previousq}}</td>
+                    <td>{{sequence.quota.d_previousx}}</td>
+            
+                    <td> <b>{{sequence.quota.d_current}}</b> </td>
+                    <td>{{sequence.quota.d_currentq}}</td>
+                    <td>{{sequence.quota.d_currentx}}</td>
+            
+                    <td> <b>{{sequence.quota.d_passed}}</b> </td>
+                    <td>{{sequence.quota.d_passedq}}</td>
+                    <td>{{sequence.quota.d_passedx}}</td>
+            
+                    <td> <b>{{sequence.quota.d_schedule}}</b> </td>
+                    <td>{{sequence.quota.d_scheduleq}}</td>
+                    <td>{{sequence.quota.d_schedulex}} </td>
+            
+                    <td> <b>{{sequence.quota.d_next}}</b> </td>
+                    <td>{{sequence.quota.d_nextq}}</td>
+                    <td>{{sequence.quota.d_nextx}}</td>
+                </tr>       
+            {% endfor %}
         {% endfor %}
     {% endfor %}
@@ -21,7 +21,7 @@ from django.core.paginator import Paginator, EmptyPage, PageNotAnInteger
 # Project imports
 from dashboard.config_pyros import ConfigPyros
 from .functions import get_global_svg_timeline, get_svg_timeline, get_proposal_svg_timeline
-from user_mgmt.models import ScientificProgram, Institute, Period, SP_Period_User, SP_Period, PyrosUser, SP_Period_Guest, ScienceTheme   #, UserLevel
+from user_mgmt.models import ScientificProgram, Institute, Period, SP_Period_User, SP_Period, PyrosUser, SP_Period_Guest, ScienceTheme, UserLevel   #, UserLevel
 from seq_submit.models import Sequence, Quota
 #from src.core.pyros_django import scientific_program
 from .forms import PeriodForm, ScienceThemeForm, ScientificProgramForm, InstituteForm, SP_PeriodForm,TACAssociationForm
@@ -810,11 +810,31 @@ def institute_list(request):
 @login_required
 @level_required("Admin", "Unit-PI", "Observer", "Operator", "TAC", "Management board member")
 def quota_sp(request):
-    institutes = Institute.objects.all()
-    scientific_programs = ScientificProgram.objects.all()
     config = OBSConfig(os.environ["PATH_TO_OBSCONF_FILE"],os.environ["unit_name"])
+    CAN_VIEW_ALL_QUOTA = request.session.get("role") in ("Admin", "Unit-PI", "Unit-board", "Management board member")
     current_night = config.fn.date2night("now")
     current_period = Period.objects.exploitation_period()
+    sp_periods_per_institute = {}
+    if CAN_VIEW_ALL_QUOTA:
+        institutes = Institute.objects.all()
+        sp_period = SP_Period.objects.filter(period=current_period)
+    else:
+        user_sp = request.user.get_scientific_programs()
+        sp_period = SP_Period.objects.filter(scientific_program__in=user_sp, period=current_period)
+        institutes = []
+    # regroup sp_period per institute
+    for sp in sp_period:
+        if sp_periods_per_institute.get(sp.scientific_program.institute.name) is None:
+            sp_periods_per_institute[sp.scientific_program.institute.name] = []
+        if sp not in sp_periods_per_institute.get(sp.scientific_program.institute.name):
+            sp_periods_per_institute[sp.scientific_program.institute.name].append(sp)
+    sequences = {}
+    # regroup institutes and regroup sequence per scientific_program
+    for institute, sp_list in sp_periods_per_institute.items():
+        if not CAN_VIEW_ALL_QUOTA and institute not in institutes:
+            institutes.append(Institute.objects.get(name=institute))
+        for sp_period in sp_list:
+            sequences[sp_period.scientific_program.name] = sp_period.scientific_program.sequences.all()
     # lowest id is period line
     quota_current_night = Quota.objects.filter(id_period=current_period.id, night_id=current_night).order_by("id").first()
     return render(request, 'scp_mgmt/quota_sp.html', locals())