created a __reload functiont hat reloads the hosts from the host file, upped the length of the backup hosts file

Author: ekultek

lib/settings.py

@@ -161,7 +161,7 @@ def backup_host_file(current, path):
         os.makedirs(path)
     new_filename = "{}/hosts_{}_{}.txt".format(
         path,
-        lib.jsonize.random_file_name(length=17),
+        lib.jsonize.random_file_name(length=22),
         str(datetime.datetime.today()).split(" ")[0]
     )
     shutil.copyfile(current, new_filename)

lib/term/terminal.py

@@ -61,9 +61,12 @@ def __init__(self, tokens, modules):
         self.modules = modules
         try:
             self.loaded_hosts = open(lib.settings.HOST_FILE).readlines()
-        except:
+        except IOError:
             lib.output.warning("no hosts file present")
-            self.loaded_hosts = []
+            self.loaded_hosts = open(lib.settings.HOST_FILE, "a+").readlines()
+
+    def __reload(self):
+        self.loaded_hosts = open(lib.settings.HOST_FILE).readlines()
 
     def reflect_memory(self, max_memory=100):
         """
@@ -551,6 +554,7 @@ def terminal_main_display(self, tokens, extra_commands=None, save_history=True):
                                 else:
                                     lib.output.error("cannot reset {} API credentials".format(choice))
                         self.history.append(choice)
+                        self.__reload()
                 except KeyboardInterrupt:
                     lib.output.warning("use the `exit/quit` command to end terminal session")
             except IndexError: