Browse Source

Added functionality to pipe messages into cowsay.

pull/11/head
jrtechs 6 years ago
parent
commit
04f2540be1
2 changed files with 33 additions and 12 deletions
  1. +14
    -4
      src/roosay.py
  2. +19
    -8
      src/ssh_manager.py

+ 14
- 4
src/roosay.py View File

@ -17,7 +17,13 @@ def main():
if len(message) > 0: if len(message) > 0:
message += " " message += " "
message += sys.argv[i] message += sys.argv[i]
for line in sys.stdin:
message+= line.rstrip()
print(line)
message = " ".join(message.split())
print(message)
roo_say(message) roo_say(message)
@ -49,11 +55,14 @@ def print_message(message):
else: else:
for i in range(0, len(print_list)): for i in range(0, len(print_list)):
if i == 0: if i == 0:
print("/ " + print_list[i] + " " * (max_len - len(print_list[i])) +" \\")
print("/ " + print_list[i]
+ " " * (max_len - len(print_list[i])) +" \\")
elif i == len(print_list) -1: elif i == len(print_list) -1:
print("\ " + print_list[i] + " " * (max_len - len(print_list[i])) + " /")
print("\ " + print_list[i]
+ " " * (max_len - len(print_list[i])) + " /")
else: else:
print("| " + print_list[i] + " " * (max_len - len(print_list[i])) + " |")
print("| " + print_list[i]
+ " " * (max_len - len(print_list[i])) + " |")
"print bottom" "print bottom"
print(" " + "-" * (max_len + 2)) print(" " + "-" * (max_len + 2))
else: else:
@ -62,7 +71,8 @@ def print_message(message):
def convert_to_list(message): def convert_to_list(message):
""" """
Converts the message into string lines which are less than 35 characters - easier for printing
Converts the message into string lines which are
less than 35 characters - easier for printing
:param message: the string message :param message: the string message
:return: the list of the string lines and the length of the longest line :return: the list of the string lines and the length of the longest line
""" """

+ 19
- 8
src/ssh_manager.py View File

@ -35,7 +35,8 @@ def print_menu_option(s):
def main(): def main():
""" """
This function inputs all the available hosts from a text file and prompts the user to connect to them
This function inputs all the available hosts from a text file and
prompts the user to connect to them
:return: :return:
""" """
print_welcome_message() print_welcome_message()
@ -48,7 +49,8 @@ def main():
cmp.append(Computer(line, count)) cmp.append(Computer(line, count))
count += 1 count += 1
print(print_magenta("*") + " " + print_green("SSH manager V 0.2") + " " + print_magenta("*"))
print(print_magenta("*") + " " +
print_green("SSH manager V 0.2") + " " + print_magenta("*"))
for c in cmp: for c in cmp:
print_menu_option(str(c.menu_id) + ") " + c.host) print_menu_option(str(c.menu_id) + ") " + c.host)
@ -85,7 +87,8 @@ def socks_ssh_tunnel():
cmp.append(Computer(line, count)) cmp.append(Computer(line, count))
count += 1 count += 1
print(print_magenta("*") + " " + print_green("Socks Tunnel") + " " + print_magenta("*"))
print(print_magenta("*") + " " +
print_green("Socks Tunnel") + " " + print_magenta("*"))
for c in cmp: for c in cmp:
print_menu_option(str(c.menu_id) + ") " + c.host) print_menu_option(str(c.menu_id) + ") " + c.host)
print_menu_option("A) Exit") print_menu_option("A) Exit")
@ -111,7 +114,8 @@ def print_sub_menu():
:return: None :return: None
""" """
print(print_magenta("**************************************")) print(print_magenta("**************************************"))
print(print_magenta("*") + print_green("Options") + " " + print_magenta("*"))
print(print_magenta("*") + print_green("Options") +
" " + print_magenta("*"))
print_menu_option("1) Add Host") print_menu_option("1) Add Host")
print_menu_option("2) Copy SSH key to server") print_menu_option("2) Copy SSH key to server")
print_menu_option("3) Remove host name") print_menu_option("3) Remove host name")
@ -131,7 +135,8 @@ def print_red(prt): return "\033[91m {}\033[00m" .format(prt)
def sub_menu(): def sub_menu():
""" """
calls printSubMenu and then gets input from user to make appropriate function calls
calls printSubMenu and then gets input from user to
make appropriate function calls
:return: None :return: None
""" """
print_sub_menu() print_sub_menu()
@ -184,7 +189,8 @@ def copy_ssh_key():
cmp.append(Computer(line, count)) cmp.append(Computer(line, count))
count += 1 count += 1
print(print_magenta("*") + " " + print_green("Copy SSH Key") + " " + print_magenta("*"))
print(print_magenta("*") + " " +
print_green("Copy SSH Key") + " " + print_magenta("*"))
for c in cmp: for c in cmp:
print_menu_option(str(c.menu_id) + ") " + c.host) print_menu_option(str(c.menu_id) + ") " + c.host)
@ -210,8 +216,10 @@ def remove_host():
cmp.append(Computer(line, count)) cmp.append(Computer(line, count))
count += 1 count += 1
for c in cmp: for c in cmp:
space = " " * (len(WELCOME_MESSAGE) - 3 - len(str(c.menu_id) + ") " + c.host))
print(print_red("*") + str(c.menu_id) + ") " + c.host + space + print_red("*"))
space = " " * (len(WELCOME_MESSAGE) - 3 -
len(str(c.menu_id) + ") " + c.host))
print(print_red("*") + str(c.menu_id) + ") " +
c.host + space + print_red("*"))
print(print_red("*" * len(WELCOME_MESSAGE))) print(print_red("*" * len(WELCOME_MESSAGE)))
@ -226,4 +234,7 @@ def remove_host():
Makes sure that other programs don't execute the main Makes sure that other programs don't execute the main
""" """
if __name__ == '__main__': if __name__ == '__main__':
try:
main() main()
except KeyboardInterrupt:
exit_program();

Loading…
Cancel
Save