manubot#337 connect command timeout with execution timeout.

manubot/cite/citations.py

@@ -31,11 +31,18 @@ class Citations:
     # `sort_csl_items=False` retains order of input_ids in get_csl_items.
     # (input_ids with the same standard_id will still be deduplicated).
     sort_csl_items: bool = True
+    # timeout for requests
+    timeout_seconds: tp.Union[tuple, int, float, None] = (3, 15)
 
     def __post_init__(self):
         input_ids = list(dict.fromkeys(self.input_ids))  # deduplicate
         self.citekeys = [
-            CiteKey(x, aliases=self.aliases, infer_prefix=self.infer_citekey_prefixes)
+            CiteKey(
+                x,
+                aliases=self.aliases,
+                infer_prefix=self.infer_citekey_prefixes,
+                timeout_seconds=self.timeout_seconds,
+            )
             for x in input_ids
         ]
 

manubot/cite/citekey.py

@@ -21,6 +21,8 @@ class CiteKey:
     """Mapping from input identifier to aliases"""
     infer_prefix: bool = True
     """Whether to infer the citekey's prefix when a prefix is missing or unhandled"""
+    timeout_seconds: tp.Union[tuple, int, float, None] = (3, 12)
+    """The time to wait when making requests"""
 
     def __post_init__(self):
         self.check_input_id(self.input_id)
@@ -208,7 +210,7 @@ def __repr__(self):
     def csl_item(self):
         from .csl_item import CSL_Item
 
-        csl_item = self.handler.get_csl_item(self)
+        csl_item = self.handler.get_csl_item(self, self.timeout_seconds)
         if not isinstance(csl_item, CSL_Item):
             csl_item = CSL_Item(csl_item)
         csl_item.set_id(self.standard_id)

manubot/command.py

@@ -41,8 +41,7 @@ def parse_arguments():
             "-t",
             "--timeout",
             dest="timeout_seconds",
-            default="3",
-            type=int,
+            default=(3, 15),
             help="timeout for web requests in seconds",
         )
     args = parser.parse_args()