feat: add models and enum

itd/models/_text.py

@@ -0,0 +1,17 @@
+from uuid import UUID
+from datetime import datetime
+
+from pydantic import BaseModel, Field
+
+from itd.models.user import UserPost
+
+
+class _TextObject(BaseModel):
+    id: UUID
+    content: str
+    author: UserPost
+    attachments: list[UUID]
+
+    created_at: datetime = Field(alias='createdAt')
+
+    model_config = {'populate_by_name': True}

itd/models/comment.py

@@ -0,0 +1,14 @@
+from uuid import UUID
+from datetime import datetime
+
+from pydantic import Field
+
+from itd.models._text import _TextObject
+
+
+class CommentShort(_TextObject):
+    likes_count: int = Field(0, alias='likesCount')
+    replies_count: int = Field(0, alias='repliesCount')
+    is_liked: bool = Field(False, alias='isLiked')
+
+    replies: list['CommentShort'] = []

itd/models/file.py

@@ -0,0 +1,10 @@
+from uuid import UUID
+
+from pydantic import BaseModel, Field
+
+class File(BaseModel):
+    id: UUID
+    url: str
+    filename: str
+    mime_type: str = Field('image/png', alias='mimeType')
+    size: int

itd/models/hashtag.py

@@ -0,0 +1,8 @@
+from uuid import UUID
+
+from pydantic import BaseModel, Field
+
+class Hashtag(BaseModel):
+    id: UUID
+    name: str
+    posts_count: int = Field(0, alias='postsCount')

itd/models/notification.py

@@ -0,0 +1,22 @@
+from uuid import UUID
+from datetime import datetime
+
+from pydantic import BaseModel, Field
+
+from itd.enums import NotificationType, NotificationTargetType
+from itd.models.user import UserNotification
+
+class Notification(BaseModel):
+    id: UUID
+    type: NotificationType
+
+    target_type: NotificationTargetType | None = Field(None, alias='targetType') # none - follows, other - NotificationTragetType.POST
+    target_id: int | None = Field(None, alias='targetId') # none - follows
+
+    preview: str | None = None # follow - none, comment/reply - content, repost - original post content, like - post content, wall_post - wall post content
+
+    read: bool = False
+    read_at: datetime | None = Field(None, alias='readAt')
+    created_at: datetime = Field(alias='createdAt')
+
+    actor: UserNotification

itd/models/post.py

@@ -0,0 +1,29 @@
+from pydantic import Field
+
+from itd.models.user import UserPost
+from itd.models._text import _TextObject
+
+
+class PostShort(_TextObject):
+    likes_count: int = Field(0, alias='likesCount')
+    comments_count: int = Field(0, alias='commentsCount')
+    reposts_count: int = Field(0, alias='repostsCount')
+    views_count: int = Field(0, alias='viewsCount')
+
+
+class OriginalPost(PostShort):
+    is_deleted: bool = Field(False, alias='isDeleted')
+
+
+class Post(PostShort):
+    is_liked: bool = Field(False, alias='isLiked')
+    is_reposted: bool = Field(False, alias='isReposted')
+    is_viewed: bool = Field(False, alias='isViewed')
+    is_owner: bool = Field(False, alias='isOwner')
+
+    comments: list = []
+
+    original_post: OriginalPost | None = None
+
+    wall_recipient_id: int | None = None
+    wall_recipient: UserPost | None = None

itd/models/report.py

@@ -0,0 +1,16 @@
+from uuid import UUID
+from datetime import datetime
+
+from pydantic import BaseModel, Field
+
+from itd.enums import ReportTargetType, ReportTargetReason
+
+class Report(BaseModel):
+    id: UUID
+    reason: ReportTargetReason
+    description: str | None = None
+
+    target_type: ReportTargetType = Field(alias='targetType')
+    target_id: UUID
+
+    created_at: datetime = Field(alias='createdAt')

itd/models/user.py

@@ -0,0 +1,37 @@
+from uuid import UUID
+from datetime import datetime
+
+from pydantic import BaseModel, Field
+
+class UserNotification(BaseModel):
+    id: UUID
+    username: str
+    display_name: str = Field(alias='displayName')
+    avatar: str
+
+    model_config = {'populate_by_name': True}
+
+
+class UserPost(UserNotification):
+    verified: bool = False
+
+
+class UserSearch(UserPost):
+    followers_count: int = Field(0, alias='followersCount')
+
+
+class User(UserSearch):
+    banner: str | None = None
+    bio: str | None = None
+    pinned_post_id: UUID | None
+
+    private: bool | None = Field(None, alias='isPrivate') # none for not me
+    wall_closed: bool = Field(False, alias='wallClosed')
+
+    following_count: int = Field(0, alias='followingCount')
+    posts_count: int = Field(0, alias='postsCount')
+
+    is_following: bool | None = Field(None, alias='isFollowing') # none for me
+    is_followed: bool | None = Field(None, alias='isFollowedBy') # none for me
+
+    created_at: datetime = Field(alias='createdAt')

itd/models/verification.py

@@ -0,0 +1,23 @@
+from uuid import UUID
+from datetime import datetime
+
+from pydantic import BaseModel, Field
+
+class Verification(BaseModel):
+    id: UUID
+    user_id: UUID = Field(alias='userId')
+    video_url: str = Field(alias='videoUrl')
+    status: str # should be enum, but we dont know all statuses (what status for accepted?)
+
+    reject_reason: str | None = Field(None, alias='rejectionReason')
+    reviewer: str | None = Field(None, alias='reviewedBy')
+    reviewed_at: datetime | None = Field(None, alias='reviewedAt')
+
+    created_at: datetime = Field(alias='createdAt')
+    updated_at: datetime = Field(alias='updatedAt')
+
+
+class VerificationStatus(BaseModel):
+    status: str # should be enum, but we dont know all statuses (what status for accepted?)
+    request_id: UUID = Field(alias='requestId')
+    submitted_at: datetime = Field(alias='submittedAt')