示例#1
0
def task_handler(service: AppService, namespace):
    if namespace.action == 'create':
        event = event_converter(namespace.event)
        task = service.create_task(user=namespace.user,
                                   name=namespace.name,
                                   description=namespace.description,
                                   start_date=namespace.start_date,
                                   end_date=namespace.end_date,
                                   parent_task_id=namespace.parent_task_id,
                                   priority=namespace.priority,
                                   event=event,
                                   status=namespace.status)
        print('Created task:')
        print(task)

    elif namespace.action == 'show':
        task_show_handler(service, namespace)

    elif namespace.action == 'edit':
        event = event_converter(namespace.event)
        task = service.update_task(user=namespace.user,
                                   task_id=namespace.task_id,
                                   name=namespace.name,
                                   description=namespace.description,
                                   status=namespace.status,
                                   priority=namespace.priority,
                                   start_date=namespace.start_date,
                                   end_date=namespace.end_date,
                                   event=event)
        print('Updated task:')
        print(task)

    elif namespace.action == 'share':
        service.share_task(user=namespace.user,
                           user_receiver=namespace.user_receiver,
                           task_id=namespace.task_id)
        print(
            f'Task(ID={namespace.task_id}) shared with user({namespace.user_receiver})'
        )

    elif namespace.action == 'unshare':
        service.unshare_task(user=namespace.user,
                             user_receiver=namespace.user_receiver,
                             task_id=namespace.task_id)

        print(
            f'Task(ID={namespace.task_id}) unshared with user({namespace.user_receiver})'
        )

    elif namespace.action == 'assign':
        service.assign_user(user=namespace.user,
                            task_id=namespace.task_id,
                            user_receiver=namespace.user_receiver)
        print(
            f'User({namespace.user_receiver}) assigned as task(ID={namespace.task_id}) executor'
        )

    elif namespace.action == 'add_subtask':
        service.add_subtask(user=namespace.user,
                            task_id=namespace.task_id,
                            parent_task_id=namespace.parent_task_id)
        print(
            f'Subtask(ID={namespace.task_id}) set as parent task of task(ID={namespace.parent_task_id})'
        )

    elif namespace.action == 'detach':
        service.detach_task(user=namespace.user, task_id=namespace.task_id)
        print(
            f'Subtask(ID={namespace.task_id}) is detached from parent task now'
        )

    elif namespace.action == 'done':
        task = service.get_task(user=namespace.user, task_id=namespace.task_id)
        service.change_task_status(user=namespace.user,
                                   task_id=namespace.task_id,
                                   status='done')
        print(
            f'Task(ID={namespace.task_id}) and its subtasks successfully done')

    elif namespace.action == 'archive':
        service.change_task_status(user=namespace.user,
                                   task_id=namespace.task_id,
                                   status='archived',
                                   apply_on_subtasks=namespace.subtasks)
        print(f'Task(ID={namespace.task_id}) has been archived')

    elif namespace.action == 'delete':
        service.delete_task(user=namespace.user, task_id=namespace.task_id)
        print(f'Task(ID={namespace.task_id}) has been deleted')

    elif namespace.action == 'search':
        tasks = service.get_tasks_by_name(user=namespace.user,
                                          name=namespace.name)
        if tasks:
            for task in tasks:
                print(task)
        else:
            print('Task(s) not found', file=sys.stderr)

    elif namespace.action == 'filter':
        if not any([
                namespace.name, namespace.description, namespace.start_date,
                namespace.end_date, namespace.parent_task_id,
                namespace.priority, namespace.status, namespace.event
        ]):
            print('You didnt specified filter arguments', file=sys.stderr)
            sys.exit(1)
        event = event_converter(namespace.event)
        tasks = service.get_filtered_tasks(
            user=namespace.user,
            name=namespace.name,
            description=namespace.description,
            start_date=namespace.start_date,
            end_date=namespace.end_date,
            parent_task_id=namespace.parent_task_id,
            priority=namespace.priority,
            status=namespace.status,
            event=event)
        if tasks:
            for task in tasks:
                print(task)
        else:
            print('Task(s) not found', file=sys.stderr)
示例#2
0
class TaskTest(unittest.TestCase):
    def setUp(self):
        session = mo.set_up_connection(DRIVER_NAME, CONNECTIONSTRING)
        self.serv = AppService(session)

    def test_create_task(self):
        parent_task = self.serv.create_task(
            user=TEST_USER,
            name=TEST_NAME,
        )
        task = self.serv.create_task(user=TEST_USER,
                                     name=TEST_NAME,
                                     description=TEST_DESCRIPTION,
                                     start_date=TEST_DATE_FIRST,
                                     end_date=TEST_DATE_SECOND,
                                     priority=TEST_PRIORITY_VALUE,
                                     status=TEST_STATUS_VALUE,
                                     parent_task_id=parent_task.id)
        self.assertEqual(task.owner, TEST_USER)
        self.assertEqual(task.name, TEST_NAME)
        self.assertEqual(task.description, TEST_DESCRIPTION)
        self.assertEqual(task.priority, TEST_PRIORITY)
        self.assertEqual(task.status, TEST_STATUS)
        self.assertEqual(task.start_date, TEST_DATE_FIRST)
        self.assertEqual(task.end_date, TEST_DATE_SECOND)
        self.assertEqual(task.parent_task_id, parent_task.id)

        with self.assertRaises(KeyError):
            self.serv.create_task(user=TEST_USER,
                                  name=TEST_NAME,
                                  status=TEST_RANDOM_STR)
        with self.assertRaises(KeyError):
            self.serv.create_task(user=TEST_USER,
                                  name=TEST_NAME,
                                  priority=TEST_RANDOM_STR)

        plan = self.serv.create_plan(user=TEST_USER,
                                     task_id=task.id,
                                     period=TEST_PERIOD_VALUE,
                                     period_amount=TEST_RANDOM_INT)

        with self.assertRaises(ValueError):
            subtask = self.serv.create_task(user=TEST_USER,
                                            name=TEST_NAME,
                                            parent_task_id=task.id)

    def test_update_task(self):
        task = self.serv.create_task(user=TEST_USER,
                                     name='TEST_NAME',
                                     description='TEST_DESCRIPTION',
                                     start_date=datetime(2018, 4, 4, 19, 00),
                                     end_date=datetime(2018, 4, 4, 21, 00),
                                     priority='low',
                                     status='todo')
        self.serv.update_task(user=TEST_USER,
                              task_id=task.id,
                              name=TEST_NAME,
                              description=TEST_DESCRIPTION,
                              start_date=TEST_DATE_FIRST,
                              end_date=TEST_DATE_SECOND,
                              priority=TEST_PRIORITY_VALUE,
                              status=TEST_STATUS_VALUE)
        self.assertEqual(task.name, TEST_NAME)
        self.assertEqual(task.description, TEST_DESCRIPTION)
        self.assertEqual(task.priority, TEST_PRIORITY)
        self.assertEqual(task.status, TEST_STATUS)
        self.assertEqual(task.start_date, TEST_DATE_FIRST)
        self.assertEqual(task.end_date, TEST_DATE_SECOND)

        with self.assertRaises(ValueError):
            self.serv.update_task(user=TEST_USER,
                                  task_id=task.id,
                                  name=TEST_NAME,
                                  start_date=TEST_DATE_SECOND,
                                  end_date=TEST_DATE_FIRST)

        with self.assertRaises(ValueError):
            self.serv.update_task(user=TEST_USER,
                                  task_id=task.id,
                                  name=TEST_NAME,
                                  end_date=TEST_DATE_FIRST)

        with self.assertRaises(ValueError):
            self.serv.update_task(user=TEST_USER,
                                  task_id=task.id,
                                  name=TEST_NAME,
                                  start_date=TEST_DATE_SECOND)

        with self.assertRaises(KeyError):
            self.serv.update_task(user=TEST_USER,
                                  task_id=task.id,
                                  name=TEST_NAME,
                                  priority=TEST_RANDOM_STR)

        with self.assertRaises(KeyError):
            self.serv.update_task(user=TEST_USER,
                                  task_id=task.id,
                                  name=TEST_NAME,
                                  status=TEST_RANDOM_STR)

        with self.assertRaises(KeyError):
            self.serv.update_task(user=TEST_USER,
                                  task_id=task.id,
                                  name=TEST_NAME,
                                  status=TEST_RANDOM_STR)

    def test_get_task_user_relation(self):
        task = self.serv.create_task(user=TEST_USER, name=TEST_NAME)
        rel = self.serv.get_task_user_relation(user=TEST_USER, task_id=task.id)
        self.assertTrue(rel)

        rel = self.serv.get_task_user_relation(user=TEST_USER, task_id=100)
        self.assertIsNone(rel)
        rel = self.serv.get_task_user_relation(user=TEST_RECEIVER,
                                               task_id=task.id)
        self.assertIsNone(rel)

    def test_get_task(self):
        task_created = self.serv.create_task(user=TEST_USER, name=TEST_NAME)
        task_expected = self.serv.get_task(user=TEST_USER,
                                           task_id=task_created.id)
        self.assertEqual(task_created, task_expected)

        with self.assertRaises(ex.ObjectNotFound):
            self.serv.get_task(user=TEST_RECEIVER, task_id=task_created.id)
            self.serv.get_task(user=TEST_USER, task_id=TEST_RANDOM_INT)

    def test_assign_user(self):
        task = self.serv.create_task(user=TEST_USER, name=TEST_NAME)

        self.serv.assign_user(user=TEST_USER,
                              task_id=task.id,
                              user_receiver=TEST_RECEIVER)

        self.assertTrue(task.assigned, TEST_RECEIVER)
        self.assertIn(TEST_RECEIVER, [rel.user for rel in task.members])

        with self.assertWarns(ex.RedundancyAction):
            self.serv.assign_user(user=TEST_USER,
                                  task_id=task.id,
                                  user_receiver=TEST_RECEIVER)

    def test_share_task(self):
        task = self.serv.create_task(user=TEST_USER, name=TEST_NAME)

        with self.assertRaises(ex.ObjectNotFound):
            self.serv.get_task(user=TEST_RECEIVER, task_id=task.id)

        self.serv.share_task(user=TEST_USER,
                             task_id=task.id,
                             user_receiver=TEST_RECEIVER)
        task_rec = self.serv.get_task(user=TEST_RECEIVER, task_id=task.id)
        self.assertEqual(task, task_rec)

        with self.assertWarns(ex.RedundancyAction):
            self.serv.share_task(user=TEST_USER,
                                 task_id=task.id,
                                 user_receiver=TEST_RECEIVER)

    def test_unshare_task(self):
        task = self.serv.create_task(user=TEST_USER, name=TEST_NAME)

        with self.assertRaises(ValueError):
            self.serv.unshare_task(user=TEST_USER,
                                   task_id=task.id,
                                   user_receiver=TEST_USER)
        with self.assertRaises(ValueError):
            self.serv.unshare_task(user=TEST_USER,
                                   task_id=task.id,
                                   user_receiver=TEST_RECEIVER)

        self.serv.share_task(user=TEST_USER,
                             task_id=task.id,
                             user_receiver=TEST_RECEIVER)
        self.serv.unshare_task(user=TEST_USER,
                               task_id=task.id,
                               user_receiver=TEST_RECEIVER)
        self.assertNotIn(TEST_RECEIVER, [rel.user for rel in task.members])

    def test_delete_task(self):
        with self.assertRaises(ex.ObjectNotFound):
            self.serv.delete_task(user=TEST_USER, task_id=TEST_RANDOM_INT)

        task = self.serv.create_task(user=TEST_USER, name=TEST_NAME)
        self.serv.delete_task(user=TEST_USER, task_id=task.id)

    def test_add_subtask(self):
        task = self.serv.create_task(user=TEST_USER, name=TEST_NAME)
        subtask = self.serv.create_task(user=TEST_USER, name=TEST_NAME)

        plan = self.serv.create_plan(user=TEST_USER,
                                     task_id=task.id,
                                     period='min',
                                     period_amount=10)

        with self.assertRaises(ValueError):

            self.serv.add_subtask(user=TEST_USER,
                                  task_id=subtask.id,
                                  parent_task_id=task.id)

        self.serv.delete_plan(user=TEST_USER, plan_id=plan.id)

        self.serv.add_subtask(user=TEST_USER,
                              task_id=subtask.id,
                              parent_task_id=task.id)

        with self.assertRaises(ValueError):
            self.serv.add_subtask(user=TEST_USER,
                                  task_id=subtask.id,
                                  parent_task_id=task.id)

        with self.assertRaises(ValueError):
            self.serv.add_subtask(user=TEST_USER,
                                  task_id=task.id,
                                  parent_task_id=task.id)
        with self.assertRaises(ValueError):
            self.serv.add_subtask(user=TEST_USER,
                                  task_id=task.id,
                                  parent_task_id=subtask.id)

    def test_detach_task(self):
        task = self.serv.create_task(user=TEST_USER, name=TEST_NAME)
        subtask = self.serv.create_task(user=TEST_USER,
                                        name=TEST_NAME,
                                        parent_task_id=task.id)

        self.serv.detach_task(TEST_USER, subtask.id)
        self.assertIsNone(subtask.parent_task_id)

        with self.assertRaises(ValueError):
            self.serv.detach_task(TEST_USER, subtask.id)

    def test_change_task_status(self):
        task = self.serv.create_task(user=TEST_USER, name=TEST_NAME)
        self.serv.change_task_status(user=TEST_USER,
                                     task_id=task.id,
                                     status=mo.TaskStatus.DONE.value)
        self.assertEqual(task.status, mo.TaskStatus.DONE)

        subtask = self.serv.create_task(user=TEST_USER, name=TEST_NAME)
        self.serv.add_subtask(user=TEST_USER,
                              task_id=subtask.id,
                              parent_task_id=task.id)

        self.serv.change_task_status(user=TEST_USER,
                                     task_id=task.id,
                                     status=mo.TaskStatus.DONE.value)

        self.assertEqual(subtask.status, mo.TaskStatus.DONE)

        with self.assertRaises(KeyError):
            self.serv.change_task_status(user=TEST_USER,
                                         task_id=task.id,
                                         status=TEST_RANDOM_STR)

    def test_get_filtered_tasks(self):
        task = self.serv.create_task(user=TEST_USER,
                                     name=TEST_NAME,
                                     description=TEST_DESCRIPTION,
                                     status=TEST_STATUS_VALUE,
                                     priority=TEST_PRIORITY_VALUE,
                                     start_date=TEST_DATE_FIRST,
                                     end_date=TEST_DATE_THIRD)
        tasks = self.serv.get_filtered_tasks(user=TEST_USER,
                                             name=TEST_NAME,
                                             description=TEST_DESCRIPTION,
                                             start_date=TEST_DATE_FIRST,
                                             end_date=TEST_DATE_THIRD,
                                             priority=TEST_PRIORITY_VALUE,
                                             status=TEST_STATUS_VALUE)
        self.assertEqual(len(tasks), 0)