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)
|
