from flask import Blueprint, request, render_template
from flask_apscheduler.utils import job_to_dict

from applications.common.tasks import tasks
from applications.common.tasks.tasks import task_list
from applications.common.utils.http import table_api, fail_api, success_api
from applications.extensions.init_apscheduler import scheduler

admin_task = Blueprint('adminTask', __name__, url_prefix='/admin/task')


@admin_task.route('/add_job', methods=['GET'])
def add_task():
    scheduler.add_job(func=tasks.get(), id='4', args=(1, 1), trigger='interval', seconds=3,
                      replace_existing=True)
    return '6'


@admin_task.get('/')
def main():
    return render_template('admin/task/main.html')


@admin_task.route('/data', methods=['GET'])
def get_task():  # 获取
    jobs = scheduler.get_jobs()
    jobs_list = []
    for job in jobs:
        jobs_list.append(job_to_dict(job))
    return table_api(data=jobs_list, count=len(jobs_list))


# 增加
@admin_task.get('/add')
def add():
    return render_template('admin/task/add.html', task_list=task_list)


@admin_task.post('/save')
def save():
    _id = request.json.get("id")
    name = request.json.get("id")
    type = request.json.get("type")
    functions = request.json.get("functions")
    datetime = request.json.get("datetime")
    time = request.json.get("time")
    if not hasattr(tasks, functions):
        return fail_api()
    if type == 'date':
        scheduler.add_job(
            func=getattr(tasks, functions),
            id=_id,
            name=name,
            args=(1, 1),
            trigger=type,
            run_date=datetime,
            replace_existing=True)
    elif type == 'interval':
        scheduler.add_job(
            func=getattr(tasks, functions),
            id=_id,
            name=name,
            args=(1, 1),
            trigger=type,
            replace_existing=True)
    elif type == 'cron':
        scheduler.add_job(
            func=getattr(tasks, functions),
            id=_id,
            name=name,
            args=(1, 1),
            trigger=type,
            replace_existing=True)

    return success_api()


# 恢复
@admin_task.put('/enable')
def enable():
    _id = request.json.get('id')
    # print(id)
    if _id:
        scheduler.resume_job(str(_id))
        return success_api(msg="启动成功")
    return fail_api(msg="数据错误")


# 暂停
@admin_task.put('/disable')
def dis_enable():
    _id = request.json.get('id')
    if _id:
        scheduler.pause_job(str(_id))
        return success_api(msg="暂停成功")
    return fail_api(msg="数据错误")


@admin_task.delete('/remove/<int:_id>')
def remove_job(_id):  # 移除
    scheduler.remove_job(str(_id))
    return success_api(msg="删除成功")

    #     scheduler.add_job(func=task1, id='2', args=(1, 1), trigger='cron', day_of_week='0-6', hour=18, minute=24,
    #                       second=10, replace_existing=True)

    #     scheduler.add_job(func=task4, id='4', args=(2, 2), trigger='interval', seconds=3,
    #                       replace_existing=True, misfire_grace_time=3)

    #     # trigger='interval' 表示是一个循环任务，每隔多久执行一次
    #     scheduler.add_job(func=task2, id='3', args=(2, 2), trigger='interval', seconds=3,
    #                       replace_existing=True)