chore: remove_dict_emptyで特定のkeyをignoreできるように

pull/105/head
yupix 10 months ago
parent 10c8c8d698
commit c2cac8f6d5
No known key found for this signature in database
GPG Key ID: 2FF705F5C56D9C06

@ -55,12 +55,16 @@ def remove_list_empty(data: list[Any]) -> list[Any]:
return [k for k in data if k] return [k for k in data if k]
def remove_dict_empty(data: dict[str, Any]) -> dict[str, Any]: def remove_dict_empty(
data: dict[str, Any], ignore_keys: list[str] | None = None
) -> dict[str, Any]:
""" """
Parameters Parameters
---------- ----------
data: dict data: dict
空のkeyを削除したいdict 空のkeyを削除したいdict
ignore_keys: list
削除したくないkeyのリスト
Returns Returns
------- -------
@ -68,7 +72,10 @@ def remove_dict_empty(data: dict[str, Any]) -> dict[str, Any]:
空のkeyがなくなったdict 空のkeyがなくなったdict
""" """
_data = {} _data = {}
_data = {k: v for k, v in data.items() if v is not None} if ignore_keys is None:
ignore_keys = []
_data = {k: v for k, v in data.items() if v is not None or k in ignore_keys}
return _data
return _data return _data

Loading…
Cancel
Save