From 6b80732bf6b9f7b6091c51b4734ff1c1655a6521 Mon Sep 17 00:00:00 2001
From: AYMENJD <53928879+AYMENJD@users.noreply.github.com>
Date: Mon, 13 Jan 2025 07:25:05 +0300
Subject: [PATCH] Add limited_seconds to TlObject.

---
 generate_files.py               | 9 ++++++++-
 pytdbot/types/td_types/types.py | 9 ++++++++-
 2 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/generate_files.py b/generate_files.py
index 99e33ae..01d7fbc 100644
--- a/generate_files.py
+++ b/generate_files.py
@@ -148,7 +148,7 @@ def generate_self_args(args, classes):
             arg_type = generate_union_types(arg_type, arg_data["type"], classes)
 
         args_list.append(
-            f"self.{arg_name}: {arg_type} = {arg_value}\n{indent * 2}r\"\"\"{escape_quotes(arg_data['description'])}\"\"\""
+            f'self.{arg_name}: {arg_type} = {arg_value}\n{indent * 2}r"""{escape_quotes(arg_data["description"])}"""'
         )
     if not args_list:
         return "pass"
@@ -403,6 +403,13 @@ def __bool__(self):
     def is_error(self): # for backward compatibility
         return isinstance(self, Error)
 
+    @property
+    def limited_seconds(self):
+        if self.is_error and self.code == 429:
+            return pytdbot.utils.get_retry_after_time(self.message)
+        else:
+            return 0
+
     def getType(self):
         raise NotImplementedError
 
diff --git a/pytdbot/types/td_types/types.py b/pytdbot/types/td_types/types.py
index 5814931..92e2f7d 100644
--- a/pytdbot/types/td_types/types.py
+++ b/pytdbot/types/td_types/types.py
@@ -1,8 +1,8 @@
 from typing import Union, Literal, List
 from base64 import b64decode
 from .bound_methods import (
-    MessageBoundMethods,
     FileBoundMethods,
+    MessageBoundMethods,
     CallbackQueryBoundMethods,
 )
 import pytdbot
@@ -27,6 +27,13 @@ def __bool__(self):
     def is_error(self):  # for backward compatibility
         return isinstance(self, Error)
 
+    @property
+    def limited_seconds(self):
+        if self.is_error and self.code == 429:
+            return pytdbot.utils.get_retry_after_time(self.message)
+        else:
+            return 0
+
     def getType(self):
         raise NotImplementedError