Lesson 7
Objects and Dictionaries
Chapter 5: Application of dictionaries
Querying the employee database
Continuing on from the previous task, let us now query employee names by searching for their IDs in the database.
You can just code in the same script file as load_database()
from earlier, since this is a continuation of the previous task.
Your follow-up task now is to write a function query_employees()
that takes two input arguments:
- a
dict
with employee IDs as keys and employee names as values (e.g. the return value of theload_database()
function above) - a
list
which contains a list of employee IDs whose names you wish to retrieve
The function should return a list of employee names corresponding to the given list of IDs. If an ID cannot be found in the database, the employee name should be set to None
.
Sample usage
>>> employee_dict = {"111": "Joe", "222": "Luca", "333": "Harry",
... "444": "William"}
...
>>> queries = ["444", "111", "555"]
>>> names = query_employees(employee_dict, queries)
>>> print(names)
['William', 'Joe', None]
>>> employee_dict = load_database("employees.txt")
>>> queries = ["14817102", "12345678", "61098940"]
>>> names = query_employees(employee_dict, queries)
>>> print(names)
['Agathe Davies', None, 'Loukios Rubio']