Slumpade sittplatser

# coding: latin-1
# Python 3.4.1

import os,random,time

# få en lista av filer i mappen
# ex:
# classes/132A.txt
# classes/132B.txt
# classes/132C.txt

# varje fil innehåller en lista med
# klasskamrater med ett namn per rad

def selectclass(folder):

    if not os.path.isdir(folder):
        print("The folder '%s' does not exist!" % folder)
        return None

    dirfiles = os.listdir(folder)
    files = {}

    # fyll dict med filnamn:klassnamn
    for index in range(len(dirfiles)):
        filename = dirfiles[index]
        basename = os.path.basename(filename)
        if os.path.isfile(os.path.join(folder,filename)) and basename[-4:] == '.txt':
            files[basename[:-4]] = basename

    if len(files) < 1:
        print("No valid files in '%s'" % folder)
        return None


    # Fråga om vilken klass
    print("Please select one of the following:")

    for key in files:
        print("- %s" % key)

    # gör dics'ens värden till lista och tar ett slumpmässigt värde
    randomfile = list(files.keys())[random.randrange(len(files))]
    print("For example, type '%s'\n(without the apostrophes)\nOr type 'cancel' to change folder" % randomfile)

    userinput = ""
    while True:
        userinput = input("> ")

        if userinput in files:
            break
        else:
            print("'%s' does not exist among the list!" % userinput)

        if userinput.lower() == "cancel":
            return None

    print("'%s' selected!" % userinput)
    return files[userinput],userinput

def selectreadfolder():
    print("Please enter the folder containing the class lists")

    userinput = ""
    while True:
        userinput = input("> ")

        if os.path.isdir(userinput):
            print("Folder '%s' selected!" % userinput)
            break
        else:
            print("'%s' is not a valid folder!" % userinput)

    return userinput

def getclassmates(path):
    try:
        fileObj = open(path,"r")
    except FileNotFoundError:
        print("Unable to open file at '%s'!" % path)
        return None

    alldata = fileObj.read()
    return alldata.split("\n")


def randomise(classmates):
    randomclass = classmates.copy()

    for index in range(len(classmates),0,-1):
        if randomclass[index-1] == '':
            del randomclass[index-1]

    random.shuffle(randomclass)

    return randomclass

def selectwritefolder():
    print("Please enter the folder where you wish to save the new text files")

    userinput = ""
    while True:
        userinput = input("> ")

        if os.path.isdir(userinput):
            return userinput

        try:
            os.mkdir(userinput)
            return userinput
        except FileExistsError:
            return userinput
        except FileNotFoundError:
            print("Unable to create folder at '%s', destination not found!" % userinput)

def getfilename(folder,name,overflow,ext=".txt"):
    name = name + " (%s)" % ("000"[:-len(str(overflow))] + str(overflow))
    return os.path.join(folder,name+ext)

def save(classname,classlist,folder):
    # skapa namn på filen, använd datum i filens namn
    filename = "%s [%s]" % (classname,time.strftime("%A, %d %b %Y",time.localtime()))

    # hitta ett unikt namn
    overflow = 0
    while os.path.exists(getfilename(folder,filename,overflow)):
        overflow += 1

    fileObj = open(getfilename(folder,filename,overflow),"w")

    for index in range(len(classlist)):
        fileObj.write("%s: %s\n" % (index+1,classlist[index]))

    fileObj.close()

    print("Saved as '%s'" % getfilename(folder,filename,overflow))

#--------------#
# MAIN PROGRAM #
#--------------#

readfolder = ""
filename = ""
classname = ""

task = "all"
while True:
    if task == 'selectclass' or task == 'all':
        #välj mapp och klassnamn
        while True:
            # om användaren misslyckas pga vadsomhelst att välja klass
            # så ska hen välja om mapp.
            readfolder = selectreadfolder()
            filename,classname = selectclass(readfolder)
            if type(filename) == str:
                break

        # path = path till filen, mapp och filnamn tillsammans
        path = os.path.join(readfolder,filename)

    if task == 'switchfolder' or task == 'all':
        # välj mapp att spara de slumpade klass-filerna
        writefolder = selectwritefolder()

    # randomisera klass
    classmates = getclassmates(path)
    randomclass = randomise(classmates)

    if task == "savenew" or task == "all":
        # spara i en mapp
        save(classname,randomclass,writefolder)

    print("To quit, type 'quit'.")
    print("To change class, type 'selectclass'.")
    print("To change location for randomised files, type 'switchfolder'")
    print("To create another randomised file just press enter.")

    userinput = input("> ").lower()

    if userinput == 'quit':
        break
    elif userinput == 'selectclass':
        task = userinput
    elif userinput == 'switchfolder':
        task = userinput
    elif userinput == '':
        task = 'savenew'
    else:
        print("Unknown command ('%s')!" % userinput)
        task = "N/A"

print("Stopped.")