Add fieldsets for document and account

Author: datamik

src/country_workspace/contrib/hope/constants.py

@@ -6,3 +6,6 @@
 HOUSEHOLD_CHECKER_NAME: Final[str] = "HOPE Household core"
 INDIVIDUAL_CHECKER_NAME: Final[str] = "HOPE Individual core"
 PEOPLE_CHECKER_NAME: Final[str] = "HOPE People core"
+
+DOCUMENT_FIELDSET_NAME = "HOPE Document"
+ACCOUNT_FIELDSET_NAME = "HOPE Account"

src/country_workspace/versioning/scripts/0009_add_document_and_account_fieldsets.py

@@ -0,0 +1,90 @@
+# Generated by HCW 0.1.0 on 2025 06 23 11:12:31
+from django.db import transaction
+from hope_flex_fields.models import Fieldset, DataChecker, DataCheckerFieldset, FieldDefinition
+from packaging.version import Version
+
+from django import forms
+
+from country_workspace.contrib.hope.constants import (
+    INDIVIDUAL_CHECKER_NAME,
+    DOCUMENT_FIELDSET_NAME,
+    ACCOUNT_FIELDSET_NAME,
+)
+from country_workspace.utils.flex_fields import Base64ImageField
+
+_script_for_version = Version("0.1.0")
+
+
+fields_to_remove = [
+    ("national_id_issuer", {"field_type": forms.CharField}, {}),
+    ("national_id_no", {"field_type": forms.CharField}, {}),
+    ("national_id_photo", {"field_type": forms.CharField}, {}),
+    ("phone_no", {"field_type": forms.CharField}, {}),
+]
+
+document_fields = [
+    ("document_number", {"field_type": forms.CharField}, {}),
+    ("photo", {"field_type": Base64ImageField}, {}),
+    ("issuance_date", {"field_type": forms.DateField}, {}),
+    ("expiry_date", {"field_type": forms.DateField}, {}),
+]
+
+account_fields = [
+    ("number", {"field_type": forms.CharField}, {}),
+    ("data", {"field_type": forms.JSONField}, {}),
+]
+
+
+def remove_fields_from_individual_fieldset() -> None:
+    fieldset = Fieldset.objects.get(name=INDIVIDUAL_CHECKER_NAME)
+    for field_name, _, _ in fields_to_remove:
+        fieldset.fields.get(name=field_name).delete()
+
+
+def create_and_add_fieldsets_to_datachecker() -> tuple[Fieldset, Fieldset]:
+    ind_datachecker = DataChecker.objects.get(name=INDIVIDUAL_CHECKER_NAME)
+
+    document_fieldset, _ = Fieldset.objects.get_or_create(name=DOCUMENT_FIELDSET_NAME)
+    account_fieldset, _ = Fieldset.objects.get_or_create(name=ACCOUNT_FIELDSET_NAME)
+    for fieldset, prefixes in (
+        (document_fieldset, ("national_id__", "national_passport__")),
+        (account_fieldset, ("photo__", "bank__")),
+    ):
+        for prefix in prefixes:
+            DataCheckerFieldset.objects.get_or_create(checker=ind_datachecker, fieldset=fieldset, prefix=prefix)
+    return document_fieldset, account_fieldset
+
+
+def add_fields_to_fieldsets(fieldset: Fieldset, fields: list) -> None:
+    for field_name, field_def, field_attrs in fields:
+        fieldset.fields.get_or_create(
+            name=field_name, definition=FieldDefinition.objects.get(**field_def), defaults={"attrs": field_attrs or {}}
+        )
+
+
+def forward() -> None:
+    with transaction.atomic():
+        remove_fields_from_individual_fieldset()
+
+        document_fieldset, account_fieldset = create_and_add_fieldsets_to_datachecker()
+
+        for fieldset, fields in ((document_fieldset, document_fields), (account_fieldset, account_fields)):
+            add_fields_to_fieldsets(fieldset, fields)
+
+
+def backward() -> None:
+    with transaction.atomic():
+        Fieldset.objects.filter(name__in=[DOCUMENT_FIELDSET_NAME, ACCOUNT_FIELDSET_NAME]).delete()
+
+        individual_fieldset, _ = Fieldset.objects.get_or_create(name=INDIVIDUAL_CHECKER_NAME)
+        for field_name, field_def, field_attrs in fields_to_remove:
+            individual_fieldset.fields.get_or_create(
+                name=field_name,
+                definition=FieldDefinition.objects.get(**field_def),
+                defaults={"attrs": field_attrs or {}},
+            )
+
+
+class Scripts:
+    requires = []
+    operations = [(forward, backward)]