Python3 初学实践案例（6）终极 CLI 密码生成程序实现

你的名字 2022-06-02 21:17 158阅读 0赞

# Python3 初学实践案例（6）终极 CLI 密码生成程序实现 #

在前面一章的学习中，我已经完成了一个一定扔出 6 的生成密码算法实现。但是只能将生成的密码输出到终端，并不能够将密码保存到文件，这样还是很有问题的。

随机生成的东西我记都记不住，回头找不回来就麻烦了。所以，我需要给它加上这个功能。

## 实现源码 ##

#!/usr/bin/env python3
    # -*- coding: UTF-8 -*-
    import random
    import argparse
    import os
    import time
    import clipboard
    
    # 设定密码保存路径
    DEFAULT_FILENAME = './password.txt'
    
    # 根据需要的结果，剪出一个数组
    def cutLength(leng, level):
        # 算法比较复杂，简化来说，就是已知数组长度，和数组内数字的和
        # 求一个随机的数组，满足上面的两个条件
        res = []
        for i in range(level, 1, -1):
            res.append(random.randint(1, leng - sum(res) - i + 1))
        res.append(leng - sum(res))
        # 因为第一位生成大数字的几率高于后面的几位，
        # 所以在得到结果后，随机排序一下,以期待更随机一些
        random.shuffle(res)
        return res
    # 根据上面生成的数组和对应的字典，制造一个密码数组
    def makePassword(dists, arr):
        res = []
        # 根据字典和数组，循环生成密码
        for i in range(len(arr)):
            # res += random.choices(dists[i], k=arr[i])
            for j in range(arr[i]):
                res += random.choice(dists[i])
        # 得到结果后，再一次随机排序
        random.shuffle(res)
        # 最后把数组变成字符串，并输出
        return ''.join(res)
    # 生成一个密码函数
    def getPassword(leng, level):
        # 根据密码长度和等级，去求一个满足条件的数组
        arr = cutLength(leng,level)
        # 制造字典
        str1 = '01'
        str2 = '23456789'
        str3 = 'abcdefghijkmnpqrstuvwxyz'
        str4 = 'ABCDEFGHJKMNPQRSTUVWXYZ'
        str5 = '_@!,.:;-=+/?'
    
        dists = {
                    1: [str1 + str2],
                    3: [str2, str3, str4],
                    4: [str2, str3, str4, str5]
                }
        # 生成密码
        return makePassword(dists[level], arr)
    
    # 将生成的密码写入文件
    def writePassword(name, passwd):
        # 如果文件不存在，则创建该文件
        if not os.path.exists(DEFAULT_FILENAME):
            f = open(DEFAULT_FILENAME, 'w')
            f.write('Here save the passwords generated by history\n\n')
            f.close()
    
        # 获取当前时间
        t = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
        # 将结果追加到文件中
        f = open(DEFAULT_FILENAME, 'a')
        f.write(name + '\t' + passwd + '\t' + t + '\n')
        f.close()
    
    # 输出结果
    def returnPassword(passwd, name):
        clipboard.copy(passwd)
        print('New password:\t\033[31m' + passwd + '\033[0m\nTip: The password has been copied to the clipboard')
        if name:
            print('And the password has been saved in the ' + DEFAULT_FILENAME + ' file')
        exit()
    
    # 主函数
    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='?')
        parser.add_argument('-n', '--name', help='Take a name for your password and Write it in ' + DEFAULT_FILENAME)
    
        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()
    
        # 默认密码等级为一般
        level = 3
    
        if args.simple:
            level = 1
        if args.difficult:
            level = 4
    
        # 去得到密码
        res = getPassword(length, level)
        # 如果密码需要保存，则写入文件
        if args.name:
            writePassword(args.name, res)
        # 返回结果
        returnPassword(res, args.name)

运行效果：

![生成密码最终运行效果图][SouthEast]

## 重点补充 ##

**新增可选命令参数**

我们需要保存密码，但是保存的是个啥密码我咋知道呢？所以，还是很有必要为这个密码取个名字的。这需要我们在过去密码的时候增加一个参数，这个参数可以缀一个值，这个值，就是我们为密码保存的名字。

代码如下：

parser.add_argument('-n', '--name', help='Take a name for your password and Write it in "./password.txt"')

> 所有加了 `-` 的参数都是可选项。

有了这样一个选项之后，我们就可以通过 `args.name` 拿到脚本执行时用户输入的名字参数是什么了。如果没有这个参数，则会返回 `None`。

**创建与写入文件以及判断文件是否存在**

我们设定了一个保存文件的路径为 `"./password.txt"` ，但是如果这个文件不存在呢？

所以我们首先需要判断这个文件存在不存在，这就需要引入 `os` 库来解决问题：