字典查找无法识别初始输入

时间:2017-10-29 17:23:53

标签: python class object dictionary instance

Hello StackOverflow社区, 我非常接近完成员工数据库程序的创建,该程序实现了存储字典。我遇到的问题是无法找到在菜单提示之前存储的初始Employee对象。它们存在于数据库中,但我无法使用我的代码来操纵它们。我之后使用菜单功能添加的任何内容都可以被查找或删除。我不确定这里的问题是什么。我试图使字典全局无济于事。我使用的陈述几乎相同。

class Employee:
    def __init__(self, name, idNo, department, title):
        self.name = name
        self.idNo = idNo
        self.department = department
        self.title = title   

    def setName(newName):
        self.name = newName

    def setIDNo(newID):
        self.idNo = newID

    def setDepartment(newDept):
        self.department = newDept

    def setTitle(newTitle):
        self.title = newTitle

    def getName():
        return self.name

    def getIDNo():
        return self.idNo

    def getDepartment():
        return self.department

    def getTitle():
        return self.title

    def __str__(self):
         return "Name: {0} \nID number: {1} \nDepartment: {2} \nTitle: {3}\n".format(self.name, self.idNo, self.department, self.title)

def main():
    empDatabase = {}
    entry = input("Do You Want to Enter New Employee Information(Y/N): ")
    entry = entry.upper()
    if entry == 'Y':
       gate = True
       newName = input("Enter Employee Name: ")
       newID = input("Enter Employee ID Number: ")
       newDept = input("Enter Employee Department Name: ")
       newTitle = input("Enter Employee Job Title: ")
       empDatabase[newID] = Employee(newName, newID, newDept, newTitle)    
       another = input("Do You Want to Enter Another Employee Information(Y/N): ")
       another = another.upper()
       if another != 'Y':
          gate = False
       while gate == True:
          newName2 = input("Enter Employee Name: ")
          newID2 = input("Enter Employee ID Number: ")
          newDept2 = input("Enter Employee Department Name: ")
          newTitle2 = input("Enter Employee Job Title: ")
          empDatabase[newID2] = Employee(newName2, newID2, newDept2, newTitle2)
          another2 = input("Do You Want to Enter Another Employee Information(Y/N): ")
          another2 = another2.upper()
          if another2 != 'Y':
               gate = False
    boolGate = True
    while boolGate == True:
        print("\nPlease Select from the Following Menu: ")
        print("1. Look up an Employee in the Dictionary")
        print("2. Add a New Employee to the Dictionary")
        print("3. Delete an Employee from the Dictionary")
        print("4. Quit the Program")
        print()
        choice = eval(input("Please Enter a Number from the Menu: "))
        if choice == 1:
            idNo = eval(input("Please Enter the Employee ID You are Looking for: "))
            dic_lookup(idNo, empDatabase)
        elif choice == 2:
            empDatabase = dic_add(empDatabase)
        elif choice == 3:
            idNo = eval(input("Please Enter the Employee ID to Delete the Employee: "))
            dic_delete(idNo, empDatabase)
        else:
            boolGate = False
def dic_lookup(idNo, database):
    if idNo in database:
        print()
        print(database.get(idNo))
    else:
        print()
        print("This Employee ID is not Available in our Database")
def dic_add(database):
    print()
    addName = input("Enter New Employee Name: ")
    addID = eval(input("Enter New Employee ID Number: "))
    addDept = input("Enter New Employee Department Name: ")
    addTitle = input("Enter New Employee Job Title: ")
    database[addID] = Employee(addName, addID, addDept, addTitle)
    return database

def dic_delete(idNo, database):
    if idNo in database.keys():
        database.pop(idNo)
    else:
        print()
        print("Employee does not exist in database")
    return database  

main()

1 个答案:

答案 0 :(得分:0)

我在代码中添加了几个打印语句。 最初添加了前两个元素,使用菜单添加了第三个元素。 这是print的输出:

{'1':< main .Employee对象位于0x037FFDB0>,'2':< main .Employee对象位于0x037FFE30>,3:< main .Employee对象位于0x037FFA30>}

正如您所看到的,前两个键是字符串,第三个键是整数。

如果您修改代码(第45和55行)以将整数键添加到字典中:

empDatabase[int(newID)] = Employee(newName, newID, newDept, newTitle)
empDatabase[int(newID2)] = Employee(newName2, newID2, newDept2, newTitle2)

它似乎工作正常。