Rubka 6.6.2__py3-none-any.whl → 7.1.17__py3-none-any.whl

rubka/exceptions.py

@@ -1,3 +1,37 @@
 class APIRequestError(Exception):
-    """Raised when an API request fails."""
+    """Base class for all API request errors."""
+    def __init__(self, status: str, dev_message: str = None):
+        self.status = status
+        self.dev_message = dev_message
+        msg = f"{self.__class__.__name__}: {status}"
+        if dev_message:
+            msg += f" | Message : {dev_message}"
+        super().__init__(msg)
+
+
+class InvalidAccessError(APIRequestError):
+    """Raised when access is invalid."""
+
+class InvalidTokenError(Exception):
     pass
+
+class InvalidInputError(APIRequestError):
+    """Raised when input is invalid."""
+
+
+class TooRequestError(APIRequestError):
+    """Raised when too many requests are made."""
+
+
+def raise_for_status(response: dict):
+    status = response.get("status")
+    dev_message = response.get("dev_message")
+
+    if status == "INVALID_ACCESS":
+        raise InvalidAccessError(status, dev_message)
+    elif status == "INVALID_INPUT":
+        raise InvalidInputError(status, dev_message)
+    elif status == "TOO_REQUEST":
+        raise TooRequestError(status, dev_message)
+    elif status != "OK":
+        raise APIRequestError(status, dev_message)

rubka/filters.py

@@ -110,6 +110,9 @@ def text_length(min_len: int = 0, max_len: int = None):
 def text_regex(pattern: str):
     regex = re.compile(pattern)
     return Filter(lambda m: getattr(m, "text", "") and regex.search(m.text))
+def regex(pattern: str):
+    regex = re.compile(pattern)
+    return Filter(lambda m: getattr(m, "text", "") and regex.search(m.text))
 def text_startswith(prefix: str):
     return Filter(lambda m: getattr(m, "text", "").startswith(prefix) if getattr(m, "text", None) else False)
 def text_endswith(suffix: str):
@@ -149,6 +152,19 @@ def sticker_id_is(sid: str):
     return Filter(lambda m: m.sticker and getattr(m.sticker, "sticker_id", None) == sid)
 def sticker_emoji_is(emoji: str):
     return Filter(lambda m: m.sticker and getattr(m.sticker, "emoji", None) == emoji)
+is_bold = Filter(lambda m: getattr(m, "is_bold", False))
+is_italic = Filter(lambda m: getattr(m, "is_italic", False))
+is_strike = Filter(lambda m: getattr(m, "is_strike", False))
+is_underline = Filter(lambda m: getattr(m, "is_underline", False))
+is_quote = Filter(lambda m: getattr(m, "is_quote", False))
+is_spoiler = Filter(lambda m: getattr(m, "is_spoiler", False))
+is_pre = Filter(lambda m: getattr(m, "is_pre", False))
+is_mono = Filter(lambda m: getattr(m, "is_mono", False))
+is_link_meta = Filter(lambda m: getattr(m, "is_link_meta", False))
+has_metadata = Filter(lambda m: getattr(m, "has_metadata", False))
+meta_links_contain = lambda keyword: Filter(lambda m: any(keyword in link for link in getattr(m, "meta_links", [])))
+meta_link_positions_contain = lambda keyword: Filter(lambda m: any(keyword in link.get("url", "") for link in getattr(m, "meta_link_positions", [])))
+meta_types_include = lambda types: Filter(lambda m: any(t in getattr(m, "meta_types", []) for t in types))
 def poll_question_contains(keyword: str):
     return Filter(lambda m: m.poll and keyword in getattr(m.poll, "question", ""))
 def poll_option_count(min_options: int = 1, max_options: int = None):