Pasted

 
# 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.")