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

from applications.common.tasks.tasks import task2, task3, task4
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')


# 暂停任务
# scheduler.pause_job('third')
# 恢复任务
# time.sleep(10)
# scheduler.resume_job('third')
# 删除任务
# scheduler.remove_job('first')


@admin_task.route('/add_job', methods=['GET'])
def add_task():
    tasks = {
        "task2": task2,
        "task3":task3,
        "task4":task4
    }
    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)


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


# 恢复
@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)