Add initial staff app

config/settings/base.py

@@ -40,6 +40,7 @@
 VCAP_SERVICES = env.json("VCAP_SERVICES", {})
 
 INSTALLED_APPS = [
+    "staff.apps.StaffConfig",
     "user",
     "authbroker_client",
     "future_years.apps.FutureYearsConfig",

config/urls.py

@@ -35,6 +35,7 @@
     path("data-lake/", include("data_lake.urls")),
     path("oscar_return/", include("oscar_return.urls")),
     path("upload_split_file/", include("upload_split_file.urls")),
+    path("staff/", include("staff.urls")),
     path("admin/", admin.site.urls),
     # TODO - split below out into develop only?
     path(

core/models.py

@@ -61,6 +61,17 @@ def future_year_dictionary(self):
         )
 
 
+class FinancialYearQuerySet(models.QuerySet):
+    def current(self):
+        return self.filter(current=True).first()
+
+    def future(self):
+        current_financial_year = self.current().financial_year
+        return self.filter(financial_year__gt=current_financial_year).order_by(
+            "-financial_year"
+        )
+
+
 class FinancialYear(BaseModel):
     """Key and representation of the financial year"""
 
@@ -69,7 +80,7 @@ class FinancialYear(BaseModel):
     current = models.BooleanField(default=False)
     archived = models.BooleanField(default=False)
     archived_at = models.DateTimeField(blank=True, null=True)
-    objects = models.Manager()  # The default manager.
+    objects = FinancialYearQuerySet.as_manager()
     financial_year_objects = FinancialYearManager()
 
     def __str__(self):

docker-compose.yml

@@ -1,5 +1,3 @@
-version: "3"
-
 services:
   web:
     build:

forecast/models.py

@@ -153,6 +153,11 @@ def __str__(self):
         return self.forecast_expenditure_type_name
 
 
+class FinancialPeriodQuerySet(models.QuerySet):
+    def months(self):
+        return self.filter(financial_period_code__lte=12)
+
+
 class FinancialPeriodManager(models.Manager):
     def month_display_list(self):
         return list(
@@ -296,7 +301,7 @@ class FinancialPeriod(BaseModel):
     actual_loaded_previous_year = models.BooleanField(default=False)
     display_figure = models.BooleanField(default=True)
 
-    objects = models.Manager()  # The default manager.
+    objects = FinancialPeriodQuerySet.as_manager()
     financial_period_info = FinancialPeriodManager()
 
     class Meta:

staff/admin.py

@@ -0,0 +1,61 @@
+from django.contrib import admin
+
+from staff.services.staff import staff_created
+
+from .models import Staff, StaffForecast, Payroll, PayElement, PayElementGroup
+
+
+@admin.register(Staff)
+class StaffAdmin(admin.ModelAdmin):
+    list_display = [
+        "employee_no",
+        "first_name",
+        "last_name",
+        "cost_centre",
+    ]
+
+    def save_model(self, request, obj, form, change):
+        super().save_model(request, obj, form, change)
+
+        if not change:
+            staff_created(obj)
+
+
+@admin.register(StaffForecast)
+class StaffForecastAdmin(admin.ModelAdmin):
+    list_display = [
+        "staff",
+        "year",
+        "period_1",
+        "period_2",
+        "period_3",
+        "period_4",
+        "period_5",
+        "period_6",
+        "period_7",
+        "period_8",
+        "period_9",
+        "period_10",
+        "period_11",
+        "period_12",
+    ]
+
+
+@admin.register(Payroll)
+class PayrollAdmin(admin.ModelAdmin):
+    list_display = [
+        "staff",
+        "pay_element",
+        "debit_amount",
+        "credit_amount",
+    ]
+
+
+@admin.register(PayElement)
+class PayElementAdmin(admin.ModelAdmin):
+    pass
+
+
+@admin.register(PayElementGroup)
+class PayElementGroupAdmin(admin.ModelAdmin):
+    pass

staff/apps.py

@@ -0,0 +1,6 @@
+from django.apps import AppConfig
+
+
+class StaffConfig(AppConfig):
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "staff"