How to let the user add 'None' to an specific array in a dictionary?

[noahverner1995]

Ok so, the question may sound confusing at first but I'm gonna do my best to explain what I would like to learn in order to improve my programming skills.

Let's say I have a path in which exists 6 folders with the following file images:

Color:

• Amarillo.png
  
• Blanco.png
  
• Rojirosado.png
  
• Turquesa.png
  
• Verde_oscuro.png
  
• Zapote.png
  

Cuerpo:

• Cuerpo_cangrejo.png
  

Fondo:

• Oceano.png
  

Ojos:

• Antenas.png
  
• Pico.png
  
• Verticales.png
  

Pinzas:

• Pinzitas.png
  
• Pinzotas.png
  
• Pinzota_pinzita.png
  

Puas:

• Arena.png
  
• Marron.png
  
• Purpura.png
  
• Verde.png
  

Now, I want the above information to be stored in a dictionary for further use, so I run the code below in the same path in which the folders previously mentioned exist:

import os
# Main method
the_dictionary_list = {}

for name in os.listdir("."):
    if os.path.isdir(name):
        path = os.path.basename(name)
        print(f'\u001b[45m{path}\033[0m')
        list_of_file_contents = os.listdir(path)
        print(f'\033[46m{list_of_file_contents}')
        the_dictionary_list[path] = list_of_file_contents
        print('\n')
print('\u001b[43mthe_dictionary_list:\033[0m')
print(the_dictionary_list)

So after compiling the program above I get my dictionary:



But here's the problem: After creating the dictionary, how can I let the user decide in which arrays to add a 'None' string as a new value (i.e. not replacing the current ones), meaning that, for instance, if the user wanted to add 'None' only to Puas Array and Pinzas Array, it would generate the following output?

the_dictionary_list: {
'Color': ['Amarillo.png', 'Blanco.png','Rojirosado.png', 'Turquesa.png', 'Verde_oscuro.png', 'Zapote.png'],
'Cuerpo': ['Cuerpo_cangrejo.png'],
'Fondo': ['Oceano.png'],
'Ojos': ['Antenas.png', 'Pico.png', 'Verticales.png'],
'Pinzas': ['None', 'Pinzitas.png', 'Pinzotas.png', 'Pinzota_pinzita.png'],
'Puas': ['None', 'Arena.png', 'Marron.png', 'Purpura.png', 'Verde.png']}

[BashBedlam]

Something like this maybe?

the_dictionary_list = {
	'Color': ['Amarillo.png', 'Blanco.png','Rojirosado.png', 'Turquesa.png', 'Verde_oscuro.png', 'Zapote.png'],
	'Cuerpo': ['Cuerpo_cangrejo.png'],
	'Fondo': ['Oceano.png'],
	'Ojos': ['Antenas.png', 'Pico.png', 'Verticales.png'],
	'Pinzas': ['Pinzitas.png', 'Pinzotas.png', 'Pinzota_pinzita.png'],
	'Puas': ['Arena.png', 'Marron.png', 'Purpura.png', 'Verde.png']}

for key, value in the_dictionary_list.items () :
	answer = input (f'Would like to add "None" to {key}? ')
	if answer.lower () == 'y' :
		the_dictionary_list [key].insert (0, 'None')
	print (key, value, '\n')

[noahverner1995]

Something like this maybe?

the_dictionary_list = {
	'Color': ['Amarillo.png', 'Blanco.png','Rojirosado.png', 'Turquesa.png', 'Verde_oscuro.png', 'Zapote.png'],
	'Cuerpo': ['Cuerpo_cangrejo.png'],
	'Fondo': ['Oceano.png'],
	'Ojos': ['Antenas.png', 'Pico.png', 'Verticales.png'],
	'Pinzas': ['Pinzitas.png', 'Pinzotas.png', 'Pinzota_pinzita.png'],
	'Puas': ['Arena.png', 'Marron.png', 'Purpura.png', 'Verde.png']}

for key, value in the_dictionary_list.items () :
	answer = input (f'Would like to add "None" to {key}? ')
	if answer.lower () == 'y' :
		the_dictionary_list [key].insert (0, 'None')
	print (key, value, '\n')

Thank you for your answer , I also developed another one sometime after I created this thread, here it is:

entrada = input('All right, do you want to add a "None" item to an array in the dictionary? (y/n):')

while True:
    
    if entrada == 'y':
        key_entrada = input('Cool, now tell me at which key do you want me to add a "None" item? Type only a valid key name:')
        while True:            
            if key_entrada in the_dictionary_list:
                the_dictionary_list[key_entrada].insert(0, 'None')
                key_entrada = input('Done, do you want to add another one? (y/n):')
                if key_entrada == 'n':
                    entrada = 'n'
                    break
                if key_entrada == 'y':
                    key_entrada = input('Cool, now tell me at which key do you want me to add a "None" item? Type only a valid key name:')
                else:
                    key_entrada = input("Invalid Input, Type 'y' or 'n' without single quotation marks: ")
            else:
                key_entrada = input('That input does not exist in the dictionary, try again, Type only a valid key name:')               
                                   
    if entrada == 'n':
        print('weno')
        break  
    
    elif entrada != 'y' or entrada != 'n':
        entrada = input("Invalid Input, Type 'y' or 'n' without single quotation marks: ")

Now the only thing that doesn't let me sleep is to figure out how to avoid executing the the_dictionary_list[key_entrada].insert(0, 'None') if the current the_dictionary_list already has a key with 'None' as value inside its respective array , do you know how could I do that to improve the code above?

[ibreeden]

if None in the_dictionary_list[key_entrada]:
    print("None is already in the list")
else:
    the_dictionary_list[key_entrada].insert(0, 'None')

[noahverner1995]

if None in the_dictionary_list[key_entrada]:
    print("None is already in the list")
else:
    the_dictionary_list[key_entrada].insert(0, 'None')

Thank you again! Once more, I will update the final version of this program, now the user can choose whether adding a 'None' value to the arrays of one of the available keys only once, or not adding anything at all :

def existe(key, value, diccionario):
    return key in diccionario and value in diccionario[key]

entrada = input('All right, do you want to add a "None" item to an array in the dictionary? (y/n):')

while True:
    
    if entrada == 'y':
        key_entrada = input('Cool, now tell me at which key do you want me to add a "None" item? Type only a valid key name:')
        while True:            
            if key_entrada in the_dictionary_list:
                if existe(key_entrada, 'None', the_dictionary_list) == False:
                    the_dictionary_list[key_entrada].insert(0, 'None')
                    print('\u001b[43mthe_dictionary_list IS UPDATED:\033[0m')
                    print(the_dictionary_list)
                    print('\n')
                    key_entrada = input('Done, do you want to add another one? (y/n):')
                    if key_entrada == 'n':
                        entrada = 'n'
                        break
                    if key_entrada == 'y':
                        key_entrada = input('Cool, now tell me at which key do you want me to add a "None" item? Type only a valid key name:')
                    else:
                        key_entrada = input("Invalid Input, Type 'y' or 'n' without single quotation marks:")
                        while True:
                            if key_entrada == 'n':
                                entrada = 'n'
                                break
                            if key_entrada == 'y':
                                key_entrada = input('Cool, now tell me at which key do you want me to add a "None" item? Type only a valid key name:')
                                break
                            elif key_entrada != 'n' or key_entrada != 'y':
                                key_entrada = input("Invalid Input, Type 'y' or 'n' without single quotation marks:")
                                
                if existe(key_entrada, 'None', the_dictionary_list) == True:
                    key_entrada = input('That input DOES ALREADY EXIST in the dictionary, try again with another key name:') 
            
            elif entrada == 'n':
                break
            
            else:
                key_entrada = input('That input does not exist in the dictionary, try again, Type only a valid key name:')               
                                   
    if entrada == 'n':
        break  
    
    elif entrada != 'y' or entrada != 'n':
        entrada = input("Invalid Input, Type 'y' or 'n' without single quotation marks:")