refactor: use get for accessing res fields

DataShades · Sep 5, 2023 · 044ac78 · 044ac78
1 parent 001c152
commit 044ac78
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ckanext/unfold/plugin.py b/ckanext/unfold/plugin.py
@@ -51,10 +51,10 @@ def form_template(self, context: Context, data_dict: DataDict) -> str:
     def before_resource_update(
         self, context: Context, current: dict[str, Any], resource: dict[str, Any]
     ) -> None:
-        if resource["url_type"] == "upload" and not resource["upload"]:
+        if resource.get("url_type") == "upload" and not resource.get("upload"):
             return
 
-        if resource["url_type"] == "url" and current["url"] == resource["url"]:
+        if resource.get("url_type") == "url" and current["url"] == resource["url"]:
             return
 
         unf_utils.delete_archive_structure(resource["id"])