main.py

#!/home/alvaro/.miniconda3/envs/fazip/bin/python

from fazip import fazip
import sys
import getpass


def do_extraction(zipname):
    host, user, pwd = fazip.get_mysql_info()
    db = fazip.connect_to_db(user, pwd, host)
    fazip.unzip_files(db, zipname)
    db.close()


def do_compression(zipname, files):
    host, user, pwd = fazip.get_mysql_info()
    db = fazip.connect_to_db(user, pwd, host)
    str_files = " ".join(files)
    fazip.zip_files(db, zipname, str_files)
    db.close()


def do_list(zipname):
    print("TODO")
    pass


def do_add_user(database):
    print("\n[fazip] Enter the new user's name")
    user = input("[fazip]> ")
    if fazip.add_user_db(database, user):
        print("[fazip] User successfully registered!")
    else:
        print("[fazip] User already registered :(")


def do_edit_user(database):
    print("\n[fazip] Enter the user's name you want to edit")
    user = input("[fazip]> ")
    if fazip.modify_user_db(database, user):
        print("[fazip] User successfully edited!")
    else:
        print("[fazip] User doesn't exists :(")


def do_remove_user(database):
    print("\n[fazip] Enter the user's name you want to remove")
    user = input("[fazip]> ")
    if fazip.remove_user_db(database, user):
        print("[fazip] User successfully removed!")
    else:
        print("[fazip] User doesn't exists :(")


def do_list_users(database):
    users = fazip.get_users_db(database)
    if users:
        print("\n[fazip] Those are the registered users:")
        for user in users:
            print("=> {}".format(user))
    else:
        print("\n[fazip] There are no users registered :(")


def do_change_pass(database):
    print("[fazip] WARNING: You will not be able to extract files" +
          "from previously created zip files")
    verify = input("[fazip] Proceed (y/[n])? ")
    if verify == 'y' or verify == 'yes':
        oldp = getpass.getpass("[fazip] Enter your actual password: ")
        if fazip.get_password_zip(database) == oldp:
            first = getpass.getpass("[fazip] Enter your new password: ")
            second = getpass.getpass("[fazip] Enter your new password again: ")
            if first == second:
                fazip.set_password_zip(database, first)
                print("\n[fazip] Password changed successfully!")
                print("[fazip] Please log in again")
                sys.exit()
            else:
                print("[fazip] Those passwords didn't match.")
        else:
            print("[fazip] Access denied")
            sys.exit()


def print_menu(options):
    print(" ====================================")
    print("|| \t   CONFIGURATION MENU\t    ||")
    print(" ====================================")
    print("\t-> {}) Exit".format(options['exit_menu']))
    print("\t-> {}) List users".format(options['list_users']))
    print("\t-> {}) Add user".format(options['add_user']))
    print("\t-> {}) Edit user".format(options['edit_user']))
    print("\t-> {}) Remove user".format(options['remove_user']))
    print("\t-> {}) Change password (CRITICAL)".format(options['change_pass']))


def menu(options, selection):
    pass


def do_configuration(login=False):
    host, user, pwd = fazip.get_mysql_info()
    db = fazip.connect_to_db(user, pwd, host)
    options = {'exit_menu': 0, 'list_users': 1, 'add_user': 2,
               'edit_user': 3, 'remove_user': 4, 'change_pass': 5}
    if not login:
        pwd = getpass.getpass("Enter password: ")
        if not fazip.get_password_zip(db) == pwd:
            print("[fazip] Access denied")
            sys.exit()

        print_menu(options)

    try:
        selection = int(input("\n[fazip]> "))
    except Exception:
        selection = -1
        print("[fazip] Seriously? That's not a number -.-")

    if options['exit_menu'] <= selection <= options['change_pass']:
        if selection == options['add_user']:
            do_add_user(db)

        elif selection == options['edit_user']:
            do_edit_user(db)

        elif selection == options['remove_user']:
            do_remove_user(db)

        elif selection == options['list_users']:
            do_list_users(db)

        elif selection == options['exit_menu']:
            sys.exit()

        elif selection == options['change_pass']:
            do_change_pass(db)

        db.close()

    else:
        print("[fazip] Wrong answer")

    do_configuration(True)


def print_help(redirected=False):
    if redirected:
        print("[fazip] It looks like you missed something :(")
    else:
        print("[fazip] Help:")
    print("\n- Extract an existing zip file with original structure:")
    print("\n  fazip x <zipfile>")
    print("\n- List the contents of a zip file:")
    print("\n  fazip l <zipfile>")
    print("\n- Archive a file or directory:")
    print("\n  fazip a <zipfile> <files>")
    print("\n- Configuration menu:")
    print("\n  fazip config")
    print("\n- Display this help:")
    print("\n  fazip h")
    print()


def main():
    if 2 < len(sys.argv) == 3 and sys.argv[1] == 'x':
        do_extraction(sys.argv[2])
    elif 2 < len(sys.argv) == 3 and sys.argv[1] == 'l':
        do_list(sys.argv[2])
    elif len(sys.argv) >= 4 and sys.argv[1] == 'a':
        do_compression(sys.argv[2], sys.argv[3:])
    elif len(sys.argv) == 2 and sys.argv[1] == 'config':
        print(fazip.utils.HEADER)
        do_configuration()
    elif len(sys.argv) == 2 and sys.argv[1] == 'h':
        print_help()
    else:
        print_help(True)


if __name__ == '__main__':
    main()