Browse Source

Switch to .save() for updating links

Trevor Elliott 4 years ago
parent
commit
b8c15098fa
1 changed files with 11 additions and 6 deletions
  1. 11 6
      lc/model.py

+ 11 - 6
lc/model.py

@@ -145,6 +145,11 @@ class Link(Model):
     def link_url(self) -> str:
         return f"/u/{self.user.name}/l/{self.id}"
 
+    @staticmethod
+    def by_id(id: int) -> Optional["Link"]:
+        return Link.get_or_none(id=id)
+
+
     @staticmethod
     def from_request(user: User, link: r.Link) -> "Link":
         l = Link.create(
@@ -163,12 +168,12 @@ class Link(Model):
         return l
 
     def update_from_request(self, link: r.Link):
-        Link.update(
-            url=link.url,
-            name=link.name,
-            description=link.description,
-            private=link.private,
-        ).where(Link.id == self.id).execute()
+        self.url = link.url
+        self.name = link.name
+        self.description = link.description
+        self.private = link.private
+        self.tags = link.tags
+        self.save()
 
 
 class Tag(Model):