Python3 初学实践案例（4）可设定长度和密码复杂级别的生成密码脚本

ゞ浴缸里的玫瑰 2022-06-03 01:45 182阅读 0赞

# Python3 初学实践案例（4）可设定长度和密码复杂级别的生成密码脚本 #

在学习 `python` 的第一次实战中，我就完成了一个生成密码的脚本。原文链接：[http://blog.csdn.net/fungleo/article/details/78753940][http_blog.csdn.net_fungleo_article_details_78753940]

但是，在那次的实战中，最终的脚本只能设定一个参数，就是长度。我希望可以做一个更加完善的脚本，就是不但可以设定长度，并且可以设定密码复杂的等级。这样，才能满足我们在不同情况下所需要的密码。

另外，由于密码是在字典中随机取得字符组合在一起的。这个结果可能并不满足我们的想要的条件，因此，必须检查一下，如果不满足，则需要重新生成一个，直到满足为止。

好，设定我的程序的目标：

1.  随机生成一个密码
2.  可以指定密码的长度
3.  可以指定密码的复杂等级  
    
    
    1.  简单：由纯数字组成
    2.  一般：数字+小写字母+大写字母
    3.  复杂：数字+小写字母+大写字母+标点符号
4.  密码必须符合指定的复杂等级才输出
5.  密码最小长度为 4 位，因为复杂模式必须由四种元素构成

好，确定目标之后，我就开始准备实现这个脚本。由于之前我已经学习了 `argparse` 命令行参数处理的库，《[argparse 命令行参数库的使用][argparse]》，所以处理命令行参数对于我来说变得异常简单，我配置好之后，只需要把努力全部放在逻辑处理上即可。

## 实现代码 ##

#!/usr/bin/env python3
    # -*- coding: UTF-8 -*-
    import random
    import argparse
    import re
    
    # 用字典和长度随机生成一个密码
    def passwordMaker(leng, dicts):
        res = ''
        dictLen = len(dicts) - 1
        for i in range(leng):
            res += dicts[random.randint(0, dictLen)]
        return res
    
    # 检查密码的等级
    def checkPassword(passwd):
        res = ''
        if re.search(r'\d', passwd):
            res = 's'
            if re.search(r'[a-z]', passwd) and re.search(r'[A-Z]', passwd):
                res = 'c'
                if re.search(r'[_@!,.<>:;-=+/?]', passwd):
                    res = 'd'
        return res
    
    # 取得密码函数
    def getPassword(leng,level,dicts):
        res = passwordMaker(leng, dicts)
        # 检查密码是否符合期望的条件，如果不负责，则重新生成一遍
        if checkPassword(res) != level:
            return getPassword(leng, level, dicts)
        return res
    
    # 主函数
    if __name__ == "__main__":
        # 设置命令行参数
        parser = argparse.ArgumentParser()
        parser.description='This program is used to generate simple or complex passwords'
        parser.add_argument("-v", "--version",action='version', version='%(prog)s 1.0')
        parser.add_argument('length', type=int, help='The length of the password (Default 8)', nargs='?')
    
        group = parser.add_mutually_exclusive_group()
        group.add_argument("-s", "--simple", help="The password is made up of pure numbers", action="store_true")
        group.add_argument("-c", "--commonly", help="The password is made up of numbers and letters (Default)", action="store_true")
        group.add_argument("-d", "--difficult", help="The password is made up of numbers, letters, and punctuation", action="store_true")
    
        # 获取命令行参数结果
        args = parser.parse_args()
    
        # 确定密码长度为命令行设置或者默认为8
        length = args.length or 8
    
        # 如果密码长度小于 4 则提示并退出
        if length < 4:
            parser.print_usage()
            print('error: The password length must be greater than 3')
            exit()
    
        # 设定三个等级的密码生成的词典
        sDict = '0123456789'
        cDict = 'abcdefghijkmnpqrstuvwxyzABCDEFGHJKMNPQRSTUVWXYZ23456789'
        dDict = cDict + '_@!,.:;-=+/?'
        # 根据命令行条件，打印最终的密码
        if args.simple:
            print(getPassword(length, 's', sDict))
        elif args.difficult:
            print(getPassword(length, 'd', dDict))
        else:
            # 默认输出为 一般复杂程度的密码
            print(getPassword(length, 'c', cDict))

**运行结果如下**

![可设定长度和密码复杂级别的生成密码脚本][SouthEast]

## 完成后小结 ##

**`main` 函数**

之前在写代码的时候，没有关注过这个问题。但是今天看到一副图之后，深以为然，决定以后写代码就按照这个规范来写。

![python 代码组织规范][python]

定义一个主函数也是非常简单的，只需要在自己写的未封装的代码前面加上如下代码即可：

if __name__ == "__main__":

**正则没有研究好**

我总感觉下面这句：

if re.search(r'[a-z]', passwd) and re.search(r'[A-Z]', passwd):

可以优化成一句正则就可以了，但是死活没有想出来。反正不是这样的，高人给我指点一下呗：

if re.search(r'[A-Za-z]', passwd):

还有就是，标点符号的正则，我用的是 `[_@!,.<>:;-=+/?]`，这能叫正则吗？反正我使用 `[:punct:]` 没能实现我的想法。希望有高人指点呀。

好吧，总结到这里。

本文由 FungLeo 原创，允许转载，但转载必须保留首发链接。

[http_blog.csdn.net_fungleo_article_details_78753940]: http://blog.csdn.net/fungleo/article/details/78753940
[argparse]: http://blog.csdn.net/fungleo/article/details/78784180
[SouthEast]: /images/20220603/8cf5102e43ce49f8b6404e747dc84e2f.png
[python]: /images/20220603/5eb1cc3ee91a4502a37931da85a2e273.png