Skip to content

Commit 0a41a48

Browse files
committed
update
1 parent b265b00 commit 0a41a48

1 file changed

Lines changed: 16 additions & 9 deletions

File tree

backend/data_wizard/views.py

Lines changed: 16 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1819,7 +1819,7 @@ def prepare_create(
18191819
) -> tuple[dict, Optional[Error]]:
18201820
domain = self.folder_id
18211821
domain_name = record.get("domain")
1822-
if not domain_name:
1822+
if domain_name:
18231823
domain = self.folders_map.get(str(domain_name).lower(), self.folder_id)
18241824

18251825
name = record.get("name")
@@ -1866,6 +1866,12 @@ class ProcessingRecordConsumer(RecordConsumer):
18661866
"filtering_labels": ["labels"],
18671867
}
18681868
)
1869+
_STATUS_KEYS: ClassVar[frozenset[str]] = frozenset(
1870+
k for k, _ in Processing.STATUS_CHOICES
1871+
)
1872+
_STATUS_BY_DISPLAY: ClassVar[Mapping[str, str]] = MappingProxyType(
1873+
{v.lower(): k for k, v in Processing.STATUS_CHOICES}
1874+
)
18691875

18701876
def _build_update_data(self, record: dict, record_data: dict) -> dict:
18711877
update_data = super()._build_update_data(record, record_data)
@@ -1881,20 +1887,21 @@ def prepare_create(
18811887
) -> tuple[dict, Optional[Error]]:
18821888
domain = self.folder_id
18831889
domain_name = record.get("domain")
1884-
if not domain_name:
1890+
if domain_name:
18851891
domain = self.folders_map.get(str(domain_name).lower(), self.folder_id)
18861892

18871893
name = record.get("name")
18881894
if not name:
18891895
return {}, Error(record=record, error="Name field is mandatory")
18901896

1891-
# Accept display value or raw key for status
1892-
status_mapping = {v: k for k, v in Processing.STATUS_CHOICES}
1893-
record_status_value = record.get("status", "privacy_draft")
1894-
status_value = None
1895-
1896-
if record_status_value in status_mapping:
1897-
status_value = status_mapping[record_status_value]
1897+
# Accept both raw key (e.g. "privacy_draft") and display label (e.g. "Draft")
1898+
record_status_value = record.get("status", "")
1899+
if record_status_value in self._STATUS_KEYS:
1900+
status_value = record_status_value
1901+
elif str(record_status_value).lower() in self._STATUS_BY_DISPLAY:
1902+
status_value = self._STATUS_BY_DISPLAY[str(record_status_value).lower()]
1903+
else:
1904+
status_value = "privacy_draft"
18981905

18991906
data = {
19001907
"name": name,

0 commit comments

Comments
 (0)