#56 Add custom dictionary factory

azely/utils.py

@@ -43,7 +43,8 @@ def wrapper(*args: Any, **kwargs: Any) -> Any:
             tab = doc.setdefault(table, {})
 
             if update or (query not in tab):
-                tab[query] = asdict(func(*args, **kwargs))
+                obj = func(*args, **kwargs)
+                tab[query] = asdict(obj, dict_factory=dropna)
 
                 if tab is not doc.last_item():
                     tab.add(nl())
@@ -53,6 +54,11 @@ def wrapper(*args: Any, **kwargs: Any) -> Any:
     return wrapper  # type: ignore
 
 
+def dropna(items: list[tuple[str, Any]]) -> dict[str, Any]:
+    """Dictionary factory that drops N/A items."""
+    return dict((k, v) for k, v in items if v is not None)
+
+
 def rename(func: TCallable, key: str) -> TCallable:
     """Update the name field of a dataclass object."""
     signature = Signature.from_callable(func)