move mesh forms from site to mesh module… wtf

src/c3nav/site/forms.py

@@ -74,96 +74,3 @@ class PositionSetForm(Form):
         self.fields['position'].queryset = Position.objects.filter(owner=request.user)
         self.fields['position'].label_from_instance = attrgetter('name')
 
-
-@dataclass
-class OTAFormGroup:
-    hardware: HardwareDescription
-    builds: Sequence[FirmwareBuild]
-    fields: dict[str, tuple[MeshNode, Any]]
-
-    @cached_property
-    def builds_by_id(self) -> dict[int, FirmwareBuild]:
-        return {build.pk: build for build in self.builds}
-
-
-class OTACreateForm(Form):
-    def __init__(self, builds: Sequence[FirmwareBuild], *args, **kwargs):
-        super().__init__(*args, **kwargs)
-
-        nodes: Sequence[MeshNode] = MeshNode.objects.prefetch_last_messages(
-            MeshMessageType.CONFIG_BOARD
-        ).prefetch_firmwares().prefetch_ota()
-
-        builds_by_hardware = {}
-        for build in builds:
-            for hardware_desc in build.hardware_descriptions:
-                builds_by_hardware.setdefault(hardware_desc, []).append(build)
-
-        nodes_by_hardware = {}
-        for node in nodes:
-            nodes_by_hardware.setdefault(node.hardware_description, []).append(node)
-
-        self._groups: list[OTAFormGroup] = []
-        for hardware, hw_nodes in sorted(nodes_by_hardware.items(), key=lambda k: len(k[1]), reverse=True):
-            try:
-                hw_builds = builds_by_hardware[hardware]
-            except KeyError:
-                continue
-            choices = [
-                ('', '---'),
-                *((build.pk, build.variant) for build in hw_builds)
-            ]
-
-            group = OTAFormGroup(
-                hardware=hardware,
-                builds=hw_builds,
-                fields={
-                    f'build_{node.pk}': (node, (
-                        ChoiceField(choices=choices, required=False)
-                        if len(hw_builds) > 1
-                        else BooleanField(required=False)
-                    )) for node in hw_nodes
-                }
-            )
-            for name, (node, hw_field) in group.fields.items():
-                self.fields[name] = hw_field
-            self._groups.append(group)
-
-    @property
-    def groups(self) -> list[OTAFormGroup]:
-        return [
-            dataclass_replace(group, fields={
-                name: (node, self[name])
-                for name, (node, hw_field) in group.fields.items()
-            })
-            for group in self._groups
-        ]
-
-    @property
-    def selected_builds(self):
-        build_nodes = {}
-        for group in self._groups:
-            for name, (node, hw_field) in group.fields.items():
-                value = self.cleaned_data.get(name, None)
-                if not value:
-                    continue
-                if len(group.builds) == 1:
-                    build_nodes.setdefault(group.builds[0], []).append(node)
-                else:
-                    build_nodes.setdefault(group.builds[0], []).append(group.builds_by_id[int(value)])
-        return build_nodes
-
-    def save(self) -> list[OTAUpdate]:
-        updates = []
-        with transaction.atomic():
-            replaced_recipients = OTAUpdateRecipient.objects.filter(
-                node__in=chain(*self.selected_builds.values()),
-                status=OTARecipientStatus.RUNNING,
-            ).select_for_update()
-            replaced_recipients.update(status=OTARecipientStatus.REPLACED)
-            for build, nodes in self.selected_builds.items():
-                update = OTAUpdate.objects.create(build=build)
-                for node in nodes:
-                    update.recipients.create(node=node)
-                updates.append(update)
-        return updates