簡體   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