import sys,re tail_symbol = "[^a-zA-Z0-9]{0,3}$" pat = [ "^digi .*[nN][yY][oO]" + tail_symbol, "^petit .*[nN][yY][uU]" + tail_symbol, "^rabi .*[a-zA-Z0-9]", "^gema .*[gG][eE][mM][aA]" + tail_symbol, "^piyo .*[pP][yY][oO]" + tail_symbol ] pattern = re.compile("|".join(pat)) for s in sys.stdin: print("CORRECT (maybe)" if pattern.match(s) else "WRONG!")