Description:
A supervisor is using the TaskManagementSystem to manage tasks for his workers. The system can add users, assign tasks to users, and get task(s) that belong to a specific user.
Restriction:
You can not modify user or task class. self.users and self.tasks in TaskManagementSystem has to remain as lists. You can introduce new methods and class variables.
Goals:
Fix the syntax and semantic bugs that prevent the program from running correctly.
Change get_user_tasks() to return list of task names instead of task objects.
Add error handling to add_user(), add_task(), and get_user_tasks() methods
Optimize TaskManagementSystem to handle large number of users and tasks
Description:
A supervisor is using the TaskManagementSystem to manage tasks for his
workers. The system can add users, assign tasks to users, and get
task(s) that belong to a specific user.
Restriction:
You can not modify user or task class. self.users and self.tasks in
`TaskManagementSystem has to remain as lists. You can introduce new
methods and class variables. Goals:
Fix the syntax and semantic bugs that prevent the program from running correctly.
Change get_user_tasks() to return list of task names instead of task objects.
Add error handling to add_user(), add_task(), and get_user_tasks() methods
Optimize TaskManagementSystem to handle large number of users and tasks
CODE PROVIDED
Code provided at interview:
#!/usr/bin/env python2.7
# encoding: utf-8
class user(object):
def __init__(self, user_id, name):
'''
Task object. This code can not be modified.
@ user_id: int
@ name: string
'''
self.user_id = user_id
self.name = name
class task(object):
def __init__(self, user_id, task_name):
'''
Task object. This code can not be modified.
@ user_id: int
@ task_name: string
'''
self.user_id = user_id
self.task_name = task_name
class TaskManagementSystem(object):
def __init__(self):
self.users = [] # stores a list of user objects
self.tasks = [] # stores a list of task objects
def add_user(self, user_name):
'''
Add a new user
@ user_name: string
Condition:
Every user should have a unique name
Every user should have a unique id
'''
unique_id = 0
for user in self.users:
if user.user_id > unique_id:
unique_id = user.id
self.users.append(user(unique_id, user_name))
def add_task(self, user_name, task_name):
'''
Add a task for a user
@ user_name: string
@ task_name: string
Condition:
The user should not have two or more tasks with the same name
'''
for user in self.users:
if user.name = user_name:
user_id = user.user_id
self.users.append(task(user_id, task_name))
def get_user_tasks(self, user_name):
'''
Get task(s) that belongs to the specified user name
@ user_name: string
'''
user_tasks = []
for user in self.users:
if user.name == user_name:
for task in self.tasks:
if task.user_id == user.user_id:
user_tasks.append(task)
return user_tasks
if __name__ == "__main__":
tms = TaskManagementSystem()
tms.add_user('Bob')
tms.add_task('Bob', 'laundry')
tms.add_task('Bob', 'grocery')
tms.add_task('Bob', 'daycare')
print tms.get_user_tasks('Bob')
# should print: ['laundry', 'grocery', 'daycare']
#!/usr/bin/env python2.7
# encoding: utf-8
class user(object):
def __init__(self, user_id, name):
'''
Task object. This code can not be modified.
@ user_id: int
@ name: string
'''
self.user_id = user_id
self.name = name
class task(object):
def __init__(self, user_id, task_name):
'''
Task object. This code can not be modified.
@ user_id: int
@ task_name: string
'''
self.user_id = user_id
self.task_name = task_name
class TaskManagementSystem(object):
def __init__(self):
self.users = [] # stores a list of user objects
self.tasks = [] # stores a list of task objects
def add_user(self, user_name):
'''
Add a new user
@ user_name: string
Condition:
Every user should have a unique name
Every user should have a unique id
'''
unique_id = 0
for user in self.users:
if user.user_id > unique_id:
unique_id = user.id
self.users.append(user(unique_id, user_name))
def add_task(self, user_name, task_name):
'''
Add a task for a user
@ user_name: string
@ task_name: string
Condition:
The user should not have two or more tasks with the same name
'''
for user in self.users:
if user.name = user_name:
user_id = user.user_id
self.users.append(task(user_id, task_name))
def get_user_tasks(self, user_name):
'''
Get task(s) that belongs to the specified user name
@ user_name: string
'''
user_tasks = []
for user in self.users:
if user.name == user_name:
for task in self.tasks:
if task.user_id == user.user_id:
user_tasks.append(task)
return user_tasks
if __name__ == "__main__":
tms = TaskManagementSystem()
tms.add_user('Bob')
tms.add_task('Bob', 'laundry')
tms.add_task('Bob', 'grocery')
tms.add_task('Bob', 'daycare')
print tms.get_user_tasks('Bob')
# should print: ['laundry', 'grocery', 'daycare']
MY ANSWER
Here is the code I handed in as my answer:
