fix #1213 (#1287)

3 files changed, 18 insertions, 26 deletions

diff --git a/lib/api/server.py b/lib/api/server.py
index 66087b4e..93b08ff1 100644
--- a/lib/api/server.py
+++ b/lib/api/server.py
@@ -11,7 +11,8 @@ import time
 from threading import Thread
 
 import requests
-from flask import Flask, g, request
+from flask import Flask, g
+from werkzeug.serving import make_server
 
 from ..utils.json import InkStitchJSONEncoder
 from .install import install
@@ -24,13 +25,14 @@ class APIServer(Thread):
         self.extension = args[0]
         Thread.__init__(self, *args[1:], **kwargs)
         self.daemon = True
-        self.shutting_down = False
         self.app = None
         self.host = None
         self.port = None
         self.ready = False
 
         self.__setup_app()
+        self.flask_server = None
+        self.server_thread = None
 
     def __setup_app(self):  # noqa: C901
         # Disable warning about using a development server in a production environment
@@ -50,20 +52,13 @@ class APIServer(Thread):
             # this request
             g.extension = self.extension
 
-        @self.app.route('/shutdown', methods=['POST'])
-        def shutdown():
-            self.shutting_down = True
-            request.environ.get('werkzeug.server.shutdown')()
-            return "shutting down"
-
         @self.app.route('/ping')
         def ping():
             return "pong"
 
     def stop(self):
-        # for whatever reason, shutting down only seems possible in
-        # the context of a flask request, so we'll just make one
-        requests.post("http://%s:%s/shutdown" % (self.host, self.port))
+        self.flask_server.shutdown()
+        self.server_thread.join()
 
     def disable_logging(self):
         logging.getLogger('werkzeug').setLevel(logging.ERROR)
@@ -76,7 +71,9 @@ class APIServer(Thread):
 
         while True:
             try:
-                self.app.run(self.host, self.port, threaded=True)
+                self.flask_server = make_server(self.host, self.port, self.app)
+                self.server_thread = Thread(target=self.flask_server.serve_forever)
+                self.server_thread.start()
             except socket.error as e:
                 if e.errno == errno.EADDRINUSE:
                     self.port += 1
diff --git a/lib/commands.py b/lib/commands.py
index 7435f753..f3875409 100644
--- a/lib/commands.py
+++ b/lib/commands.py
@@ -286,7 +286,6 @@ def add_group(document, node, command):
         "transform": get_correction_transform(node)
     })
     parent.insert(parent.index(node) + 1, group)
-    
     return group
 
 
diff --git a/lib/extensions/print_pdf.py b/lib/extensions/print_pdf.py
index a1a8c6c0..0facdf92 100644
--- a/lib/extensions/print_pdf.py
+++ b/lib/extensions/print_pdf.py
@@ -15,10 +15,10 @@ from datetime import date
 from threading import Thread
 
 import appdirs
-import requests
 from flask import Flask, Response, jsonify, request, send_from_directory
 from jinja2 import Environment, FileSystemLoader, select_autoescape
 from lxml import etree
+from werkzeug.serving import make_server
 
 from ..gui import open_url
 from ..i18n import get_languages
@@ -66,7 +66,8 @@ class PrintPreviewServer(Thread):
         self.realistic_color_block_svgs = kwargs.pop('realistic_color_block_svgs')
         Thread.__init__(self, *args, **kwargs)
         self.daemon = True
-        self.shutting_down = False
+        self.flask_server = None
+        self.server_thread = None
 
         self.__setup_app()
 
@@ -89,15 +90,9 @@ class PrintPreviewServer(Thread):
         def index():
             return self.html
 
-        @self.app.route('/shutdown', methods=['POST'])
-        def shutdown():
-            self.shutting_down = True
-            request.environ.get('werkzeug.server.shutdown')()
-            return "shutting down"
-
         @self.app.route('/resources/<path:resource>', methods=['GET'])
         def resources(resource):
-            return send_from_directory(self.resources_path, resource, cache_timeout=1)
+            return send_from_directory(self.resources_path, resource, max_age=1)
 
         @self.app.route('/settings/<field_name>', methods=['POST'])
         def set_field(field_name):
@@ -158,9 +153,8 @@ class PrintPreviewServer(Thread):
             return Response(self.realistic_overview_svg, mimetype='image/svg+xml')
 
     def stop(self):
-        # for whatever reason, shutting down only seems possible in
-        # the context of a flask request, so we'll just make one
-        requests.post("http://%s:%s/shutdown" % (self.host, self.port))
+        self.flask_server.shutdown()
+        self.server_thread.join()
 
     def disable_logging(self):
         logging.getLogger('werkzeug').setLevel(logging.ERROR)
@@ -173,7 +167,9 @@ class PrintPreviewServer(Thread):
 
         while True:
             try:
-                self.app.run(self.host, self.port, threaded=True)
+                self.flask_server = make_server(self.host, self.port, self.app)
+                self.server_thread = Thread(target=self.flask_server.serve_forever)
+                self.server_thread.start()
             except socket.error as e:
                 if e.errno == errno.EADDRINUSE:
                     self.port += 1