Begin refactor of cost centre page for payroll

uktrade · Oct 28, 2024 · ed0ddb0 · ed0ddb0
1 parent 68b63f7
commit ed0ddb0
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 2 deletions.
diff --git a/forecast/templates/forecast/edit/choose_cost_centre.html b/forecast/templates/forecast/edit/choose_cost_centre.html
@@ -16,6 +16,7 @@ <h3 class="govuk-heading-l">My cost centres</h3>
 {% endblock %}
 {% block scripts %}
     <script>
+        window.isForecastOrPayroll = {{ view.get_is_forecast_or_payroll|safe }}
         window.costCentres = {{ view.get_user_cost_centres|safe }};
         window.currentFinancialYear = {{ view.get_financial_year|safe }};
         window.currentFinancialYearDisplay = "{{ view.get_financial_year_display|safe }}";

diff --git a/forecast/views/edit_select_cost_centre.py b/forecast/views/edit_select_cost_centre.py
@@ -80,6 +80,12 @@ def get_user_cost_centres(self):
 
         return json.dumps(cost_centres)
 
+    def get_is_forecast_or_payroll(self):
+        if "forecast" in self.request.path:
+            return ["forecast"]
+
+        return ["payroll"]
+
     def get_form_kwargs(self):
         kwargs = super(ChooseCostCentreView, self).get_form_kwargs()
         kwargs["user"] = self.request.user

diff --git a/front_end/src/Components/CostCentreList/index.jsx b/front_end/src/Components/CostCentreList/index.jsx
@@ -6,6 +6,7 @@ const CostCentreList = ({rowIndex, cellKey, format}) => {
     const [financialYears, setFinancialYears] = useState([])
     const [forecastFinYearDisplay, setForecastFinYearDisplay] = useState(null)
     const [forecastFinYear, setForecastFinYear] = useState(null)
+    const [isForecastOrPayroll, setIsForecastOrPayroll] = useState([])
 
     useEffect(() => {
         const timer = () => {
@@ -18,6 +19,7 @@ const CostCentreList = ({rowIndex, cellKey, format}) => {
                     if (window.financialYears){
                         setFinancialYears(window.financialYears)
                     }
+                    setIsForecastOrPayroll(window.isForecastOrPayroll)
                 } else {
                     timer()
                 }
@@ -68,7 +70,7 @@ const CostCentreList = ({rowIndex, cellKey, format}) => {
             <ul className="cost-centre-list">
               {displayedCentres.map((costCentre, index) => {
                 return <li key={index}>
-                    <a href={ `/forecast/edit/${costCentre.code}/${forecastFinYear}` } className="govuk-link">{costCentre.code} - {costCentre.name}</a>
+                    <a href={ `/${isForecastOrPayroll}/edit/${costCentre.code}/${forecastFinYear}` } className="govuk-link">{costCentre.code} - {costCentre.name}</a>
                 </li>
               })}
             </ul>

diff --git a/payroll/urls.py b/payroll/urls.py
@@ -1,12 +1,13 @@
 from django.urls import path
 
+from forecast.views.edit_select_cost_centre import ChooseCostCentreView
+
 from . import views
 
 
 app_name = "payroll"
 
 urlpatterns = [
-    # TODO: Add choose financial year and cost centre url.
     path(
         "edit/<str:cost_centre_code>/<int:financial_year>/",
         views.edit_payroll_page,
@@ -17,4 +18,9 @@
         views.PayrollView.as_view(),
         name="api",
     ),
+    path(
+        "edit/choose-cost-centre/",
+        ChooseCostCentreView.as_view(),
+        name="choose_cost_centre",
+    ),
 ]