Move setting leaf title from CLI to model

src/safaribookmarks/cli.py

@@ -40,10 +40,6 @@ def with_bookmarks(
     def lookup(self, title: str, root: ChildrenType) -> Optional[ChildrenType]:
         if title.lower() == str(root.web_bookmark_uuid).lower():
             return root
-        elif isinstance(root, WebBookmarkTypeLeaf):
-            leaf = cast(WebBookmarkTypeLeaf, root)
-            if leaf.uri_dictionary.get("title") == title:
-                return leaf
         elif getattr(root, "title", None) == title:
             return root
         elif isinstance(root, WebBookmarkTypeList):
@@ -185,12 +181,7 @@ def edit(self, args):
             if target is None:
                 raise ValueError("Target not found")
             if title := args.title:
-                if isinstance(target, WebBookmarkTypeList):
-                    target.title = title
-                elif isinstance(target, WebBookmarkTypeLeaf):
-                    target.uri_dictionary["title"] = title
-                else:
-                    raise ValueError("Cannot update target title")
+                target.title = title
             if url := args.url:
                 if isinstance(target, WebBookmarkTypeLeaf):
                     target.url_string = url

src/safaribookmarks/models.py

@@ -33,6 +33,14 @@ class WebBookmarkTypeLeaf(WebBookmarkType):
     url_string: str = Field(alias="URLString")
     uri_dictionary: dict[str, str] = Field(alias="URIDictionary", default_factory=dict)
 
+    @property
+    def title(self) -> str:
+        return self.uri_dictionary.get("title", "")
+
+    @title.setter
+    def title(self, value) -> None:
+        self.uri_dictionary["title"] = value
+
 
 class WebBookmarkTypeList(WebBookmarkType):
     web_bookmark_type: Literal["WebBookmarkTypeList"] = Field(

tests/support/fixtures/list-leaf.txt

@@ -0,0 +1 @@
+Safari Bookmarks CLI                              leaf  B441CA58-1880-4151-929E-743090B66587  https://github.com/evilmarty/safari-bookmarks-cli