繁体   English   中英

无法编写条件来更新python中的字典值

[英]Unable to write condition to update dictionary value in python

请帮忙! 如何在我现有的代码中编写条件,将检查如下。 请看注释块。 我无法在字典/列表中附加值。 我需要编写/更改什么?

  1. 当用户返回主菜单并使用相同的密钥文件再次运行加密过程但以不同的名称保存加密文件时。 用户可以使用相同的密钥创建尽可能多的加密文件。

我的尝试:

import Encrypt
from collections import defaultdict

key_value_dict = {}

def encrypt_file():
    try:

        txtfile = input("Your plain txt file that you want to encrypt : ")
        encrypt_file = input("Directory to save the encrypted file : ")
        key = input("Key for encryption : ")

        # process for encryption
        Encrypt.encrypt_file(txtfile, encrypt_file, key)

        # ----------------------------------------------------------------
        key_value_dict = defaultdict(list, {key: encrypt_file})

        print(key_value_dict)

        key_value_dict [key].append(encrypt_file)
        print(key_value_dict )
        # --------------------------------------------------------------------
    except FileNotFoundError:
        print("File Not Found!")


def menu():
    selection = True
    while selection:
        print("""
MAIN MENU:
[1] Encrypt files using existing keys.
[2] Exit.
        """)
        try:  
            selection = input("Please select : ")

            if selection == "1":
                encrypt_file()  # call function
            elif selection == "2":
                print("\n[4] Keys are used for encryption:", "\n", key_value_dict)
                selection = None
            else:
                print("\n Invalid selection! Try again")
        except ValueError:  
            print("\n Exception: Invalid user input!")

# Main function
if __name__ == '__main__':
    menu()  

如果我理解正确,我认为您不需要 defaultdict

尝试这个:

# define this first outside the function like you have
encrypt_key_value_dict = {}

# Within your code block, get rid of the assignment to default dict
if encrypt_key_value_dict[key]:
    encrypt_key_value_dict.append(encrypt_file)
else:
    encrypt_key_value_dict[key] = [encrypt_file]

我无法从您的代码中看到您的密钥在哪里被传递等,但我相信您可以弄清楚。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM