Python: Dateierweiterung prüfen und fordert erneut Eingabe, wenn ungültige Dateierweiterung

Question

Problem

eine while-Schleife Doing Dateierweiterung zu bestätigen. Wenn eine Dateierweiterung nicht .exe oder .bat ist, fragen Sie die Benutzereingabe erneut. Ich suche nach einer Lösung ohne importendswithbreak Funktionen zu verwenden.

Code

format = " " 
while file[:-4] != ".bat" and file[:-4] != ".exe": 
    format = input("Enter file you like to open: ") 
    if format[:-4] == ".bat" or format[:-4] == ".exe": 
     callFunction(format) 
    else: 
     file = input("Enter file you like to open: ") 

Answer 1

Sie brauchen nicht eine Schleife

def ask_exe(prompt='Executable file name? '): 
    name = input(prompt) 
    if name[-4:] in {'.exe', '.bat'}: return name 
    return ask_exe(prompt='The name has to end in ".exe" or ".bat", please retry: ') 

[keine Brüche, keine Importe, fast kein Code ...] Als

von ShadowRanger meinen Code festgestellt, dass set Notation für die Mitgliedschaft Test verwendet werden, suboptimal für Python-Versionen vor 3.2. Bei diesen älteren Versionen vermeidet die Verwendung einer tuple Berechnung das Set zur Laufzeit jedes Mal, wenn die Funktion ausgeführt wird.

... 
    # for python < 3.2 
    if name[-4:] in ('.exe', '.bat'): return name 
... 

Answer 2

Asking the user for input until they give a valid response zu folgen und mit os.path.splitext() die Dateierweiterung zu extrahieren:

import os 

ALLOWED_EXTENSTIONS = {".bat", ".exe"} 
while True: 
    filename = input("Enter file you like to open: ") 
    extension = os.path.splitext(filename)[1] 
    if extension in ALLOWED_EXTENSTIONS: 
     break 

with open(filename) as f: 
    # do smth with f 

---

Ohne break:

import os 

ALLOWED_EXTENSTIONS = {".bat", ".exe"} 
extension = None 
while extension not in ALLOWED_EXTENSTIONS: 
    filename = input("Enter file you like to open: ") 
    extension = os.path.splitext(filename)[1] 

with open(filename) as f: 
    # do smth with f 

---

Ohne break und ohne Importe:

ALLOWED_EXTENSTIONS = (".bat", ".exe") 
filename = "" 
while not filename.endswith(ALLOWED_EXTENSTIONS): 
    filename = input("Enter file you like to open: ") 

with open(filename) as f: 
    # do smth with f 

---

Ohne break und ohne Importe und ohne endswith():

ALLOWED_EXTENSTIONS = {"bat", "exe"} 
filename = "" 
while filename.rsplit(".",1)[-1] not in ALLOWED_EXTENSTIONS: 
    filename = input("Enter file you like to open: ") 

with open(filename) as f: 
    # do smth with f