from db.db_conn import db_conn
from db.repository.config_rep import ConfigRepository
from task.request_task import retry
from api.base_request import my_Request
from api.site_data_api import fetch_dust_data
from enumeration.enum_site_status import EnumSiteStatus
from datetime import datetime, timedelta
import time
from utils.log_utils import LogUtils
from scheduler.task_scheduler import TaskScheduler
from config.config import Config
from controller.main_controller import MainController
import json
from db.models.tables import SiteLatestTime, RequestTask
from utils.date_utils import DateUtils


@retry
def task_test() -> bool:
    return False


def task():
    LogUtils.info("done")


def sub_scheduler():
    sc = TaskScheduler()
    now = datetime.now()
    LogUtils.info('now')
    start_time = now + timedelta(seconds=2)
    end_time = now + timedelta(seconds=12)
    sc.add_task(task, start_time, 60*60)
    sc.start_task()
    # sc.shutdown_task()
    print('--------end')
    return sc


if __name__ == '__main__':
    MainController(True).execute()
    # a = db_conn._create_session()
    # b = db_conn._create_session()
    # print(id(a))
    # print(id(b))

    # config_rep1 = ConfigRepository()
    # res = config_rep1.read_request_config()
    # print(res[0].request_start_time)

    # task_test(4, 4, "测试")

    # session = my_Request.get_session()
    # session.headers['Cookie'] = "JSESSIONID=6E5ECD90B5BC62BD9496C1EE0DCF8B61; GOVSSOSSOTOKEN=2c9483d18ab14502018b36e3dfff7ab6"
    # res = fetch_dust_data(sTime="2023-10-16 12:00:00", eTime="2023-10-16 12:14:59", site="45879")
    # print(res)

    # print(EnumSiteStatus.Online)
    # print(EnumSiteStatus.Online.name)
    # print(EnumSiteStatus.Online.value)

    # c = Config().get_request_config()
    # print(c)
    # session = my_Request.get_session()
    # cookie = "{'GOVSSOSSOTOKEN': '2c9483d18ab14502018b40a6972d7662', 'JSESSIONID': 'A8F809955228FA268D0AEA211A11AAB1'}"
    # cookie = eval(cookie)
    # my_Request.set_cookie(cookie)
    # print(session.cookies)
    # c = dict(session.cookies)
    # s = ""
    # for k in c.keys():
    #     s += f"{k}={c[k]};"
    # s = s[:-1]
    # print(s)

    # l = db_conn.query_sql(lambda session: session.query(SiteLatestTime).all())
    # obj = l[0]
    # print(obj)
    # obj.device_status = 2

    # str = "{'GOVSSOSSOTOKEN': '2c9483d18b4bc697018b6e9ee0d860e5', 'JSESSIONID': '885EC3E03A5766EDEB1AA3EFBA8211CA', 'JSESSIONID': '9C30220C12B31F7C074BE3A55AA02E5A'}"
    # cookie = eval(str)
    # print(cookie)

    # def f(session):
    #         task = session.query(RequestTask).filter_by(id=16).one()
    #         # task.end_time = datetime.now()
    #         # task.running_status = False
    #         # task.success = s
    #         # session.commit()
    #         return task
    # r = db_conn.query_sql(f, new_session=True)
    # print(r)

    # timestamp = datetime(2023, 11, 15, 8, 30, 00)
    # v = DateUtils.time_distance(timestamp, type="hour")
    # print(v > 6)