feat(captcha): add validation and thread editing

modules/lolz.py

@@ -9,16 +9,19 @@ logger = logging.getLogger(__name__)
 
 class LolzAPIError(Exception):
     """Base exception for Lolz API errors"""
+
     pass
 
 
 class LolzNetworkError(LolzAPIError):
     """Network-related errors"""
+
     pass
 
 
 class LolzResponseError(LolzAPIError):
     """API returned invalid or empty response"""
+
     pass
 
 
@@ -45,17 +48,68 @@ class Lolz:
             thread = (response.json()).get("thread", {})
 
             if not thread:
-                raise LolzResponseError(f"Тема {thread_id} не найдена или вернула пустой ответ")
+                raise LolzResponseError(
+                    f"Тема {thread_id} не найдена или вернула пустой ответ"
+                )
 
             return thread
 
         except asyncio.TimeoutError as e:
-            logger.error(f"Превышено время ожидания при получении темы {thread_id}: {e}")
-            raise LolzNetworkError(f"Превышено время ожидания при получении темы {thread_id}") from e
+            logger.error(
+                f"Превышено время ожидания при получении темы {thread_id}: {e}"
+            )
+            raise LolzNetworkError(
+                f"Превышено время ожидания при получении темы {thread_id}"
+            ) from e
         except Exception as e:
             logger.error(f"Ошибка при получении темы {thread_id}: {e}")
             raise LolzAPIError(f"Не удалось получить тему {thread_id}") from e
 
+    async def edit_thread(
+        self, thread_id: Union[str, int], post_body: str = None, **kwargs
+    ) -> bool:
+        logger.debug(f"Редактирую данные темы {thread_id}...")
+        try:
+            if post_body:
+                thread = await self.client.threads.get(thread_id=thread_id)
+
+                # Чтобы отредактировать соддержимое темы - надо отредактировать первый пост
+                post = thread.get("first_post", {})
+                post_id = post.get("post_id")
+
+                response = await self.client.posts.edit(
+                    post_id=post_id, post_body=post_body
+                )
+
+                if response.status_code != 200:
+                    logger.error(
+                        f"Не удалось отредактировать пост {post_id}. Статус: {response.status_code}"
+                    )
+                    return False
+
+            response = await self.client.threads.edit(thread_id=thread_id, **kwargs)
+
+            if response.status_code == 200:
+                logger.info(f"Тема {thread_id} успешно отредактирована.")
+                return True
+            else:
+                logger.error(
+                    f"Не удалось отредактировать тему {thread_id}. Статус: {response.status_code}"
+                )
+                return False
+
+        except asyncio.TimeoutError as e:
+            logger.error(
+                f"Превышено время ожидания при редактировании темы {thread_id}: {e}"
+            )
+            raise LolzNetworkError(
+                f"Превышено время ожидания при редактировании темы {thread_id}"
+            ) from e
+
+        except Exception as e:
+            logger.error(f"Ошибка при редактировании темы {thread_id}: {e}")
+            raise LolzAPIError(f"Не удалось отредактировать тему {thread_id}") from e
+
     async def can_bump(
         self, thread_id: Optional[int] = None, thread: Dict[str, Any] = None
     ) -> bool:
@@ -79,12 +133,16 @@ class Lolz:
                 logger.info(f"Тема {thread_id} успешно поднята.")
                 return True
             else:
-                logger.error(f"Не удалось поднять тему {thread_id}. Статус: {response.status_code}")
+                logger.error(
+                    f"Не удалось поднять тему {thread_id}. Статус: {response.status_code}"
+                )
                 return False
 
         except asyncio.TimeoutError as e:
             logger.error(f"Превышено время ожидания при поднятии темы {thread_id}: {e}")
-            raise LolzNetworkError(f"Превышено время ожидания при поднятии темы {thread_id}") from e
+            raise LolzNetworkError(
+                f"Превышено время ожидания при поднятии темы {thread_id}"
+            ) from e
         except Exception as e:
             logger.error(f"Ошибка при поднятии темы {thread_id}: {e}")
             raise LolzAPIError(f"Не удалось поднять тему {thread_id}") from e
@@ -96,13 +154,17 @@ class Lolz:
             post = (response.json()).get("post", {})
 
             if not post:
-                raise LolzResponseError(f"Пост {post_id} не найден или вернул пустой ответ")
+                raise LolzResponseError(
+                    f"Пост {post_id} не найден или вернул пустой ответ"
+                )
 
             return post
 
         except asyncio.TimeoutError as e:
             logger.error(f"Превышено время ожидания при получении поста {post_id}: {e}")
-            raise LolzNetworkError(f"Превышено время ожидания при получении поста {post_id}") from e
+            raise LolzNetworkError(
+                f"Превышено время ожидания при получении поста {post_id}"
+            ) from e
         except Exception as e:
             logger.error(f"Ошибка при получении поста {post_id}: {e}")
             raise LolzAPIError(f"Не удалось получить пост {post_id}") from e
@@ -137,11 +199,19 @@ class Lolz:
                 await asyncio.sleep(1)
 
             except asyncio.TimeoutError as e:
-                logger.error(f"Превышено время ожидания при получении постов из темы {thread_id}, страница {page}: {e}")
-                raise LolzNetworkError(f"Превышено время ожидания при получении постов из темы {thread_id}") from e
+                logger.error(
+                    f"Превышено время ожидания при получении постов из темы {thread_id}, страница {page}: {e}"
+                )
+                raise LolzNetworkError(
+                    f"Превышено время ожидания при получении постов из темы {thread_id}"
+                ) from e
             except Exception as e:
-                logger.error(f"Ошибка при получении постов из темы {thread_id}, страница {page}: {e}")
-                raise LolzAPIError(f"Не удалось получить посты из темы {thread_id}") from e
+                logger.error(
+                    f"Ошибка при получении постов из темы {thread_id}, страница {page}: {e}"
+                )
+                raise LolzAPIError(
+                    f"Не удалось получить посты из темы {thread_id}"
+                ) from e
 
         if all_posts:
             logger.info(f"Всего получено {len(all_posts)} постов из темы {thread_id}.")
@@ -165,11 +235,17 @@ class Lolz:
             return comments
 
         except asyncio.TimeoutError as e:
-            logger.error(f"Превышено время ожидания при получении комментариев к посту {post_id}: {e}")
-            raise LolzNetworkError(f"Превышено время ожидания при получении комментариев к посту {post_id}") from e
+            logger.error(
+                f"Превышено время ожидания при получении комментариев к посту {post_id}: {e}"
+            )
+            raise LolzNetworkError(
+                f"Превышено время ожидания при получении комментариев к посту {post_id}"
+            ) from e
         except Exception as e:
             logger.error(f"Ошибка при получении комментариев к посту {post_id}: {e}")
-            raise LolzAPIError(f"Не удалось получить комментарии к посту {post_id}") from e
+            raise LolzAPIError(
+                f"Не удалось получить комментарии к посту {post_id}"
+            ) from e
 
     async def has_comments(
         self, post_id: Optional[int] = None, post: Dict[str, Any] = None
@@ -192,12 +268,21 @@ class Lolz:
                 logger.info(f"Комментарий к посту {post_id} успешно опубликован.")
                 return True
             else:
-                logger.error(f"Не удалось опубликовать комментарий к посту {post_id}. Статус: {response.status_code}")
+                logger.error(
+                    f"Не удалось опубликовать комментарий к посту {post_id}. Статус: {response.status_code}"
+                )
                 return False
 
         except asyncio.TimeoutError as e:
-            logger.error(f"Превышено время ожидания при создании комментария к посту {post_id}: {e}")
-            raise LolzNetworkError(f"Превышено время ожидания при создании комментария к посту {post_id}") from e
+            logger.error(
+                f"Превышено время ожидания при создании комментария к посту {post_id}: {e}"
+            )
+            raise LolzNetworkError(
+                f"Превышено время ожидания при создании комментария к посту {post_id}"
+            ) from e
+
         except Exception as e:
             logger.error(f"Ошибка при создании комментария к посту {post_id}: {e}")
-            raise LolzAPIError(f"Не удалось создать комментарий к посту {post_id}") from e
+            raise LolzAPIError(
+                f"Не удалось создать комментарий к посту {post_id}"
+            ) from e