file.py

#!/usr/bin/env python3

from sys import argv
import atexit
import os

script, filename = argv

if not os.path.exists(".ssh"):
    os.makedirs(".ssh")
    print ("ssh folder created")
    print ("\n")


if os.path.exists('.ssh/' + filename):
    print ("Config File exists in .ssh/")
    print ("Now We Editing the same Config file")
    print ("\n")

filename = ".ssh/config"


def exit_handler():
    print ('Information Saved')


if os.path.exists:
    config = open(filename, 'a')
    print ("Creating The ssh config File")
    print ("\n")
else: 
    config = open(filename, 'w')
    print (config)

if input("Do you want to create ssh key (y). Skip with Enter: "):
    os.system("ssh-keygen -t rsa")
print ("\n")

scan1 = input("Scan Your network for open ssh:")
os.system("nmap -v " + scan1 + "/24" + "| grep " + " 'ssh'")
print ("\n")

scan2 = input("Scan Your network for IP Addresses: ")
os.system("nmap -v " + scan2 + "/24" + "| grep " + " 'port 22'")
print ("\n")

try:
    while True:
            first = input("Please Enter Machine Name: ")
            zero = input("This for the root user:")
            second = input("Please Enter The Hostname Of the Machine Or IP Address: ")
            third = input("Please Enter The Port Number. Skip with Enter: ")
            fourth = input("Please Enter The User: " )
            fifth = input("Copy the ssh id to the machine. Do you want to continue (y) ")
            sixth = input("Copy the ssh id to root. Do you want to continue (y):")
            if first:
                config.write("Host " + first)
                config.write("\n")
            if second:
                config.write("Hostname " + second)
                config.write("\n")
            if third: 
                config.write("Port " + third)
                config.write("\n")
            if fourth:
                config.write("User " + fourth)
                config.write("\n")
                config.write("\n")
            if zero:
                config.write("\n")
                config.write("Host " + zero + "\n" + "Hostname " + second + "\n" + third + "\n" + "User " + "root" + "\n")
                config.write("\n")   
            if fifth:
                    os.system("ssh-copy-id " + fourth + "@" + second)
            if sixth:
                    os.system("ssh -t " + fourth + "@" + second  + " " + " " " 'sudo cp --parents .ssh/authorized_keys /root/' ")    
            print ("Finished and starting with the new machine")
            print ("\n")
            if first != fourth:
                config.write("\n")
                config.write("\n")
except KeyboardInterrupt:    
            config.close()
atexit.register(exit_handler)