8_inK SSKJr SSKrSSKJrJrJr SSKJrJrJ r J r J r SSK r SSK r SSKJr SSKJr "SS 5rS"S \\\ 4S \ \S \4S jjrS\S \ \\\ 44SjrS\S \ \\\ 44SjrS \ \\\ 44SjrS#S\S\S \\\ 44SjjrS\S \4SjrS\S \\\ 4S \4SjrS \S \\\\ 44SjrSrSrS$Sjr S$Sjr!Sr"Sr#S"Sjr$S%S jr%S!r&g)&datetimeN)dbChallengeRecordUser)DictListOptionalAnyUnion) current_app) flag_modifiedc @\rSrSrSr\SS\\\4S\ \ S\ 4Sjj5r \S\ S\ \\\44S j5r \S \S\ \\\44S j5r \S\ \\\44S j5r\SS \ S\ S\\\44Sjj5r\S\ S\4Sj5r\S\ S\\\4S\4Sj5rSrg)ChallengeManager u; 题目管理类：处理题目记录的增删改查操作 Ndatauser_idreturncnSU;a+USRSS5nUSRSS5nO$URSS5nURSS5n[RRUS9R 5nU(a UnX%lO[5nX%lX5lURS5UlURS5UlURS S 5Ul URS S 5Ul URS 5Ul SU;a USUl U(aXl U(d[RR!U5 [RR#5 UR$$![&a%n[RR)5 UeSnAff=f)u 保存题目生成记录到数据库 Args: data: 包含题目详细信息的字典 user_id: 用户ID，如果提供则关联到此用户 Returns: int: 新创建或更新的记录ID challenge_infonameu未命名题目 descriptionr output_dirlog_fileestimated_timeu 1-2小时generation_status completedgeneration_warning deployableN)getrquery filter_byfirstrrrrrrr!r"rrsessionaddcommitid Exceptionrollback)rrrrexisting_challenge challengees =/Users/yu22x/Desktop/ge/ctf/app/models/database/operations.pysave_challenge_record&ChallengeManager.save_challenge_records| t #()--f6GHD/044]BGK88F$56D((="5K-22<<$<GMMO *I!N()I!N!,#xx 5 !XXj1 #'88,DDF   ""c[RR[RR 55R 5nU(dgUR 5$)u 获取最新生成的题目记录 Returns: Optional[Dict]: 最新题目详细信息字典，如果不存在则返回None N)rr$order_by created_atdescr&r9)r.s r0get_latest_challenge%ChallengeManager.get_latest_challengexsG$))22?3M3M3R3R3TU[[]   ""rFpageper_pagec X[RR[RR 55R XSS9n/nUR HnURURURURUR(aURROSUR(aURR5OSS.5 M URURURUR UR"UR$US.$)u 获取题目列表（分页） Args: page: 页码 per_page: 每页记录数 Returns: Dict: 包含分页信息和题目列表的字典 F)rMrN error_outN)r*rr difficultyrI)totalpages current_pagerNhas_nexthas_prev challenges)rr$rHrIrJpaginateitemsappendr*rrdifficulty_level isoformatrRrSrMrNrUrV)rMrN paginationrWr.s r0list_challenges ChallengeManager.list_challengess%**33O4N4N4S4S4UV__E`  #))I   ll!(44AJA[A[i88==aeBKBVBVi22<<>\`  * %%%%&OO"++"++"++$  rFc[RRU5nU(dgSSKnSSKJn UR (aURRUR 5(aTURUR 5nU(a[SUR 35 O1[SUR 35 O[SUR 35 UR(a_URRUR5(a5UR"UR5 [SUR35 [R R#U5 [R R%5 [S US UR&35 g ![a!n[S [U535 SnANSnAff=f![a%n[R R)5 UeSnAff=f) u 删除题目记录（同时删除相关文件） Args: challenge_id: 题目记录ID Returns: bool: 是否成功删除 FrN)challenge_file_manageru✅ 已删除输出目录: u!⚠️ 删除输出目录失败: u⚠️ 输出目录不存在: u✅ 已删除日志文件: u!⚠️ 删除日志文件失败: u✅ 已删除题目记录: ID=z, Name=T)rr$r#osapp.services.storagerarpathexistsdelete_directoryr;rremover+r<rr'deleter)rr,)r3r.rbrasuccessr/s r0delete_challenge!ChallengeManager.delete_challenges$))--l;    C##77>>)"6"6774EEiFZFZ[G ;I": # #! c! ! d38n! ! F,s,t,,\ s $sCx. T  rFrrrrc,[RX5$)u<保存题目生成记录到数据库（向后兼容函数）)rr1)rrs r0r1r1s  1 1$ @@rFr3c,[RU5$)u2根据ID获取题目记录（向后兼容函数）)rrAr3s r0rArAs  0 0 >>rFrc,[RU5$)u6根据名称获取题目记录（向后兼容函数）)rrDrs r0rDrDs  1 1$ 77rFc*[R5$)u9获取最新生成的题目记录（向后兼容函数）)rrKr}rFr0rKrK s  0 0 22rFrMrNc,[RX5$)u*获取题目列表（向后兼容函数）)rr^)rMrNs r0r^r^s  + +D ;;rFc,[RU5$)u*删除题目记录（向后兼容函数）)rrjrs r0rjrjs  , ,\ ::rFc,[RX5$)u*更新题目记录（向后兼容函数）)rr{)r3rs r0r{r{s  , ,\ @@rFcZ[RRUS9R[RR 55R 5nUVs/sHo"R5PM sn$s snf![a#n[S[U535 /sSnA$SnAff=f)um 获取指定用户的所有题目记录 Args: user_id: 用户ID Returns: List[Dict]: 题目列表 ru 获取用户题目列表失败: N) rr$r%rHrIrJallr9r+r;r<)rrWr.r/s r0get_challenges_by_userrs$**44W4ENNOiOiOnOnOpquuw 5?@Z !!#Z@@@  0Q9: s0AA=A85A=8A== B*B%B*%B*cSSKJnJn SU;aSU;aUSUS'URU5nURR U5 URR 5 U$![aOnWRR5 [S[U535 SSK nUR5 SnAgSnAff=f)u创建部署记录 Args: deployment_data: 部署数据字典 Returns: DeploymentRecord: 创建的部署记录对象 rDeploymentRecordrdeployment_uuidr*u创建部署记录失败: N) app.models.database.modelsrr from_dictr'r(r)r+r,r;r<r=r>)deployment_datarrrecordr/r=s r0create_deployment_recordr-sC O 38O1@1FO- ."++O< v    *3q6(34 sA#A&& B?0AB::B?cSSKJn URRUS9R 5nU(aUR 5$g![ a!n[S[U535 SnAgSnAff=f)uo通过UUID获取部署记录 Args: deployment_uuid: 部署UUID Returns: Dict: 部署记录字典 rrrNu获取部署记录失败: ) rrr$r%r&r9r+r;r<)rrrr/s r0get_deployment_by_uuidrLsf ?!''11/1RXXZ >># #  *3q6(34sAA A2A--A2cSSKJn URRUS9R UR R 55RU5RU5R5nUVs/sHoUR5PM sn$s snf![a#n[S[U535 /sSnA$SnAff=f)u获取用户的部署记录列表 Args: user_id: 用户ID limit: 返回结果数量限制 offset: 结果偏移量 Returns: List[Dict]: 部署记录字典列表 rrru 获取用户部署记录失败: Nrrr$r%rHrIrJoffsetlimitrr9r+r;r<)rrrrrecordsrr/s r0get_deployments_by_userr`s ?"((2272C X&11668 9 VF^EE%L 077wV w777  0Q9: 0A6B8BBB C#C;CCcSSKJn URRUS9R UR R 55RU5RU5R5nUVs/sHoUR5PM sn$s snf![a#n[S[U535 /sSnA$SnAff=f)u获取题目的部署记录列表 Args: challenge_id: 题目ID limit: 返回结果数量限制 offset: 结果偏移量 Returns: List[Dict]: 部署记录字典列表 rrru 获取题目部署记录失败: Nr)r3rrrrrr/s r0get_deployments_by_challengerws ?"((22 2M X&11668 9 VF^EE%L 077wV w777  0Q9: rc 8SSKJnJn URR US9R 5nU(dgUR nXlURSS5nUR5H#up[XX5(dM[XXU 5 M% Xa:wa U(dSUSU3nURX5 URR5 g ![a;n WRR5 [!S [#U 535 Sn A gSn A ff=f) u更新部署状态 Args: deployment_uuid: 部署UUID status: 新状态 **kwargs: 其他要更新的字段（包括message用于日志） Returns: bool: 更新是否成功 rrrFmessageNu 状态从 u 变更为 Tu更新部署状态失败: )rrrr$r%r&statuspoprYrvrwadd_status_logr'r)r+r,r;r<) rrkwargsrrr old_status log_messagerxryr/s r0update_deployment_statusrs"C"''11/1RXXZ]]  jjD1 !,,.JCv##U+)   *:,k&J  ! !& 6    *3q6(34s$6CACAC D1DDcnSSKJnJn URR US9R 5nU(dgUR RU5 UR R5 g![a;nWR R5 [S[U535 SnAgSnAff=f)ue删除部署记录 Args: deployment_uuid: 部署UUID Returns: bool: 删除是否成功 rrrFTu删除部署记录失败: N) rrrr$r%r&r'rhr)r+r,r;r<)rrrrr/s r0delete_deployment_recordrsC"''11/1RXXZ &!    *3q6(34s6A/5A// B491B//B4cSSKJnJn URR US9R 5nU(dgUR X5 URR5 UR5nU(aUS$S$![a;nWRR5 [S[U535 SnAgSnAff=f)u添加部署状态日志 Args: deployment_uuid: 部署UUID status: 状态 message: 消息 Returns: DeploymentStatusLog: 创建的日志对象，如果失败则返回None rrrNu添加部署日志失败: )rrrr$r%r&rr'r)get_status_historyr+r,r;r<)rrrrrrhistoryr/s r0add_deployment_logrsC"''11/1RXXZ f. ++-%wr{/4/   *3q6(34s#6BABB C 1CCc  SSKJn SSKJn URR US9R 5nU(d/$UR 5n/n[U5HupxURS5n [U [5(a$UR"U RSS55n O![X5(dUR"5n URUS-URURS S 5URS S 5U S .5 M UR!S SS9 USU$! UR"5n Ns=f!["a7n [%S[U 535 SSKn U R)5 /sSn A $Sn A ff=f)u获取部署日志 Args: deployment_uuid: 部署UUID limit: 返回结果数量限制 Returns: List[Dict]: 日志字典列表 rrrr timestampZz+00:00rrrr)r* deployment_idrrrc US$)Nrr})xs r0%get_deployment_logs.."s+rFT)rxreverseNu获取部署日志失败: )rrrr$r%r&r enumerater# isinstancer< fromisoformatreplacenowrZr*sortr+r;r=r>) rrrrrrlogsi log_entryrr/r=s r0get_deployment_logsrs^(?%"''11/1RXXZI++-%g.LA! k2I)S))/ ( 6 6y7H7Hh7W XI 44$LLN KK!e!'#--"5$==B7&  /( . =FU|/ ( I  *3q6(34 s<;EAE"D))A?E)D><E F ,E=7F=FcSSKJnJn SSKJnJn UR "UR 5nURRURU:URS:g5R5nSnUHnSUl US- nM URR5 U$![a;nWRR5 [!S[#U535 SnAgSnAff=f)uE清理过期的部署记录 Returns: int: 清理的记录数量 rr)rtimezonedeletedru 清理过期部署记录失败: N)rrrrrrutcr$filter expires_atrrr'r)r+r,r;r<) rrrrrexpiredcountrr/s r0cleanup_expired_deploymentsr*s C/ll8<<("((//  ' '# -  # #y 0  #%  F%FM QJE    0Q9:sB"B%% C*/1C%%C*r~r)rr))'rjsonmodelsrrrtypingr r r r r rbglobflaskrsqlalchemy.orm.attributesrrr<rr1rArDrKr^rrjr{rrrrrrrrrrr}rFr0rs^ --33 3oohAS#XA#ARUA?s?xS#X/G?88c3h(@83htCH~63<#(..-^4>2hrF