Upload edgar_client.py

edgar_client.py

@@ -67,80 +67,55 @@ class EdgarDataClient:
         if not self.edgar:
             return
         
-        # Wrap get_submissions and get_company_facts with timeout
+        # Wrap get_submissions and get_company_facts with timeout (thread-based, Gradio compatible)
         original_get_submissions = self.edgar.get_submissions
         original_get_company_facts = self.edgar.get_company_facts
         
         def get_submissions_with_timeout(cik):
-            import signal
-            
-            def timeout_handler(signum, frame):
-                raise TimeoutError("SEC API request timeout (30s)")
-            
-            # Set signal alarm (Unix only, Windows needs different approach)
-            try:
-                signal.signal(signal.SIGALRM, timeout_handler)
-                signal.alarm(self.timeout)
-                result = original_get_submissions(cik)
-                signal.alarm(0)  # Cancel alarm
-                return result
-            except AttributeError:
-                # Windows doesn't support signal.SIGALRM, use threading.Timer fallback
-                result = [None]
-                exception = [None]
-                
-                def wrapper():
-                    try:
-                        result[0] = original_get_submissions(cik)
-                    except Exception as e:
-                        exception[0] = e
-                
-                thread = threading.Thread(target=wrapper, daemon=True)
-                thread.start()
-                thread.join(timeout=self.timeout)
-                
-                if thread.is_alive():
-                    raise TimeoutError("SEC API request timeout (30s)")
-                
-                if exception[0]:
-                    raise exception[0]
-                
-                return result[0]
+            """Thread-based timeout wrapper for get_submissions (Gradio compatible)"""
+            result = [None]
+            exception = [None]
+            
+            def wrapper():
+                try:
+                    result[0] = original_get_submissions(cik)
+                except Exception as e:
+                    exception[0] = e
+            
+            thread = threading.Thread(target=wrapper, daemon=True)
+            thread.start()
+            thread.join(timeout=self.timeout)
+            
+            if thread.is_alive():
+                raise TimeoutError(f"SEC API request timeout ({self.timeout}s)")
+            
+            if exception[0]:
+                raise exception[0]
+            
+            return result[0]
         
         def get_company_facts_with_timeout(cik):
-            import signal
-            
-            def timeout_handler(signum, frame):
-                raise TimeoutError("SEC API request timeout (30s)")
-            
-            try:
-                signal.signal(signal.SIGALRM, timeout_handler)
-                signal.alarm(self.timeout)
-                result = original_get_company_facts(cik)
-                signal.alarm(0)
-                return result
-            except AttributeError:
-                # Windows fallback
-                result = [None]
-                exception = [None]
-                
-                def wrapper():
-                    try:
-                        result[0] = original_get_company_facts(cik)
-                    except Exception as e:
-                        exception[0] = e
-                
-                thread = threading.Thread(target=wrapper, daemon=True)
-                thread.start()
-                thread.join(timeout=self.timeout)
-                
-                if thread.is_alive():
-                    raise TimeoutError("SEC API request timeout (30s)")
-                
-                if exception[0]:
-                    raise exception[0]
-                
-                return result[0]
+            """Thread-based timeout wrapper for get_company_facts (Gradio compatible)"""
+            result = [None]
+            exception = [None]
+            
+            def wrapper():
+                try:
+                    result[0] = original_get_company_facts(cik)
+                except Exception as e:
+                    exception[0] = e
+            
+            thread = threading.Thread(target=wrapper, daemon=True)
+            thread.start()
+            thread.join(timeout=self.timeout)
+            
+            if thread.is_alive():
+                raise TimeoutError(f"SEC API request timeout ({self.timeout}s)")
+            
+            if exception[0]:
+                raise exception[0]
+            
+            return result[0]
         
         self.edgar.get_submissions = get_submissions_with_timeout
         self.edgar.get_company_facts = get_company_facts_with_timeout