Refactored welcome.py to use a class so its functions would be callable by remote-install.sh. This ensure the setup uses the same IP address that will be displayed by Dialog.

scripts/remote-install.sh

@@ -450,3 +450,9 @@ service gns3 start
 log "Download http://$MY_IP_ADDR:8003/$UUID/$HOSTNAME.ovpn to setup your OpenVPN client after rebooting the server"
 
 fi
+
+if [ $WELCOME_SETUP == 1 ]
+then
+python3 -c 'import sys; sys.path.append("/usr/local/bin/"); import welcome; ws = welcome.Welcome_dialog(); ws.repair_remote_install()'
+su gns3 & cd ~
+fi

scripts/welcome.py

@@ -27,13 +27,21 @@ from json import loads as convert
 import urllib.request
 from dialog import Dialog, PythonDialogBug
 
-try:
+
+class Welcome_dialog():
+    def __init__(self):
+        try:
             locale.setlocale(locale.LC_ALL, '')
-except locale.Error:
+        except locale.Error:
             # Not supported via SSH
             pass
+        self.d = Dialog(dialog="dialog", autowidgetsize=True)
+        if self.gns3_version() is None:
+            self.d.set_background_title("GNS3")
+        else:
+            self.d.set_background_title("GNS3 {}".format(self.gns3_version()))
 
-def get_ip():
+    def get_ip(self):
         """
         Return the active IP
         """
@@ -57,12 +65,17 @@ def get_ip():
             
         return ip_addr
 
-def repair_remote_install():
+    def repair_remote_install(self):
-    ip_addr = get_ip()
+        """
-    subprocess.run(["sed", "-i", f"'s/host = 0.0.0.0/host = {ip_addr}/'", "/etc/gns3/gns3_server.conf"],capture_output=False)
+        This method is only called by remote-install.sh during setup to ensure it is setting the same IP as shown by Dialog
+        """        
+        ip_addr = self.get_ip()
+        subprocess.run(["sed", "-i", f"s/host = 0.0.0.0/host = {ip_addr}/", "/etc/gns3/gns3_server.conf"],capture_output=False)
+        subprocess.run(["service", "gns3", "stop"],capture_output=False)
+        subprocess.run(["service", "gns3", "start"],capture_output=False)
 
 
-def get_config():
+    def get_config(self):
         """
         Read the config
         """
@@ -72,7 +85,7 @@ def get_config():
         return config
 
 
-def write_config(config):
+    def write_config(self, config):
         """
         Write the config file
         """
@@ -81,19 +94,19 @@ def write_config(config):
             config.write(f)
 
 
-def gns3_major_version():
+    def gns3_major_version(self):
         """
         Returns the GNS3 major server version
         """
 
-    version = gns3_version()
+        version = self.gns3_version()
         if version:
             match = re.search(r"\d+.\d+", version)
             return match.group(0)
         return ""
 
 
-def gns3_version():
+    def gns3_version(self):
         """
         Return the GNS3 server version
         """
@@ -103,7 +116,7 @@ def gns3_version():
             return None
 
 
-def gns3vm_version():
+    def gns3vm_version(self):
         """
         Return the GNS3 VM version
         """
@@ -114,30 +127,23 @@ def gns3vm_version():
             return "Remote Install"
 
 
-d = Dialog(dialog="dialog", autowidgetsize=True)
+    def mode(self):
-if gns3_version() is None:
+        if self.d.yesno("This feature is for testers only. You may break your GNS3 installation. Are you REALLY sure you want to continue?", yes_label="Exit (Safe option)", no_label="Continue") == self.d.OK:
-    d.set_background_title("GNS3")
-else:
-    d.set_background_title("GNS3 {}".format(gns3_version()))
-
-
-def mode():
-    if d.yesno("This feature is for testers only. You may break your GNS3 installation. Are you REALLY sure you want to continue?", yes_label="Exit (Safe option)", no_label="Continue") == d.OK:
             return
-    code, tag = d.menu("Select the GNS3 version",
+        code, tag = self.d.menu("Select the GNS3 version",
                         choices=[("2.1", "Stable release for this GNS3 VM (RECOMMENDED)"),
                                     ("2.1dev", "Development version for stable release"),
                                     ("2.2", "Latest stable release")])
-    d.clear()
+        self.d.clear()
         if code == Dialog.OK:
             os.makedirs(os.path.expanduser("~/.config/GNS3"), exist_ok=True)
             with open(os.path.expanduser("~/.config/GNS3/gns3_release"), "w+") as f:
                 f.write(tag)
 
-        update(force=True)
+            self.update(force=True)
 
 
-def get_release():
+    def get_release():
         try:
             with open(os.path.expanduser("~/.config/GNS3/gns3_release")) as f:
                 content = f.read()
@@ -155,13 +161,13 @@ def get_release():
             return "1.5"
 
 
-def update(force=False):
+    def update(self, force=False):
         if not force:
-        if d.yesno("PLEASE SNAPSHOT THE VM BEFORE RUNNING THE UPGRADE IN CASE OF FAILURE. The server will reboot at the end of the upgrade process. Continue?") != d.OK:
+            if self.d.yesno("PLEASE SNAPSHOT THE VM BEFORE RUNNING THE UPGRADE IN CASE OF FAILURE. The server will reboot at the end of the upgrade process. Continue?") != self.d.OK:
                 return
-    release = get_release()
+        release = self.get_release()
         if release == "2.2":
-        if d.yesno("It is recommended to run GNS3 version 2.2 with lastest GNS3 VM based on Ubuntu 18.04 LTS, please download this VM from our website or continue at your own risk!") != d.OK:
+            if self.d.yesno("It is recommended to run GNS3 version 2.2 with lastest GNS3 VM based on Ubuntu 18.04 LTS, please download this VM from our website or continue at your own risk!") != self.d.OK:
                 return
         if release.endswith("dev"):
             ret = os.system("curl -Lk https://raw.githubusercontent.com/GNS3/gns3-vm/unstable/scripts/update_{}.sh > /tmp/update.sh && bash -x /tmp/update.sh".format(release))
@@ -172,21 +178,21 @@ def update(force=False):
             time.sleep(15)
 
 
-def migrate():
+    def migrate(self):
         """
         Migrate GNS3 VM data.
         """
 
-    code, option = d.menu("Select an option",
+        code, option = self.d.menu("Select an option",
                             choices=[("Setup", "Configure this VM to send data to another GNS3 VM"),
                                     ("Send", "Send images and projects to another GNS3 VM")])
-    d.clear()
+        self.d.clear()
         if code == Dialog.OK:
-        (answer, destination) = d.inputbox("What is IP address or hostname of the other GNS3 VM?", init="172.16.1.128")
+            (answer, destination) = self.d.inputbox("What is IP address or hostname of the other GNS3 VM?", init="172.16.1.128")
-        if answer != d.OK:
+            if answer != self.d.OK:
                 return
-        if destination == get_ip():
+            if destination == self.get_ip():
-            d.msgbox("The destination cannot be the same as this VM IP address ({})".format(destination))
+                self.d.msgbox("The destination cannot be the same as this VM IP address ({})".format(destination))
                 return
             if option == "Send":
                 # first make sure they are no files belonging to root
@@ -196,33 +202,33 @@ def migrate():
                 ret = os.system('bash -c "{}"'.format(command))
                 time.sleep(10)
                 if ret != 0:
-                d.msgbox("Could not send data to the other GNS3 VM located at {}".format(destination))
+                    self.d.msgbox("Could not send data to the other GNS3 VM located at {}".format(destination))
                 else:
-                d.msgbox("Images and projects have been successfully sent to the other GNS3 VM located at {}".format(destination))
+                    self.d.msgbox("Images and projects have been successfully sent to the other GNS3 VM located at {}".format(destination))
             elif option == "Setup":
                 script = """
-if [ ! -f ~/.ssh/gns3-vm-key ]
+    if [ ! -f ~/.ssh/gns3-vm-key ]
-then
+    then
         ssh-keygen -f ~/.ssh/gns3-vm-key -N '' -C gns3@{}
-fi
+    fi
-ssh-copy-id -i ~/.ssh/gns3-vm-key gns3@{}
+    ssh-copy-id -i ~/.ssh/gns3-vm-key gns3@{}
-""".format(get_ip(), destination)
+    """.format(self.get_ip(), destination)
                 ret = os.system('bash -c "{}"'.format(script))
                 time.sleep(10)
                 if ret != 0:
-                d.msgbox("Error while setting up the migrate feature")
+                    self.d.msgbox("Error while setting up the migrate feature")
                 else:
-                d.msgbox("Configuration successful, you can now send data to the GNS3 VM located at {} without password".format(destination))
+                    self.d.msgbox("Configuration successful, you can now send data to the GNS3 VM located at {} without password".format(destination))
 
 
-def shrink_disk():
+    def shrink_disk(self):
 
         ret = os.system("lspci | grep -i vmware")
         if ret != 0:
-        d.msgbox("Shrinking the disk is only supported when running inside VMware")
+            self.d.msgbox("Shrinking the disk is only supported when running inside VMware")
             return
 
-    if d.yesno("Would you like to shrink the VM disk? The VM will reboot at the end of the process. Continue?") != d.OK:
+        if self.d.yesno("Would you like to shrink the VM disk? The VM will reboot at the end of the process. Continue?") != self.d.OK:
             return
 
         os.system("sudo service gns3 stop")
@@ -230,81 +236,81 @@ def shrink_disk():
         os.system("sudo vmware-toolbox-cmd disk shrink /opt")
         os.system("sudo vmware-toolbox-cmd disk shrink /")
 
-    d.msgbox("The GNS3 VM will reboot")
+        self.d.msgbox("The GNS3 VM will reboot")
         os.execvp("sudo", ['/usr/bin/sudo', "reboot"])
 
-def vm_information():
+    def vm_information(self):
         """
         Show IP, SSH settings....
         """
 
         content = "Welcome to GNS3 appliance\n\n"
 
-    version = gns3_version()
+        version = self.gns3_version()
         if version is None:
             content += "GNS3 is not installed please install it with sudo pip3 install gns3-server. Or download a preinstalled VM.\n\n"
         else:
             content = "GNS3 version: {gns3_version}\nVM version: {gns3vm_version}\nKVM support available: {kvm}\n\n".format(
-            gns3vm_version=gns3vm_version(),
+                gns3vm_version=self.gns3vm_version(),
                 gns3_version=version,
-            kvm=kvm_support())
+                kvm=self.kvm_support())
 
-    ip = get_ip()
+        ip = self.get_ip()
 
         if ip:
             content += "IP: {ip}\n\nTo log in using SSH:\nssh gns3@{ip}\nPassword: gns3\n\nImages and projects are located in /opt/gns3""".format(ip=ip)
         else:
             content += "eth0 is not configured. Please manually configure it via the Networking menu."
 
-    content += "\n\nRelease channel: " + get_release()
+        content += "\n\nRelease channel: " + self.get_release()
 
         try:
-        d.msgbox(content)
+            self.d.msgbox(content)
         # If it's an scp command or any bugs
         except:
             os.execvp("bash", ['/bin/bash'])
 
 
-def check_internet_connectivity():
+    def check_internet_connectivity(self):
-    d.pause("Please wait...\n\n")
+        self.d.pause("Please wait...\n\n")
         try:
             response = urllib.request.urlopen('http://pypi.python.org/', timeout=5)
         except urllib.request.URLError as err:
-        d.infobox("Can't connect to Internet (pypi.python.org): {}".format(str(err)))
+            self.d.infobox("Can't connect to Internet (pypi.python.org): {}".format(str(err)))
             time.sleep(15)
             return
-    d.infobox("Connection to Internet: OK")
+        self.d.infobox("Connection to Internet: OK")
         time.sleep(2)
 
 
-def keyboard_configuration():
+    def keyboard_configuration():
         """
         Allow user to change the keyboard layout
         """
         os.system("/usr/bin/sudo dpkg-reconfigure keyboard-configuration")
 
 
-def set_security():
+    def set_security(self):
-    config = get_config()
+        config = self.get_config()
-    if d.yesno("Enable server authentication?") == d.OK:
+        if self.d.yesno("Enable server authentication?") == self.d.OK:
             if not config.has_section("Server"):
                 config.add_section("Server")
             config.set("Server", "auth", True)
-        (answer, text) = d.inputbox("Login?")
+            (answer, text) = self.d.inputbox("Login?")
-        if answer != d.OK:
+            if answer != self.d.OK:
                 return
             config.set("Server", "user", text)
-        (answer, text) = d.passwordbox("Password?")
+            (answer, text) = self.d.passwordbox("Password?")
-        if answer != d.OK:
+            if answer != self.d.OK:
                 return
             config.set("Server", "password", text)
         else:
             config.set("Server", "auth", False)
 
-    write_config(config)
+        self.write_config(config)
 
 
-def log():
+    def log(self):
         os.system("/usr/bin/sudo chmod 755 /var/log/upstart/gns3.log")
         with open("/var/log/upstart/gns3.log") as f:
             try:
@@ -315,37 +321,37 @@ def log():
                 return
 
 
-def edit_config():
+    def edit_config(self):
         """
         Edit GNS3 configuration file
         """
 
-    major_version = gns3_major_version()
+        major_version = self.gns3_major_version()
         if major_version == "2.2":
             os.system("nano ~/.config/GNS3/{}/gns3_server.conf".format(major_version))
         else:
             os.system("nano ~/.config/GNS3/gns3_server.conf")
 
 
-def edit_network():
+    def edit_network(self):
         """
         Edit network configuration file
         """
-    if d.yesno("The server will reboot at the end of the process. Continue?") != d.OK:
+        if self.d.yesno("The server will reboot at the end of the process. Continue?") != self.d.OK:
             return
         os.system("sudo nano /etc/network/interfaces")
         os.execvp("sudo", ['/usr/bin/sudo', "reboot"])
 
 
-def edit_proxy():
+    def edit_proxy(self):
         """
         Configure proxy settings
         """
-    res, http_proxy = d.inputbox(text="HTTP proxy string, for example http://<user>:<password>@<proxy>:<port>. Leave empty for no proxy.")
+        res, http_proxy = self.d.inputbox(text="HTTP proxy string, for example http://<user>:<password>@<proxy>:<port>. Leave empty for no proxy.")
-    if res != d.OK:
+        if res != self.d.OK:
             return
-    res, https_proxy = d.inputbox(text="HTTPS proxy string, for example http://<user>:<password>@<proxy>:<port>. Leave empty for no proxy.")
+        res, https_proxy = self.d.inputbox(text="HTTPS proxy string, for example http://<user>:<password>@<proxy>:<port>. Leave empty for no proxy.")
-    if res != d.OK:
+        if res != self.d.OK:
             return
 
         with open('/tmp/00proxy', 'w+') as f:
@@ -364,48 +370,45 @@ def edit_proxy():
         os.system("sudo chmod 744 /etc/profile.d/proxy.sh")
         os.system("sudo cp /etc/profile.d/proxy.sh /etc/default/docker")
 
-    d.msgbox("The GNS3 VM will reboot")
+        self.d.msgbox("The GNS3 VM will reboot")
         os.execvp("sudo", ['/usr/bin/sudo', "reboot"])
 
 
-def kvm_support():
+    def kvm_support(self):
         """
         Returns true if KVM is available
         """
         return subprocess.call("kvm-ok") == 0
 
 
-def kvm_control():
+    def kvm_control(self):
         """
         Check if KVM is correctly configured
         """
 
-    kvm_ok = kvm_support()
+        kvm_ok = self.kvm_support()
-    config = get_config()
+        config = self.get_config()
         try:
             if config.getboolean("Qemu", "enable_kvm") is True:
                 if kvm_ok is False:
-                if d.yesno("KVM is not available!\n\nQemu VM will crash!!\n\nThe reason could be unsupported hardware or another virtualization solution is already running.\n\nDisable KVM and get lower performances?") == d.OK:
+                    if self.d.yesno("KVM is not available!\n\nQemu VM will crash!!\n\nThe reason could be unsupported hardware or another virtualization solution is already running.\n\nDisable KVM and get lower performances?") == self.d.OK:
                         config.set("Qemu", "enable_kvm", False)
-                    write_config(config)
+                        self.write_config(config)
                         os.execvp("sudo", ['/usr/bin/sudo', "reboot"])
             else:
                 if kvm_ok is True:
-                if d.yesno("KVM is available on your computer.\n\nEnable KVM and get better performances?") == d.OK:
+                    if self.d.yesno("KVM is available on your computer.\n\nEnable KVM and get better performances?") == self.d.OK:
                         config.set("Qemu", "enable_kvm", True)
-                    write_config(config)
+                        self.write_config(config)
                         os.execvp("sudo", ['/usr/bin/sudo', "reboot"])
         except configparser.NoSectionError:
             return
 
 
-vm_information()
+    def display_loop(self):
-kvm_control()
+        try:
-
-
-try:
             while True:
-        code, tag = d.menu("GNS3 {}".format(gns3_version()),
+                code, tag = self.d.menu("GNS3 {}".format(self.gns3_version()),
                                 choices=[("Information", "Display VM information"),
                                     ("Upgrade", "Upgrade GNS3"),
                                     ("Migrate", "Migrate data to another GNS3 VM"),
@@ -422,12 +425,12 @@ try:
                                     ("Restore", "Restore the VM (if you have trouble for upgrade)"),
                                     ("Reboot", "Reboot the VM"),
                                     ("Shutdown", "Shutdown the VM")])
-        d.clear()
+                self.d.clear()
                 if code == Dialog.OK:
                     if tag == "Shell":
                         os.execvp("bash", ['/bin/bash'])
                     elif tag == "Version":
-                mode()
+                        self.mode()
                     elif tag == "Restore":
                         os.execvp("sudo", ['/usr/bin/sudo', "/usr/local/bin/gns3restore"])
                     elif tag == "Reboot":
@@ -435,28 +438,32 @@ try:
                     elif tag == "Shutdown":
                         os.execvp("sudo", ['/usr/bin/sudo', "poweroff"])
                     elif tag == "Upgrade":
-                update()
+                        self.update()
                     elif tag == "Information":
-                vm_information()
+                        self.vm_information()
                     elif tag == "Log":
-                log()
+                        self.log()
                     elif tag == "Migrate":
-                migrate()
+                        self.migrate()
                     elif tag == "Configure":
-                edit_config()
+                        self.edit_config()
                     elif tag == "Networking":
-                edit_network()
+                        self.edit_network()
                     elif tag == "Security":
-                set_security()
+                        self.set_security()
                     elif tag == "Keyboard":
-                keyboard_configuration()
+                        self.keyboard_configuration()
                     elif tag == "Test":
-                check_internet_connectivity()
+                        self.check_internet_connectivity()
                     elif tag == "Proxy":
-                edit_proxy()
+                        self.edit_proxy()
                     elif tag == "Shrink":
-                shrink_disk()
+                        self.shrink_disk()
-            elif tag == "Repair":
+        except KeyboardInterrupt:
-                repair_remote_install()
-except KeyboardInterrupt:
             sys.exit(0)
+
+if __name__ == "__main__":
+    ws = Welcome_dialog()
+    ws.vm_information()
+    ws.kvm_control()
+    ws.display_loop()