-- coding: utf-8 --
“”"
@file: async_zt_flow_hcproblem
@author: maxiaolong01
@date: 2025/7/17 10:14
“”"
import datetime
import pprint
from tortoise import run_async
from app.models.product import RequirementPool
from app.settings import TORTOISE_ORM
from app.utils.db import MySqlClient
estimated_finish_date = datetime.date(1971, 1, 1)
async def get_zt_materiallib(sql: str = ‘select * from zt_materiallib’):
result = {}
async with MySqlClient(TORTOISE_ORM) as client:
for item in await client.execute_query(sql):
release_date = item.get(
“releaseDate”, datetime.date(
1971, 1, 1))
result["release_date"] = str(release_date) result["software_project"] = item.get("name", "未关联") print("release_date:", release_date, datetime.date( 1971, 1, 1)) if release_date == datetime.date( 1971, 1, 1): result["requirement_status"] = "超期未关闭" if datetime.date.today( ) > estimated_finish_date else "按期未关闭" print("requirement_status:", datetime.date.today(), estimated_finish_date, result["requirement_status"]) else: result["requirement_status"] = "按期关闭" if release_date < estimated_finish_date else "超期关闭" print("requirement_status:", release_date, estimated_finish_date, result["requirement_status"]) return result
async def get_zt_build(sql: str = ‘select * from zt_build’):
result = {}
async with MySqlClient(TORTOISE_ORM) as client:
for item in await client.execute_query(sql):
associated_version = item.get(“name”, “未关联”)
result[“associated_version”] = associated_version
result[“associated_version_id”] = item.get(“id”, 0)
sql = f""" SELECT m.releaseDate,p.name FROM zt_materiallib m LEFT JOIN zt_project p ON p.id = m.project WHERE m.topubbaseline = '{associated_version}' AND m.status = '1' LIMIT 1 """ d = await get_zt_materiallib(sql) if not d: d["release_date"] = str(datetime.date( 1971, 1, 1)) d["software_project"] = "未关联" d["requirement_status"] = "超期未关闭" if datetime.date.today( ) > estimated_finish_date else "按期未关闭" print("requirement_status:", datetime.date.today(), estimated_finish_date, d["requirement_status"]) result.update(d) return result
async def get_zt_hctransform(sql: str = ‘select * from zt_hctransform where fromid = 1 and type = “hcrequirementtostory”’):
result = []
async with MySqlClient(TORTOISE_ORM) as client:
transformid = None
for item in await client.execute_query(sql):
transformid = item.get(“transformid”)
sql = f"“”
SELECT * FROM zt_build
WHERE stories LIKE ‘%{transformid}%’ AND deleted = ‘0’
ORDER BY id DESC
LIMIT 1
“”"
build_data = await get_zt_build(sql)
if not build_data:
build_data.update({
“rd_req_id”: 0,
“release_date”: str(datetime.date(
1971, 1, 1)),
“software_project”: “未关联”,
“requirement_status”:“超期未关闭” if datetime.date.today(
) > estimated_finish_date else “按期未关闭”
})
else:
build_data.update({“rd_req_id”: transformid})
result.append(build_data) return result
async def get_zt_flow_hcrequirement_item(sqlfile: str = ‘./app/tasks/sql/zt_flow_hcrequirement.sql’):
result = []
global estimated_finish_date
with open(sqlfile, ‘r’, encoding=‘utf-8’) as f:
sql = f.read()
async with MySqlClient(TORTOISE_ORM) as client:
for item in await client.execute_query(sql):
fromid = item.get(“zt_id”)
sql = f’select * from zt_hctransform where fromid = {fromid} and type = “hcrequirementtostory”’
print(sql)
estimated_finish_date = datetime.date.fromisoformat(item.get(
“estimated_finish_date”, “1971-01-01”))
rd_req_data_list = await get_zt_hctransform(sql)
data = {"" "zt_id": item.get("zt_id"), "project_no": item.get("project_no"), "equip": item.get("equip"), "req_name": item.get("req_name"), "req_sub_date": datetime.datetime.strptime(item.get("req_sub_date", "1971-01-01"), '%Y-%m-%d').strftime('%Y-%m-%d %H:%M:%S'), "estimated_finish_date": datetime.datetime.strptime(item.get("estimated_finish_date", "1971-01-01"), '%Y-%m-%d').strftime('%Y-%m-%d %H:%M:%S'), "req_dev_leader": item.get("req_dev_leader") } # pprint.pprint(item) # pool, _ = await RequirementPool.update_or_create(**data) rd_red_data = [] for i in rd_req_data_list: d = { "associated_version": i.get("associated_version", "未关联"), "associated_version_id": i.get("associated_version_id", 0), "rd_req_id": i.get("rd_req_id", None), "release_date": i.get("release_date", datetime.datetime.strptime("1971-01-01", '%Y-%m-%d').strftime('%Y-%m-%d %H:%M:%S')), "requirement_status": i.get("requirement_status", "未关闭"), "software_project": i.get("software_project", "未关联"), } data.update(d) print(data) rd_red_data.append(RequirementPool(**data)) if rd_red_data: print(rd_red_data) await RequirementPool.bulk_create(rd_red_data) # await RDReqData.bulk_create(rd_red_data)
async def insert_product_requirement_pool():
await get_zt_flow_hcrequirement_item()
if name == ‘main’:
run_async(get_zt_flow_hcrequirement_item())
再造成分批并发同步 展示优化后的全部代码
最新发布
本文介绍Python中datetime类型转换的方法，重点讲解如何使用strptime函数将字符串转换为datetime对象，并列出常用的格式化指令。
扫一扫
1346
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
