fix: revert str UUID

This commit is contained in:
firedotguy
2026-02-06 22:19:56 +03:00
parent 2d27507338
commit f33ed4f76a
2 changed files with 45 additions and 107 deletions

View File

@@ -158,14 +158,14 @@ class Client:
return self.get_user('me') return self.get_user('me')
@refresh_on_error @refresh_on_error
def update_profile(self, username: str | None = None, display_name: str | None = None, bio: str | None = None, banner_id: UUID | str | None = None) -> UserProfileUpdate: def update_profile(self, username: str | None = None, display_name: str | None = None, bio: str | None = None, banner_id: UUID | None = None) -> UserProfileUpdate:
"""Обновить профиль """Обновить профиль
Args: Args:
username (str | None, optional): username. Defaults to None. username (str | None, optional): username. Defaults to None.
display_name (str | None, optional): Отображаемое имя. Defaults to None. display_name (str | None, optional): Отображаемое имя. Defaults to None.
bio (str | None, optional): Биография (о себе). Defaults to None. bio (str | None, optional): Биография (о себе). Defaults to None.
banner_id (UUID | str | None, optional): UUID баннера. Defaults to None. banner_id (UUID | None, optional): UUID баннера. Defaults to None.
Raises: Raises:
ValidationError: Ошибка валидации ValidationError: Ошибка валидации
@@ -173,9 +173,6 @@ class Client:
Returns: Returns:
UserProfileUpdate: Обновленный профиль UserProfileUpdate: Обновленный профиль
""" """
if isinstance(banner_id, str):
banner_id = UUID(banner_id)
res = update_profile(self.token, bio, display_name, username, banner_id) res = update_profile(self.token, bio, display_name, username, banner_id)
if res.status_code == 422 and 'found' in res.json(): if res.status_code == 422 and 'found' in res.json():
raise ValidationError(*list(res.json()['found'].items())[0]) raise ValidationError(*list(res.json()['found'].items())[0])
@@ -376,13 +373,14 @@ class Client:
@refresh_on_error @refresh_on_error
def add_comment(self, post_id: UUID | str, content: str, attachment_ids: list[UUID | str] = []) -> Comment: def add_comment(self, post_id: UUID, content: str, attachment_ids: list[UUID] = []) -> Comment:
"""Добавить комментарий """Добавить комментарий
Args: Args:
post_id (str | UUID): UUID поста post_id (str): UUID поста
content (str): Содержание content (str): Содержание
attachment_ids (list[UUID | str]): Список UUID прикреплённых файлов attachment_ids (list[UUID]): Список UUID прикреплённых файлов
reply_comment_id (UUID | None, optional): ID коммента для ответа. Defaults to None.
Raises: Raises:
ValidationError: Ошибка валидации ValidationError: Ошибка валидации
@@ -391,12 +389,7 @@ class Client:
Returns: Returns:
Comment: Комментарий Comment: Комментарий
""" """
if isinstance(post_id, str): res = add_comment(self.token, post_id, content, attachment_ids)
post_id = UUID(post_id)
attachment_ids = list(map(lambda id: UUID(id) if isinstance(id, str) else id, attachment_ids))
res = add_comment(self.token, post_id, content, cast(list[UUID], attachment_ids))
if res.status_code == 422 and 'found' in res.json(): if res.status_code == 422 and 'found' in res.json():
raise ValidationError(*list(res.json()['found'].items())[0]) raise ValidationError(*list(res.json()['found'].items())[0])
if res.json().get('error', {}).get('code') == 'NOT_FOUND': if res.json().get('error', {}).get('code') == 'NOT_FOUND':
@@ -407,14 +400,14 @@ class Client:
@refresh_on_error @refresh_on_error
def add_reply_comment(self, comment_id: UUID | str, content: str, author_id: UUID | str, attachment_ids: list[UUID | str] = []) -> Comment: def add_reply_comment(self, comment_id: UUID, content: str, author_id: UUID, attachment_ids: list[UUID] = []) -> Comment:
"""Добавить ответный комментарий """Добавить ответный комментарий
Args: Args:
comment_id (str | UUID): UUID комментария comment_id (str): UUID комментария
content (str): Содержание content (str): Содержание
author_id (UUID | str): ID пользователя, отправившего комментарий. author_id (UUID | None, optional): ID пользователя, отправившего комментарий. Defaults to None.
attachment_ids (list[UUID | str]): Список UUID прикреплённых файлов attachment_ids (list[UUID]): Список UUID прикреплённых файлов
Raises: Raises:
ValidationError: Ошибка валидации ValidationError: Ошибка валидации
@@ -423,13 +416,7 @@ class Client:
Returns: Returns:
Comment: Комментарий Comment: Комментарий
""" """
if isinstance(comment_id, str): res = add_reply_comment(self.token, comment_id, content, author_id, attachment_ids)
comment_id = UUID(comment_id)
if isinstance(author_id, str):
author_id = UUID(author_id)
attachment_ids = list(map(lambda id: UUID(id) if isinstance(id, str) else id, attachment_ids))
res = add_reply_comment(self.token, comment_id, content, author_id, cast(list[UUID], attachment_ids))
if res.status_code == 500 and 'Failed query' in res.text: if res.status_code == 500 and 'Failed query' in res.text:
raise NotFound('User') raise NotFound('User')
if res.status_code == 422 and 'found' in res.json(): if res.status_code == 422 and 'found' in res.json():
@@ -442,11 +429,11 @@ class Client:
@refresh_on_error @refresh_on_error
def get_comments(self, post_id: UUID | str, limit: int = 20, cursor: int = 0, sort: str = 'popular') -> tuple[list[Comment], Pagination]: def get_comments(self, post_id: UUID, limit: int = 20, cursor: int = 0, sort: str = 'popular') -> tuple[list[Comment], Pagination]:
"""Получить список комментариев """Получить список комментариев
Args: Args:
post_id (UUID | str): UUID поста post_id (UUID): UUID поста
limit (int, optional): Лимит. Defaults to 20. limit (int, optional): Лимит. Defaults to 20.
cursor (int, optional): Курсор (сколько пропустить). Defaults to 0. cursor (int, optional): Курсор (сколько пропустить). Defaults to 0.
sort (str, optional): Сортировка. Defaults to 'popular'. sort (str, optional): Сортировка. Defaults to 'popular'.
@@ -458,9 +445,6 @@ class Client:
list[Comment]: Список комментариев list[Comment]: Список комментариев
Pagination: Пагинация Pagination: Пагинация
""" """
if isinstance(post_id, str):
post_id = UUID(post_id)
res = get_comments(self.token, post_id, limit, cursor, sort) res = get_comments(self.token, post_id, limit, cursor, sort)
if res.json().get('error', {}).get('code') == 'NOT_FOUND': if res.json().get('error', {}).get('code') == 'NOT_FOUND':
raise NotFound('Post') raise NotFound('Post')
@@ -470,11 +454,11 @@ class Client:
return [Comment.model_validate(comment) for comment in data['comments']], Pagination(page=(cursor // limit) or 1, limit=limit, total=data['total'], hasMore=data['hasMore'], nextCursor=None) return [Comment.model_validate(comment) for comment in data['comments']], Pagination(page=(cursor // limit) or 1, limit=limit, total=data['total'], hasMore=data['hasMore'], nextCursor=None)
@refresh_on_error @refresh_on_error
def like_comment(self, id: UUID | str) -> int: def like_comment(self, id: UUID) -> int:
"""Лайкнуть комментарий """Лайкнуть комментарий
Args: Args:
id (UUID | str): UUID комментария id (UUID): UUID комментария
Raises: Raises:
NotFound: Комментарий не найден NotFound: Комментарий не найден
@@ -482,9 +466,6 @@ class Client:
Returns: Returns:
int: Количество лайков int: Количество лайков
""" """
if isinstance(id, str):
id = UUID(id)
res = like_comment(self.token, id) res = like_comment(self.token, id)
if res.json().get('error', {}).get('code') == 'NOT_FOUND': if res.json().get('error', {}).get('code') == 'NOT_FOUND':
raise NotFound('Comment') raise NotFound('Comment')
@@ -493,11 +474,11 @@ class Client:
return res.json()['likesCount'] return res.json()['likesCount']
@refresh_on_error @refresh_on_error
def unlike_comment(self, id: UUID | str) -> int: def unlike_comment(self, id: UUID) -> int:
"""Убрать лайк с комментария """Убрать лайк с комментария
Args: Args:
id (UUID | str): UUID комментария id (UUID): UUID комментария
Raises: Raises:
NotFound: Комментарий не найден NotFound: Комментарий не найден
@@ -505,9 +486,6 @@ class Client:
Returns: Returns:
int: Количество лайков int: Количество лайков
""" """
if isinstance(id, str):
id = UUID(id)
res = unlike_comment(self.token, id) res = unlike_comment(self.token, id)
if res.json().get('error', {}).get('code') == 'NOT_FOUND': if res.json().get('error', {}).get('code') == 'NOT_FOUND':
raise NotFound('Comment') raise NotFound('Comment')
@@ -516,19 +494,16 @@ class Client:
return res.json()['likesCount'] return res.json()['likesCount']
@refresh_on_error @refresh_on_error
def delete_comment(self, id: UUID | str) -> None: def delete_comment(self, id: UUID) -> None:
"""Удалить комментарий """Удалить комментарий
Args: Args:
id (UUID | str): UUID комментария id (UUID): UUID комментария
Raises: Raises:
NotFound: Комментарий не найден NotFound: Комментарий не найден
Forbidden: Нет прав на удаление Forbidden: Нет прав на удаление
""" """
if isinstance(id, str):
id = UUID(id)
res = delete_comment(self.token, id) res = delete_comment(self.token, id)
if res.status_code == 204: if res.status_code == 204:
return return
@@ -567,22 +542,19 @@ class Client:
return [Hashtag.model_validate(hashtag) for hashtag in res.json()['data']['hashtags']] return [Hashtag.model_validate(hashtag) for hashtag in res.json()['data']['hashtags']]
@refresh_on_error @refresh_on_error
def get_posts_by_hashtag(self, hashtag: str, limit: int = 20, cursor: UUID | str | None = None) -> tuple[Hashtag | None, list[Post], Pagination]: def get_posts_by_hashtag(self, hashtag: str, limit: int = 20, cursor: UUID | None = None) -> tuple[Hashtag | None, list[Post], Pagination]:
"""Получить посты по хэштэгу """Получить посты по хэштэгу
Args: Args:
hashtag (str): Хэштэг (без #) hashtag (str): Хэштэг (без #)
limit (int, optional): Лимит. Defaults to 20. limit (int, optional): Лимит. Defaults to 20.
cursor (UUID | str | None, optional): Курсор (UUID последнего поста, после которого брать данные). Defaults to None. cursor (UUID | None, optional): Курсор (UUID последнего поста, после которого брать данные). Defaults to None.
Returns: Returns:
Hashtag | None: Хэштэг Hashtag | None: Хэштэг
list[Post]: Посты list[Post]: Посты
Pagination: Пагинация Pagination: Пагинация
""" """
if isinstance(cursor, str):
cursor = UUID(cursor)
res = get_posts_by_hashtag(self.token, hashtag, limit, cursor) res = get_posts_by_hashtag(self.token, hashtag, limit, cursor)
res.raise_for_status() res.raise_for_status()
data = res.json()['data'] data = res.json()['data']
@@ -611,18 +583,15 @@ class Client:
) )
@refresh_on_error @refresh_on_error
def mark_as_read(self, id: UUID | str) -> bool: def mark_as_read(self, id: UUID) -> bool:
"""Прочитать уведомление """Прочитать уведомление
Args: Args:
id (UUID | str): UUID уведомления id (UUID): UUID уведомления
Returns: Returns:
bool: Успешно (False - уже прочитано) bool: Успешно (False - уже прочитано)
""" """
if isinstance(id, str):
id = UUID(id)
res = mark_as_read(self.token, id) res = mark_as_read(self.token, id)
res.raise_for_status() res.raise_for_status()
@@ -649,13 +618,13 @@ class Client:
@refresh_on_error @refresh_on_error
def create_post(self, content: str, wall_recipient_id: UUID | str | None = None, attachment_ids: list[UUID | str] = []) -> NewPost: def create_post(self, content: str, wall_recipient_id: UUID | None = None, attach_ids: list[UUID] = []) -> NewPost:
"""Создать пост """Создать пост
Args: Args:
content (str): Содержимое content (str): Содержимое
wall_recipient_id (UUID | str | None, optional): UUID пользователя (чтобы создать пост ему на стене). Defaults to None. wall_recipient_id (UUID | None, optional): UUID пользователя (чтобы создать пост ему на стене). Defaults to None.
attachment_ids (list[UUID | str], optional): UUID вложений. Defaults to []. attach_ids (list[UUID], optional): UUID вложений. Defaults to [].
Raises: Raises:
NotFound: Пользователь не найден NotFound: Пользователь не найден
@@ -664,11 +633,7 @@ class Client:
Returns: Returns:
NewPost: Новый пост NewPost: Новый пост
""" """
if isinstance(wall_recipient_id, str): res = create_post(self.token, content, wall_recipient_id, attach_ids)
wall_recipient_id = UUID(wall_recipient_id)
attachment_ids = list(map(lambda id: UUID(id) if isinstance(id, str) else id, attachment_ids))
res = create_post(self.token, content, wall_recipient_id, cast(list[UUID], attachment_ids))
if res.json().get('error', {}).get('code') == 'NOT_FOUND': if res.json().get('error', {}).get('code') == 'NOT_FOUND':
raise NotFound('Wall recipient') raise NotFound('Wall recipient')
if res.status_code == 422 and 'found' in res.json(): if res.status_code == 422 and 'found' in res.json():
@@ -696,11 +661,11 @@ class Client:
return [Post.model_validate(post) for post in data['posts']], PostsPagintaion.model_validate(data['pagination']) return [Post.model_validate(post) for post in data['posts']], PostsPagintaion.model_validate(data['pagination'])
@refresh_on_error @refresh_on_error
def get_post(self, id: UUID | str) -> Post: def get_post(self, id: UUID) -> Post:
"""Получить пост """Получить пост
Args: Args:
id (UUID | str): UUID поста id (UUID): UUID поста
Raises: Raises:
NotFound: Пост не найден NotFound: Пост не найден
@@ -708,9 +673,6 @@ class Client:
Returns: Returns:
Post: Пост Post: Пост
""" """
if isinstance(id, str):
id = UUID(id)
res = get_post(self.token, id) res = get_post(self.token, id)
if res.json().get('error', {}).get('code') == 'NOT_FOUND': if res.json().get('error', {}).get('code') == 'NOT_FOUND':
raise NotFound('Post') raise NotFound('Post')
@@ -719,11 +681,11 @@ class Client:
return Post.model_validate(res.json()['data']) return Post.model_validate(res.json()['data'])
@refresh_on_error @refresh_on_error
def edit_post(self, id: UUID | str, content: str) -> str: def edit_post(self, id: UUID, content: str) -> str:
"""Редактировать пост """Редактировать пост
Args: Args:
id (UUID | str): UUID поста id (UUID): UUID поста
content (str): Содержимое content (str): Содержимое
Raises: Raises:
@@ -734,9 +696,6 @@ class Client:
Returns: Returns:
str: Новое содержимое str: Новое содержимое
""" """
if isinstance(id, str):
id = UUID(id)
res = edit_post(self.token, id, content) res = edit_post(self.token, id, content)
if res.json().get('error', {}).get('code') == 'NOT_FOUND': if res.json().get('error', {}).get('code') == 'NOT_FOUND':
@@ -750,19 +709,16 @@ class Client:
return res.json()['content'] return res.json()['content']
@refresh_on_error @refresh_on_error
def delete_post(self, id: UUID | str) -> None: def delete_post(self, id: UUID) -> None:
"""Удалить пост """Удалить пост
Args: Args:
id (UUID | str): UUID поста id (UUID): UUID поста
Raises: Raises:
NotFound: Пост не найден NotFound: Пост не найден
Forbidden: Нет доступа Forbidden: Нет доступа
""" """
if isinstance(id, str):
id = UUID(id)
res = delete_post(self.token, id) res = delete_post(self.token, id)
if res.status_code == 204: if res.status_code == 204:
return return
@@ -774,19 +730,16 @@ class Client:
res.raise_for_status() res.raise_for_status()
@refresh_on_error @refresh_on_error
def pin_post(self, id: UUID | str): def pin_post(self, id: UUID):
"""Закрепить пост """Закрепить пост
Args: Args:
id (UUID | str): UUID поста id (UUID): UUID поста
Raises: Raises:
NotFound: Пост не найден NotFound: Пост не найден
Forbidden: Нет доступа Forbidden: Нет доступа
""" """
if isinstance(id, str):
id = UUID(id)
res = pin_post(self.token, id) res = pin_post(self.token, id)
if res.json().get('error', {}).get('code') == 'NOT_FOUND': if res.json().get('error', {}).get('code') == 'NOT_FOUND':
@@ -796,11 +749,11 @@ class Client:
res.raise_for_status() res.raise_for_status()
@refresh_on_error @refresh_on_error
def repost(self, id: UUID | str, content: str | None = None) -> NewPost: def repost(self, id: UUID, content: str | None = None) -> NewPost:
"""Репостнуть пост """Репостнуть пост
Args: Args:
id (UUID | str): UUID поста id (UUID): UUID поста
content (str | None, optional): Содержимое (доп. комментарий). Defaults to None. content (str | None, optional): Содержимое (доп. комментарий). Defaults to None.
Raises: Raises:
@@ -812,9 +765,6 @@ class Client:
Returns: Returns:
NewPost: Новый пост NewPost: Новый пост
""" """
if isinstance(id, str):
id = UUID(id)
res = repost(self.token, id, content) res = repost(self.token, id, content)
if res.json().get('error', {}).get('code') == 'NOT_FOUND': if res.json().get('error', {}).get('code') == 'NOT_FOUND':
@@ -830,18 +780,15 @@ class Client:
return NewPost.model_validate(res.json()) return NewPost.model_validate(res.json())
@refresh_on_error @refresh_on_error
def view_post(self, id: UUID | str) -> None: def view_post(self, id: UUID) -> None:
"""Просмотреть пост """Просмотреть пост
Args: Args:
id (UUID | str): UUID поста id (UUID): UUID поста
Raises: Raises:
NotFound: Пост не найден NotFound: Пост не найден
""" """
if isinstance(id, str):
id = UUID(id)
res = view_post(self.token, id) res = view_post(self.token, id)
if res.status_code == 204: if res.status_code == 204:
return return
@@ -861,27 +808,19 @@ class Client:
@refresh_on_error @refresh_on_error
def report(self, id: str | UUID, type: str = 'post', reason: str = 'other', description: str = ''): def report(self, id: str, type: str = 'post', reason: str = 'other', description: str = ''):
if isinstance(id, str):
id = UUID(id)
return report(self.token, id, type, reason, description) return report(self.token, id, type, reason, description)
@refresh_on_error @refresh_on_error
def report_user(self, id: str | UUID, reason: str = 'other', description: str = ''): def report_user(self, id: str, reason: str = 'other', description: str = ''):
if isinstance(id, str):
id = UUID(id)
return report(self.token, id, 'user', reason, description) return report(self.token, id, 'user', reason, description)
@refresh_on_error @refresh_on_error
def report_post(self, id: str | UUID, reason: str = 'other', description: str = ''): def report_post(self, id: str, reason: str = 'other', description: str = ''):
if isinstance(id, str):
id = UUID(id)
return report(self.token, id, 'post', reason, description) return report(self.token, id, 'post', reason, description)
@refresh_on_error @refresh_on_error
def report_comment(self, id: str | UUID, reason: str = 'other', description: str = ''): def report_comment(self, id: str, reason: str = 'other', description: str = ''):
if isinstance(id, str):
id = UUID(id)
return report(self.token, id, 'comment', reason, description) return report(self.token, id, 'comment', reason, description)

View File

@@ -1,5 +1,4 @@
from itd.request import fetch from itd.request import fetch
from uuid import UUID
def report(token: str, id: UUID, type: str = 'post', reason: str = 'other', description: str = ''): def report(token: str, id: str, type: str = 'post', reason: str = 'other', description: str = ''):
return fetch(token, 'post', 'reports', {'targetId': id, 'targetType': type, 'reason': reason, 'description': description}) return fetch(token, 'post', 'reports', {'targetId': id, 'targetType': type, 'reason': reason, 'description': description})