def main():
""" The function is the entry point to the Py_Tracker app.
It calls the parser and processes the parser output."""
logger = get_logger().getChild('console_main')
configure_logger(logger)
logger.info('Started')
parser = get_parser()
command_dict = vars(parser.parse_args())
if len(sys.argv) == 1:
parser.print_help(sys.stderr)
sys.exit(1)
command = command_dict.pop(ParserCommands.COMMAND)
logger.debug('command {}'.format(command))
user = ConfigManager().get_default_user()
if command == ParserObjects.USER:
subcommand = command_dict.get(ParserCommands.SUBCOMMAND, None)
if subcommand == UserCommands.SET:
ConfigManager().set_default_user(command_dict[UserCommands.NAME])
print("Hi, {}! Glad to see you!".format(
command_dict[UserCommands.NAME]))
return 0
if user is None:
print(
"Hi, dear User. Please introduce yourself to py_tracker by calling 'py_tracker user set YOUR_NAME''."
)
return 0
inteface = Interface(DB(), user)
subcommand = None
if command is not None and command != CheckCommand.CHECK:
subcommand = command_dict.pop(ParserCommands.SUBCOMMAND)
logger.debug('subcommand {}'.format(subcommand))
try:
if command == ParserObjects.TASK:
if subcommand == TaskCommands.AddSubcommand.ADD:
try:
task_id = inteface.create_task(**command_dict)
print("Task created. Its ID is {}".format(str(task_id)))
except EndTimeOverflowError as e:
print(
"! Oh! The subtask can't end later than its parent. Please check task id {}"
.format(e.args),
file=sys.stderr)
except ValueError:
print(
"! Oh! The start time couldn't be less or than end time",
file=sys.stderr)
elif subcommand == TaskCommands.EditSubcommand.EDIT:
task_id = command_dict[TaskCommands.EditSubcommand.ID]
if command_dict[TaskCommands.EditSubcommand.
EDIT_KIND] == TaskCommands.EditSubcommand.SET:
if command_dict[TaskAttributes.PRIORITY] is not None:
priority = command_dict[TaskAttributes.PRIORITY]
inteface.task_set_attribute(task_id,
TaskAttributes.PRIORITY,
priority)
elif command_dict[TaskAttributes.STATUS] is not None:
status = command_dict[TaskAttributes.STATUS]
inteface.task_set_attribute(task_id,
TaskAttributes.STATUS,
status)
elif command_dict[TaskAttributes.TITLE] is not None:
title = command_dict[TaskAttributes.TITLE]
inteface.task_set_attribute(task_id,
TaskAttributes.TITLE,
title)
elif command_dict[TaskAttributes.START_TIME] is not None:
try:
date = command_dict[TaskAttributes.START_TIME]
inteface.task_set_attribute(
task_id, TaskAttributes.START_TIME, date)
except EndTimeOverflowError:
print(
"! Oh! The start time conflicts with task end time",
file=sys.stderr)
elif command_dict[TaskAttributes.END_TIME] is not None:
try:
date = command_dict[TaskAttributes.END_TIME]
inteface.task_set_attribute(
task_id, TaskAttributes.END_TIME, date)
except EndTimeOverflowError:
print(
"! Oh! The end time conflicts with the start time or with parent task end time",
file=sys.stderr)
elif command_dict[
TaskCommands.EditSubcommand.
EDIT_KIND] == TaskCommands.EditSubcommand.UNSET:
if command_dict[TaskAttributes.START_TIME]:
try:
inteface.task_set_attribute(
task_id, TaskAttributes.START_TIME, None)
except EndTimeOverflowError:
print(
"! Oh! The start time conflicts with task end time",
file=sys.stderr)
elif command_dict[TaskAttributes.END_TIME]:
try:
inteface.task_set_attribute(
task_id, TaskAttributes.END_TIME, None)
except EndTimeOverflowError:
print(
"! Oh! The end time conflicts with the start time or with parent task end time",
file=sys.stderr)
elif command_dict[
TaskCommands.EditSubcommand.
EDIT_KIND] == TaskCommands.EditSubcommand.ADD:
if command_dict[TaskAttributes.TAGS] is not None:
tag = command_dict[TaskAttributes.TAGS]
inteface.task_add_attribute(task_id,
TaskAttributes.TAGS, tag)
elif command_dict[TaskAttributes.CAN_EDIT] is not None:
user = command_dict[TaskAttributes.CAN_EDIT]
inteface.task_add_attribute(task_id,
TaskAttributes.CAN_EDIT,
user)
elif command_dict[TaskAttributes.REMIND_TIMES] is not None:
reminder = command_dict[TaskAttributes.REMIND_TIMES]
inteface.task_add_attribute(
task_id, TaskAttributes.REMIND_TIMES, reminder)
elif command_dict[TaskCommands.EditSubcommand.
EDIT_KIND] == TaskCommands.EditSubcommand.RM:
if command_dict[TaskAttributes.TAGS] is not None:
tag = command_dict[TaskAttributes.TAGS]
inteface.task_remove_attribute(task_id,
TaskAttributes.TAGS,
tag)
elif command_dict[TaskAttributes.CAN_EDIT] is not None:
user = command_dict[TaskAttributes.CAN_EDIT]
inteface.task_remove_attribute(task_id,
TaskAttributes.CAN_EDIT,
user)
elif command_dict[TaskAttributes.REMIND_TIMES] is not None:
reminder = command_dict[TaskAttributes.REMIND_TIMES]
inteface.task_remove_attribute(
task_id, TaskAttributes.REMIND_TIMES, reminder)
elif subcommand == TaskCommands.RemoveSubcommand.RM:
try:
if command_dict[TaskCommands.RemoveSubcommand.F]:
for k in command_dict[TaskAttributes.UID]:
inteface.remove_with_subtasks(k)
else:
for k in command_dict[TaskAttributes.UID]:
inteface.remove_task(k)
except SubtasksNotRemovedError:
print(
"! Oh! The task you're trying to delete has subtasks. Please remove them or use -f flag",
file=sys.stderr)
elif subcommand == TaskCommands.FindSubcommand.FIND:
tasks = list(inteface.find_tasks(**command_dict).values())
for t in tasks:
printers.simple_task_printer(t)
elif subcommand == TaskCommands.PrintSubcommand.PRINT:
if command_dict[TaskCommands.PrintSubcommand.ID]:
task = inteface.get_task(
command_dict[TaskCommands.PrintSubcommand.ID])
if command_dict[TaskCommands.PrintSubcommand.SUBTASKS]:
printers.complex_task_printer(
task,
inteface.tasks_manager.tasks,
wide_print=command_dict[
TaskCommands.PrintSubcommand.WIDE])
else:
printers.simple_task_printer(
task,
wide_print=command_dict[
TaskCommands.PrintSubcommand.WIDE])
else:
tasks_d = inteface.tasks_manager.tasks
printers.hierarchy_printer(
tasks_d,
wide_print=command_dict[
TaskCommands.PrintSubcommand.WIDE])
elif subcommand == TaskCommands.CompleteSubcommand.COMPLETE:
inteface.complete_task(command_dict[TaskAttributes.UID])
print('Congratulations! :)')
elif command == CheckCommand.CHECK:
date = dt.datetime.now()
actual_tasks, reminders = inteface.check_time(date)
priority = command_dict[CheckCommand.PRIORITY]
status = command_dict[CheckCommand.STATUS]
for each in reminders:
if printers.task_satisfies(each,
priority=priority,
status=status):
printers.simple_reminder_printer(each)
printers.simple_actual_tasks_printer(**actual_tasks,
priority=priority,
status=status)
elif command == ParserObjects.PLAN:
if subcommand == PlanCommands.AddSubcommand.ADD:
period = parse_period(command_dict)
end_time = command_dict[PlanCommands.AddSubcommand.FINISH]
task_id = command_dict[PlanCommands.AddSubcommand.TASK_ID]
task_template = inteface.get_task(task_id)
try:
inteface.add_periodic_plan(period, task_template, task_id,
end_time)
except NoTimeValueError:
print(
"! Oh! Task '{}' has no start time specified. Can't set up a plan for it"
.format(task_id),
file=sys.stderr)
except SubtasksNotRemovedError:
print(
"! Oh! Plans can only work correctly with tasks without subtasks. Please try another task ",
file=sys.stderr)
inteface.check_plans()
elif subcommand == PlanCommands.RemoveSubcommand.RM:
for k in command_dict[PlanCommands.RemoveSubcommand.PLAN_ID]:
inteface.rm_periodic_plan(k)
elif subcommand == PlanCommands.PrintSubcommand.PRINT:
if command_dict[PlanCommands.PrintSubcommand.ID] is not None:
printers.simple_plan_printer(inteface.plans_manager.plans[
command_dict[PlanCommands.PrintSubcommand.ID]])
elif command_dict[
PlanCommands.PrintSubcommand.TASK_ID] is not None:
for plan in inteface.get_plans_by_task_id(command_dict[
PlanCommands.PrintSubcommand.TASK_ID]):
printers.simple_plan_printer(plan)
else:
for k, v in inteface.plans_manager.plans.items():
printers.simple_plan_printer(v)
elif command == ParserObjects.USER:
if subcommand == UserCommands.GET:
print('USERNAME: {}'.format(inteface.current_user))
logger.info('Exited successfully')
return 0
except PermissionError as e:
logger.error(e)
print("! Oh! Permission denied for user: {}".format(
inteface.current_user),
file=sys.stderr)
return 1
except KeyError as e:
logger.error(e)
print(
"! Oh! The key {} can't be found. Please check if it is correct or if it exists"
.format(e),
file=sys.stderr)
return 1
except Exception as e:
logger.error(e)
print('! Oh! Something went wrong. Program quited', file=sys.stderr)
return 1
class InterfaceTestCase(unittest.TestCase):
TITLE_1 = 'title'
TITLE_2 = 'title 2'
TITLE_3 = 'title 3'
TITLE_4 = 'title 4'
USER_1 = 'user_1'
USER_2 = 'user_2'
CHECK_TIME = dt.datetime.now()
TIMEDELTA = dt.timedelta(minutes=30)
def setUp(self):
self.task_starts_at_check = Task(title=self.TITLE_1,
author=self.USER_1,
start_time=self.CHECK_TIME,
end_time=self.CHECK_TIME +
4 * self.TIMEDELTA)
self.task_starts_at_check_id = self.task_starts_at_check.get_attribute(
TaskAttributes.UID)
self.task_around_check = Task(
title=self.TITLE_2,
author=self.USER_2,
start_time=self.CHECK_TIME - self.TIMEDELTA,
end_time=self.CHECK_TIME + self.TIMEDELTA)
self.task_around_check_id = self.task_around_check.get_attribute(
TaskAttributes.UID)
self.db_emulator = DBEmulator()
self.db_emulator.tasks_storage = {
self.task_starts_at_check_id: self.task_starts_at_check,
self.task_around_check_id: self.task_around_check
}
self.interface = Interface(self.db_emulator, self.USER_1)
def tearDown(self):
self.task_starts_at_check = None
self.task_starts_at_check_id = None
self.task_around_check = None
self.task_around_check_id = None
self.task_late = None
self.task_late_id = None
self.task_ends_at_check = None
self.task_ends_at_check_id = None
self.interface = None
self.db_emulator = None
def test_add_task(self):
task_id = self.interface.create_task(title=self.TITLE_3,
start_time=self.CHECK_TIME +
self.TIMEDELTA)
self.assertTrue(task_id in self.db_emulator.tasks_storage)
def test_db_equalty_task(self):
self.assertDictEqual(self.db_emulator.tasks_storage,
self.interface.tasks_manager.tasks)
self.assertDictEqual(self.db_emulator.plans_storage,
self.interface.plans_manager.plans)
def test_check_time(self):
actuals, reminders = self.interface.check_time(self.CHECK_TIME)
self.assertDictEqual(
actuals['continuing'],
{self.task_around_check_id: self.task_around_check})
self.assertDictEqual(
actuals['starting'],
{self.task_starts_at_check_id: self.task_starts_at_check})
self.assertDictEqual(actuals['ending'], {})
def test_complete_task(self):
self.interface.complete_task(self.task_starts_at_check_id)
status = self.interface.get_task(
self.task_starts_at_check_id).get_attribute(TaskAttributes.STATUS)
self.assertTrue(status == TaskStatus.COMPLETE)
