No restart after update (#2242)

src/Ui/UiWebsocket.py

@@ -1124,6 +1124,7 @@ class UiWebsocket(object):
 
             import main
             main.update_after_shutdown = True
+            main.restart_after_shutdown = True
             SiteManager.site_manager.save()
             main.file_server.stop()
             main.ui_server.stop()
@@ -1234,4 +1235,4 @@ class UiWebsocket(object):
             else:
                 gevent.spawn(main.file_server.checkSites, check_files=False, force_port_check=True)
 
-        self.response(to, "ok")
+        self.response(to, "ok")

zeronet.py

@@ -38,8 +38,9 @@ def main():
             import update
             print("Updating...")
             update.update()
-            print("Restarting...")
-            restart()
+            if main.restart_after_shutdown:
+                print("Restarting...")
+                restart()
         else:
             print("Shutting down...")
             prepareShutdown()