Menu.py

from tkinter import *
from tkinter import filedialog
import ntpath
import mimetypes
import os


class Menu:
    def __init__(self):
        print("Welcome!")

    @staticmethod
    def printoptions():
        print("What do you want to do, sir?\n"
              "1. Show me the list of last added files\n"
              "2. Upload file\n"
              "3. Download file\n"
              "4. Search for a File\n"
              "5. Send all files from SYNC folder\n"
              "If you want to finish working just write 'exit'\n"
              "If you want to clear screen, write 'cls'\n"
              "If you want to see instructions write 'help'\n")

    @staticmethod
    def userchoice(drive_functions, service):
        itemlist = []
        choice = input("> ")
        while choice != 'exit':
            if choice == '1':
                print("Write the amount of last added files you want to see: ")
                amountoffiles = input("> ")
                if amountoffiles.isdigit():
                    itemlist = drive_functions.files_list(amountoffiles, service)

            elif choice == '2':
                window = Tk()
                window.filename = filedialog.askopenfilename(initialdir='/', title="Select document",
                                                             filetypes=(("All Files", "*.*"), ("Text Files", "*.txt")))
                path = window.filename
                if path == '':
                    print("no file chosen")
                else:
                    filename = ntpath.basename(path)
                    mime = mimetypes.guess_type(path)
                    print(mime)
                    mimetype = mime[0]
                    drive_functions.files_upload(filename, path, mimetype)

            elif choice == '3':
                print("From the last chosen by you list choose the file number you want to download")
                file_number = input("> ")
                if file_number.isdigit():
                    file_number = int(file_number)
                    if len(itemlist) == 0:
                        print("You haven't seen any files list yet")
                    else:
                        googlefiletype = "vnd.google"
                        if googlefiletype in itemlist[file_number]['mimeType']:
                            print("You can't download Google file, sir")
                        else:
                            chosenid = itemlist[file_number]['id']
                            chosenname = itemlist[file_number]['name']
                            drive_functions.files_download(chosenid, chosenname)
                else:
                    print('wrong my friend')

            elif choice == '4':
                print("choose by what phrase do you want to find the file")
                phrase = input("> ")
                itemlist = drive_functions.files_search(100, phrase)
                print(itemlist)

            elif choice == '5':
                pathsync = r"D:\\PROGRAMOWANIE\\Python programy\\GoogleManager\\SYNC\\"
                listfiles = os.listdir(pathsync)
                if len(listfiles) == 0:
                    print("There are no files in SYNC dir")
                else:
                    for i in range(len(listfiles)):
                        temp_path = pathsync + listfiles[i]
                        mimesync = mimetypes.guess_type(temp_path)
                        filenamesync = ntpath.basename(temp_path)
                        mimetypesync = mimesync[0]
                        drive_functions.files_upload(filenamesync, temp_path, mimetypesync)
            elif choice == "cls":
                os.system("cls")
            elif choice == "help":
                Menu.printoptions()
            choice = input("> ")