如果有三個未知數，優化的最佳方法是什么？

Question

我制作了這段代碼來生成密碼作為練習，但我想知道是否有更聰明的方法來使用if 。 愚蠢地列出所有可能的情況，我感到很糟糕。 if變得難以理解，則嘗試將if放在其他人中。

import random
import re
import exrex

len_pwd = int(input('Length of the password : '))
upc = int(input('Upper case ? (0 = No ; 1 = Yes) : '))
nbr = int(input('Numbers ? (0 = No ; 1 = Yes) : '))
spe = int(input('Special char ? (0 = No ; 1 = Yes) : '))

pwd = ''

for i in range(len_pwd):
    if upc and not nbr and not spe:
        a = exrex.getone('[a-zA-Z]')
        pwd += a
    elif nbr and not upc and not spe:
        a = exrex.getone('[a-z\d]')
        pwd += a
    elif spe and not upc and not nbr:
        a = exrex.getone('[a-z"_$.+-/*=]')
        pwd += a
    elif nbr and spe and not upc:
        a = exrex.getone('[a-z\d"_$.+-/*=]')
        pwd += a
    elif upc and nbr and not spe:
        a = exrex.getone('[a-zA-Z\d]')
        pwd += a
    elif upc and spe and not nbr:
        a = exrex.getone('[a-zA-Z"_$.+-/*=]')
        pwd += a
    elif not upc and not nbr and not spe:
        print("Don't bother me.")
        break
    else:
        a = exrex.getone('[a-zA-Z\d"_$.+-/*=]')
        pwd += a

print(pwd)

謝謝：另外，如果代碼中的其他內容很丑，請告訴我）

Answer 1

您可以根據輸入構建正則表達式字符串：

reg = '[a-z'
if upc: reg += "A-Z"
if nbr: reg += "\\d"
if spe: reg += "_$.+-/*="
reg += "]"