fid

main.py

@@ -18,6 +18,7 @@ except ImportError:
     logger.warning("RdRand is not available.")
 
 tmp_dir = None
+push_timeout = 10
 
 
 class TemporaryFile:
@@ -242,7 +243,7 @@ def push(pool_url, data, secret):
     logger.info(f"Pushing {len(data)}b.")
 
     resp = requests.post(
-        f"{pool_url}/api/pool", data=data, headers={"X-Secret": secret}
+        f"{pool_url}/api/pool", data=data, headers={"X-Secret": secret}, timeout=(push_timeout, push_timeout)
     )
 
     (logger.success if resp.status_code == 200 else logger.error)(
@@ -294,6 +295,7 @@ if __name__ == "__main__":
     parser.add_argument("--secret-file", type=str, default="./.secret")
     parser.add_argument("--cooldown", type=int, default=0)
     parser.add_argument("--pool-url", type=str, default="https://trng.iike.ru")
+    parser.add_argument("--push-timeout", type=int, default=10)
     parser.add_argument("--tmp-dir", type=str)
 
     args = parser.parse_args()
@@ -303,6 +305,8 @@ if __name__ == "__main__":
 
         logger.info(f"Changed temp-dir: '{tmp_dir}'")
 
+    push_timeout = max(args.push_timeout, 1)
+
     with open(args.secret_file, "r") as f:
         lines = f.read().strip().split("\n")
         ident = lines[0].strip()