Tweaks to waitress command

ipmb · ipmb · commit 6a49df9df24e · 2018-10-26T17:13:08.000-06:00
diff --git a/django_webserver/management/commands/waitress.py b/django_webserver/management/commands/waitress.py
@@ -1,6 +1,6 @@
 from __future__ import absolute_import
 
-from waitress.runner import run
+import waitress.runner
 
 from ...base_command import WebserverCommand
 from ...utils import wsgi_app_name
@@ -14,5 +14,8 @@ class Command(WebserverCommand):
 
     help = "Start waitress server"
 
+    def prep_server_args(self, argv):
+        return argv[1:] + [wsgi_app_name()]
+
     def start_server(self, *args):
-        run(argv=args[1:] + (wsgi_app_name(),))
+        waitress.runner.run(argv=args)
