将python generate_domain修改成可以任意生成我想要的字典例如所有数字def generate_domain num = randomrandint1 4 # num = 3 num_list = 888 666 999 777 555 333 222 111 000 num_list2 = 5678 6789 365 888 666 999 777 55

def generate_domain(charset): num = random.randint(1, 4) if charset == 'num': num_list = ['888', '666', '999', '777', '555', '333', '222', '111', '000'] domain_name = ''.join(random.choice(num_list) for i in range(random.randint(1, 3))) elif charset == 'alpha': letters = 'abcdefghijklmnopqrstuvwxyz' domain_name = ''.join(random.choice(letters) for i in range(random.randint(4, 10))) elif charset == 'alphanum': letters = 'abcdefghijklmnopqrstuvwxyz0123456789' domain_name = ''.join(random.choice(letters) for i in range(random.randint(4, 10))) elif charset == 'custom': custom_set = input('Enter custom character set: ') domain_name = ''.join(random.choice(custom_set) for i in range(random.randint(4, 10))) else: return "Invalid charset" shuffled_domain = ''.join(random.sample(domain_name, len(domain_name))) return shuffled_domain + '.com