from odoo import http
from odoo.http import request
import datetime as DT


class HelpDeskDashboard(http.Controller):

    @http.route(['/helpdesk_dashboard'], type='json', auth="public")
    def helpdesk_dashboard(self):
        stage_new = request.env['ticket.stage'].search(
            [('name', '=', 'Inbox')], limit=1).id
        stage_draft = request.env['ticket.stage'].search(
            [('name', '=', 'Draft')], limit=1).id
        stage_inprogress = request.env['ticket.stage'].search(
            [('name', '=', 'In Progress')], limit=1).id
        stage_canceled = request.env['ticket.stage'].search(
            [('name', '=', 'Canceled')], limit=1).id
        stage_done = request.env['ticket.stage'].search(
            [('name', '=', 'Done')], limit=1).id
        stage_closed = request.env['ticket.stage'].search(
            [('name', '=', 'Closed')], limit=1).id
        stage_ids = [stage_new,stage_draft]
        new = request.env["help.ticket"].search_count([('stage_id', 'in', stage_ids)])
        new_id = request.env["help.ticket"].search([('stage_id', 'in', stage_ids)])
        new_id_ls = []
        for data in new_id:
            new_id_ls.append(data.id)

        in_progress = request.env["help.ticket"].search_count([('stage_id', '=', stage_inprogress)])
        in_progress_id = request.env["help.ticket"].search([('stage_id', '=', stage_inprogress)])
        in_progress_ls = []
        for data in in_progress_id:
            in_progress_ls.append(data.id)
        canceled = request.env["help.ticket"].search_count([('stage_id', '=', stage_canceled)])
        canceled_id = request.env["help.ticket"].search([('stage_id', '=', stage_canceled)])
        canceled_id_ls = []
        for data in canceled_id:
            canceled_id_ls.append(data.id)
        done = request.env["help.ticket"].search_count([('stage_id', '=', stage_done)])
        done_id = request.env["help.ticket"].search([('stage_id', '=', stage_done)])
        done_id_ls = []
        for data in done_id:
            done_id_ls.append(data.id)
        closed = request.env["help.ticket"].search_count([('stage_id', '=', stage_closed)])
        closed_id = request.env["help.ticket"].search([('stage_id', '=', stage_closed)])
        closed_id_ls = []
        for data in closed_id:
            closed_id_ls.append(data.id)

        dashboard_values = {
            'new':new,
            'in_progress':in_progress,
            'canceled':canceled,
            'done':done,
            'closed':closed,
            'new_id':new_id_ls,
            'in_progress_id':in_progress_ls,
            'canceled_id': canceled_id_ls,
            'done_id': done_id_ls,
            'closed_id': closed_id_ls,
        }
        return dashboard_values

    @http.route(['/helpdesk_dashboard_week'], type='json', auth="public")
    def helpdesk_dashboard_week(self):
        today = DT.date.today()
        stage_new = request.env['ticket.stage'].search(
            [('name', '=', 'Inbox')], limit=1).id
        stage_draft = request.env['ticket.stage'].search(
            [('name', '=', 'Draft')], limit=1).id
        stage_inprogress = request.env['ticket.stage'].search(
            [('name', '=', 'In Progress')], limit=1).id
        stage_canceled = request.env['ticket.stage'].search(
            [('name', '=', 'Canceled')], limit=1).id
        stage_done = request.env['ticket.stage'].search(
            [('name', '=', 'Done')], limit=1).id
        stage_closed = request.env['ticket.stage'].search(
            [('name', '=', 'Closed')], limit=1).id
        stage_ids = [stage_new, stage_draft]
        week_ago = str(today - DT.timedelta(days=7)) + ' '
        new = request.env["help.ticket"].search_count(
            [('stage_id', 'in', stage_ids), ('create_date', '>', week_ago)])
        new_id = request.env["help.ticket"].search(
            [('stage_id', 'in', stage_ids), ('create_date', '>', week_ago)])
        new_id_ls = []
        for data in new_id:
            new_id_ls.append(data.id)
        in_progress = request.env["help.ticket"].search_count(
            [('stage_id', '=', stage_inprogress), ('create_date', '>', week_ago)])
        in_progress_id = request.env["help.ticket"].search(
            [('stage_id', '=', stage_inprogress), ('create_date', '>', week_ago)])
        in_progress_ls = []
        for data in in_progress_id:
            in_progress_ls.append(data.id)
        canceled = request.env["help.ticket"].search_count(
            [('stage_id', '=', stage_canceled), ('create_date', '>', week_ago)])
        canceled_id = request.env["help.ticket"].search(
            [('stage_id', '=', stage_canceled), ('create_date', '>', week_ago)])
        canceled_id_ls = []
        for data in canceled_id:
            canceled_id_ls.append(data.id)
        done = request.env["help.ticket"].search_count(
            [('stage_id', '=', stage_done), ('create_date', '>', week_ago)])
        done_id = request.env["help.ticket"].search(
            [('stage_id', '=', stage_done), ('create_date', '>', week_ago)])
        done_id_ls = []
        for data in done_id:
            done_id_ls.append(data.id)
        closed = request.env["help.ticket"].search_count(
            [('stage_id', '=', stage_closed), ('create_date', '>', week_ago)])
        closed_id = request.env["help.ticket"].search(
            [('stage_id', '=', stage_closed), ('create_date', '>', week_ago)])
        closed_id_ls = []
        for data in closed_id:
            closed_id_ls.append(data.id)

        dashboard_values = {
            'new': new,
            'in_progress': in_progress,
            'canceled': canceled,
            'done': done,
            'closed': closed,
            'new_id': new_id_ls,
            'in_progress_id': in_progress_ls,
            'canceled_id': canceled_id_ls,
            'done_id': done_id_ls,
            'closed_id': closed_id_ls,
        }
        return dashboard_values

    @http.route(['/helpdesk_dashboard_month'], type='json', auth="public")
    def helpdesk_dashboard_month(self):
        today = DT.date.today()
        stage_new = request.env['ticket.stage'].search(
            [('name', '=', 'Inbox')], limit=1).id
        stage_draft = request.env['ticket.stage'].search(
            [('name', '=', 'Draft')], limit=1).id
        stage_inprogress = request.env['ticket.stage'].search(
            [('name', '=', 'In Progress')], limit=1).id
        stage_canceled = request.env['ticket.stage'].search(
            [('name', '=', 'Canceled')], limit=1).id
        stage_done = request.env['ticket.stage'].search(
            [('name', '=', 'Done')], limit=1).id
        stage_closed = request.env['ticket.stage'].search(
            [('name', '=', 'Closed')], limit=1).id
        stage_ids = [stage_new, stage_draft]
        week_ago = str(today - DT.timedelta(days=30)) + ' '
        new = request.env["help.ticket"].search_count(
            [('stage_id', 'in', stage_ids), ('create_date', '>', week_ago)])
        new_id = request.env["help.ticket"].search(
            [('stage_id', 'in', stage_ids), ('create_date', '>', week_ago)])
        new_id_ls = []
        for data in new_id:
            new_id_ls.append(data.id)
        in_progress = request.env["help.ticket"].search_count(
            [('stage_id', '=', stage_inprogress),
             ('create_date', '>', week_ago)])
        in_progress_id = request.env["help.ticket"].search(
            [('stage_id', '=', stage_inprogress),
             ('create_date', '>', week_ago)])
        in_progress_ls = []
        for data in in_progress_id:
            in_progress_ls.append(data.id)
        canceled = request.env["help.ticket"].search_count(
            [('stage_id', '=', stage_canceled),
             ('create_date', '>', week_ago)])
        canceled_id = request.env["help.ticket"].search(
            [('stage_id', '=', stage_canceled),
             ('create_date', '>', week_ago)])
        canceled_id_ls = []
        for data in canceled_id:
            canceled_id_ls.append(data.id)
        done = request.env["help.ticket"].search_count(
            [('stage_id', '=', stage_done), ('create_date', '>', week_ago)])
        done_id = request.env["help.ticket"].search(
            [('stage_id', '=', stage_done), ('create_date', '>', week_ago)])
        done_id_ls = []
        for data in done_id:
            done_id_ls.append(data.id)
        closed = request.env["help.ticket"].search_count(
            [('stage_id', '=', stage_closed), ('create_date', '>', week_ago)])
        closed_id = request.env["help.ticket"].search(
            [('stage_id', '=', stage_closed), ('create_date', '>', week_ago)])
        closed_id_ls = []
        for data in closed_id:
            closed_id_ls.append(data.id)

        dashboard_values = {
            'new': new,
            'in_progress': in_progress,
            'canceled': canceled,
            'done': done,
            'closed': closed,
            'new_id': new_id_ls,
            'in_progress_id': in_progress_ls,
            'canceled_id': canceled_id_ls,
            'done_id': done_id_ls,
            'closed_id': closed_id_ls,
        }
        return dashboard_values

    @http.route(['/helpdesk_dashboard_year'], type='json', auth="public")
    def helpdesk_dashboard_year(self):
        today = DT.date.today()
        stage_new = request.env['ticket.stage'].search(
            [('name', '=', 'Inbox')], limit=1).id
        stage_draft = request.env['ticket.stage'].search(
            [('name', '=', 'Draft')], limit=1).id
        stage_inprogress = request.env['ticket.stage'].search(
            [('name', '=', 'In Progress')], limit=1).id
        stage_canceled = request.env['ticket.stage'].search(
            [('name', '=', 'Canceled')], limit=1).id
        stage_done = request.env['ticket.stage'].search(
            [('name', '=', 'Done')], limit=1).id
        stage_closed = request.env['ticket.stage'].search(
            [('name', '=', 'Closed')], limit=1).id
        stage_ids = [stage_new, stage_draft]
        week_ago = str(today - DT.timedelta(days=360)) + ' '
        new = request.env["help.ticket"].search_count(
            [('stage_id', 'in', stage_ids), ('create_date', '>', week_ago)])
        new_id = request.env["help.ticket"].search(
            [('stage_id', 'in', stage_ids), ('create_date', '>', week_ago)])
        new_id_ls = []
        for data in new_id:
            new_id_ls.append(data.id)
        in_progress = request.env["help.ticket"].search_count(
            [('stage_id', '=', stage_inprogress),
             ('create_date', '>', week_ago)])
        in_progress_id = request.env["help.ticket"].search(
            [('stage_id', '=', stage_inprogress),
             ('create_date', '>', week_ago)])
        in_progress_ls = []
        for data in in_progress_id:
            in_progress_ls.append(data.id)
        canceled = request.env["help.ticket"].search_count(
            [('stage_id', '=', stage_canceled),
             ('create_date', '>', week_ago)])
        canceled_id = request.env["help.ticket"].search(
            [('stage_id', '=', stage_canceled),
             ('create_date', '>', week_ago)])
        canceled_id_ls = []
        for data in canceled_id:
            canceled_id_ls.append(data.id)
        done = request.env["help.ticket"].search_count(
            [('stage_id', '=', stage_done), ('create_date', '>', week_ago)])
        done_id = request.env["help.ticket"].search(
            [('stage_id', '=', stage_done), ('create_date', '>', week_ago)])
        done_id_ls = []
        for data in done_id:
            done_id_ls.append(data.id)
        closed = request.env["help.ticket"].search_count(
            [('stage_id', '=', stage_closed), ('create_date', '>', week_ago)])
        closed_id = request.env["help.ticket"].search(
            [('stage_id', '=', stage_closed), ('create_date', '>', week_ago)])
        closed_id_ls = []
        for data in closed_id:
            closed_id_ls.append(data.id)

        dashboard_values = {
            'new': new,
            'in_progress': in_progress,
            'canceled': canceled,
            'done': done,
            'closed': closed,
            'new_id': new_id_ls,
            'in_progress_id': in_progress_ls,
            'canceled_id': canceled_id_ls,
            'done_id': done_id_ls,
            'closed_id': closed_id_ls,
        }
        return dashboard_values