Question

所以我是python的新手，几天前开始学习...我遇到了一个练习项目，我们将创建一个密码储物柜，用于存储帐户及其各自的密码。

#! python3
# pw.py - An insecure password locker program.

PASSWORDS = {'email': 'F7minlBDDuvMJuxESSKHFhTxFtjVB6',
             'blog': 'VmALvQyKAxiVH5G8v01if1MLZF3sdt',
             'luggage': '12345'}

import sys
if len(sys.argv) < 2:
    print('Usage: python pw.py [account] - copy account password')
    sys.exit()

account = sys.argv[1]      # first command line arg is the account name
if account in PASSWORDS:
    pyperclip.copy(PASSWORDS[account])
    print('Password for ' + account + ' copied to clipboard.')
else:
    print('There is no account named ' + account)

那是代码。因此，我想到了一种方法，如果该帐户先前不存在，则使用新信息来更新PASSWORDS词典。所以我将这些代码行添加到else语句中

    print('Input the password for the said account to update the info.')
    PASSWORDS[account] = input()
    print('Info updated!')

我现在面临的问题是，在程序完成后，添加到字典中的值不会持续存在。

Answer 1

密码管理始终是一个非常棘手的话题，但是由于您只是开始而不是在询问安全性，因此让我们来看一下如何以一种非常简单的方式加载和编写简单的JSON文件。

首先，让我们将实际数据移动到一个单独的文件中，我们可以将其命名为database.json：

{"email": "F7minlBDDuvMJuxESSKHFhTxFtjVB6",
 "blog": "VmALvQyKAxiVH5G8v01if1MLZF3sdt",
 "luggage": "12345"}

然后我们像这样更改应用程序代码：

#!/usr/env python3
"""
 pw.py - An insecure password locker program.
"""
import sys
import pyperclip
import json
import os


# The path to the database file
DB_FILE = "~/Desktop/database.json"


def main():
    if len(sys.argv) < 2:
        print('Usage: python pw.py [account] - copy account password')
        sys.exit()

    account = sys.argv[1]      # first command line arg is the account name
    db = os.path.expanduser(DB_FILE)
    with open(db) as f:
        data = json.load(f)  # Load a dictionary for disk

    if account in data:
        pyperclip.copy(data[account])
        print('Password for ' + account + ' copied to clipboard.')
    else:
        print('There is no account named ' + account)
        print('Input the password for the said account to update the info.')
        new_passwd = {account: input()}  # create a dictionary with the new account data
        data.update(new_passwd)  # update the main dictionary with the new data

        with open(db, 'w') as f:
            json.dump(data, f)  # convert to JSON and write to disk

        print('Info updated!')


if __name__ == "__main__":
    main()

对于这样一个简单的任务，可以使用多种文件格式。 JSON是一种以逻辑且易于理解的方式映射到Python词典的格式，但是您也可以在CSV中实现它而无需增加太多复杂性。但是在这种应用程序中，您很快就会感到需要更实用，更强大的工具。例如，随着数据库的增长，将整个数据库加载到内存中可能不切实际。并可能引发更多的安全问题。

花点时间学习诸如此类的基础知识，然后探索module sqlite。它允许在单文件关系数据库中使用SQL，并为强大的应用程序和进一步的学习铺平了道路。

更新字典的键和值

1 个答案: