Handle both Image and Document types

2021-04-17 11:47:48 -07:00
parent 239d927e94
commit 7c85b554cf
3 changed files with 9 additions and 4 deletions
--- a/bookwyrm/activitypub/image.py
+++ b/bookwyrm/activitypub/image.py
@ -5,9 +5,14 @@ from .base_activity import ActivityObject

@dataclass(init=False)
 class Document(ActivityObject):
-    """ image block """
+    """ a document """

    url: str
    name: str = ""
    type: str = "Document"
    id: str = None
+
+@dataclass(init=False)
+class Image(Document):
+    """ an image """
+    type: str = "Image"