def append_plans(self, report): """ Добавить плановое по к-во вопросов в отчет """ plans = PlanCtrl.get_on_month(self.year, self.month) for r in report: plan = find(plans, lambda p: p.employee.name == r['employee_name']) if plan is not None: r['qty_plan'] = plan.qty_question return report
def test_get_on_month_NOT_create(self): # Планы НЕ создаются plans = PlanCtrl.get_on_month(2017, 12) self.assertEqual(len(plans), 1)
def test_get_on_month_create(self): # Планы создаются, если не созданы plans = PlanCtrl.get_on_month(3000, 12) self.assertEqual(len(plans), 2)
def get(self): """ Получение планов на год/месяц """ plans = PlanCtrl.get_on_month(self.context.year, self.context.month) return self.serializator.dump(plans, many=True).data