Safe from Timeout, Works Fine for Javatpoint

Author: GOSHROW

app.py

@@ -28,13 +28,15 @@ def input_post():
 @app.route('/<outFile>')
 def getDownload(outFile) :
     try:
-        files = [f for f in os.listdir('.') if os.path.isfile(f)]
+        files = [str(f) for f in os.listdir('.') if os.path.isfile(f)]
         for f in files:
             print(f)
-        return send_file(filename_or_fp = str(outFile + '.pdf'), as_attachment=True), None
+        if str(outFile + '.pdf') in files:
+            return send_file(filename_or_fp = str(outFile + '.pdf'), as_attachment=True), None
+        return send_file(filename_or_fp = str(outFile + '.html'), as_attachment=True), None
     except Exception as E:
         print(E)
         return 'Error'
 
 if __name__ == '__main__':
-    app.run(host="localhost", port=8000, debug=True)
+    app.run(host="localhost", port=8000, debug=False)

main.py

@@ -1,12 +1,15 @@
 from urllib import request
 from bs4 import BeautifulSoup
+from multiprocessing import Process
 import json
 import os
 import sys
 import pdfkit
 import subprocess
 import platform
 import re 
+import time
+
 
 def getPDF(filename = 'out'):
 
@@ -170,7 +173,11 @@ def getAbsolute(self, content, domain):
     def util(self, URL, iterations = 1, filename = 'out'):
         print(str(iterations) + " pages to go . . .")
         if iterations < 1:
-            return getPDF(filename)
+            action_process = Process(target=lambda : getPDF(filename))
+            action_process.start()
+            action_process.join(timeout=15)
+            action_process.terminate()
+            return None
         req = request.urlopen(URL)
         html = req.read().decode('utf-8')
         domain = self.domain