securedrop.py

#!/usr/bin/env python3

import sys
import registration
import login
import contacts
from broadcast import start_networking, print_online_contacts
import send_file

def get_user_input():
    try:
        while(action := input()):
                if action == "help":
                    options =\
                        """
                        \"add\"\t-> Add a new contact
                        \"list\"\t-> List all online contacts
                        \"send\"\t-> Transfer file to contact
                        \"exit\"\t-> Exit SecureDrop
                        """
                    print(options)
                elif action == "add":
                    contacts.main();
                elif action == "list":
                    print("Current contacts online:")
                    print_online_contacts()
                elif action.split()[0] == "send":
                    # begin sending packets
                    # call send func with arg 1 and 2 which are contact's email and filepath
                    send_file.main(action.split()[1], action.split()[2])
                    print('initiating file transfer')
                elif action == "exit":
                    sys.exit("Exited SecureDrop")
                else:
                    print("Command not Recognized")

    except KeyboardInterrupt:
        sys.exit("Exited SecureDrop")

def main():
    registration.main()
    login.main()
    start_networking()
    get_user_input()

if __name__ == '__main__':
    main()