o cj+2 @sUdZddlmZddlZddlmZmZddlmZmZddl m Z m Z ddl m Z mZddlmZdd lmZdd lmZmZed gdfZiad ed <dZddddZdGddZdHddZdIdd ZdId!d"ZdId#d$ZdId%d&Z d'dd(ed)d*d+dd(ed,d*d-dd(ed.d*d/d0d(e d1d*d2Z!d ed3<dJd5d6Z"dKd7d8Z#dLd:d;Z$dMd=d>Z%dNd@dAZ&eGdBdCdCe Z'eGdDdEdEe Z(eGdFd d e Z)dS)Oua 积分商店：数据库 shop_stock 为唯一货架；内存 SHOP_ITEMS 由库表同步。 每周一 8:00 清空后：随机上架 4 个称号（库存各 2，售价按稀有度 common=3 / rare=6 / legendary=10）， 并合并固定功能商品（FIXED_FUNCTION_ITEMS）。 功能向效果存于 shop_user_buffs 表，由打卡/抽卡插件消费。 ) annotationsN)datetime timedelta)AnyCallable)PluginTimedHeartbeatPlugin)attext)logger)register_plugin) TITLE_DEFS get_title_defRedeemShopPluginzdict[str, dict[str, Any]] SHOP_ITEMS commonrare legendarytdefdict[str, Any] | NonereturnintcCs<|pidpd}t|tr|}tt|tdS)Nrarityr)get isinstancestrstriplowerrTITLE_PRICE_BY_RARITY)rrr%0/home/dore/onebot/Bot_Ero/plugins/redeem_shop.pytitle_price_from_def#s  r'plugin'RedeemShopPlugin'title_idNonecCsL|jj}|dur td|j||rtd|jj||dds$tddS)N无法识别用户u你已拥有该称号Fcommitu称号发放失败) bot_eventuser_id RuntimeError dbmanager has_title unlock_title)r(r*uidr%r%r& _grant_title*sr6cCsF|jj}|dur tdttddd}|jj||dddS)Nr,r)days%Y-%m-%dFr-) r/r0r1rnowrstrftimer2set_extra_draw_pack_until)r(r5untilr%r%r&_grant_extra_draw_pack4s r=cC.|jj}|dur td|jj|ddddSNr,rFr-)r/r0r1r2add_shop_checkin_luckr(r5r%r%r&_grant_checkin_boost=rBcCr>r?)r/r0r1r2add_shop_lottery_waiverrAr%r%r&_grant_lottery_waiverDrCrEcCs<|jj}|dur tdtd}|jj||dddS)Nr,r8Fr-)r/r0r1rr9r:r2!clear_lottery_draw_count_for_date)r(r5todayr%r%r&_grant_lottery_refreshKs rHuB额外抽卡补充包（7 天内每日 +2 次额外抽卡额度）uT兑换成功，已获得额外抽卡补充（7 日有效）。剩余积分 {rest}。) descriptioncoststockapply success_tipuF打卡增强（接下来 10 次打卡，每次 10% 概率 +1 积分）uC兑换成功，打卡增强次数已入账。剩余积分 {rest}。uK抽奖增强（接下来 10 次付费抽奖，每次 30% 不消耗积分）uC兑换成功，抽奖增强次数已入账。剩余积分 {rest}。u6抽奖刷新（立刻清空今日已用抽卡次数）uC兑换成功，今日抽卡次数已重置。剩余积分 {rest}。)fn_extra_draw_packfn_checkin_boostfn_lottery_boostfn_lottery_refreshFIXED_FUNCTION_ITEMSdict[str, int]cCs*i}tD] \}}t|d||<q|S)NrL)rTitemsr)outpidmetar%r%r&_fixed_shop_stock_mappingssrZc Cs0|}t|D]\}}t|}|drtz t|ddd}Wn ttfy.Yq wt |p4i}| dd}| dpAd}t |trM| }dd d d  |t|}t|} d |d |d| t||fdddt|<q |tvrt|} t| dt| dt|| d| ddt|<q dS)uY根据 shop_stock 重建内存中的 SHOP_ITEMS（进程重启后与数据库一致）。title__rOname?rru普通u稀有u传奇ru解锁称号「u」（u）cSs t||SN)r6)pttr%r%r&s z2refresh_shop_items_from_database..)rJrK initial_stockrMrJrKrMrN)rJrKrcrMrNN)get_all_shop_stockrclearr startswithrsplit ValueError IndexErrorrrrr!r"r'rT) dbrowsrX_stocktidrnmrrcrKrYr%r%r& refresh_shop_items_from_databasezsB           rp list[int]cCshtt}tt}g}|r)tdt|}tj||d}|D] }t |d|<q| |t ||S)uK清空商店：随机 4 个称号 + 固定功能商品，并同步内存。)kr[) listr keysdictrZminlenrandomsampleWEEKLY_TITLE_STOCKreplace_entire_shop_shelfrp)rjall_idsmappingpickedrsrmr%r%r&weekly_refresh_shop_shelfs   r list[str]cCst|g}ttD]@}t|}|d}|d}||}|dur&d}n |dkr-d}nt|}|d|d||d |d ||d q |S) uB返回本周货架每条商品的文案行（含空行分隔）。rKrJNu 未上架rIu不限[z] u 售价 u 积分｜剩余 )rpsortedrruget_shop_stockr append)rjlinesrXrYrKdescstock_nrLr%r%r&format_shop_shelf_liness   rr cCs8t|}gd}|r||n|dd|S)u-每周刷新后发到群里的公告文案。)u积分商店已刷新~ud发送 /商店 查看完整列表，或 /商店 <商品id> 兑换。（方括号里的是商品id）ru—— 本周货架 ——u本周暂无上架商品。 )rextendrjoinrstrip)rjshelfrr%r%r&format_shop_weekly_announcements   rc@s.eZdZdZdZdZdgZddZddZd S) ShopWeeklyRotationPluginshop_weekly_rotationuB每周一 8:00 刷新积分商店并在群内公告本周货架。z08:00rOcCs ||Sr_)should_run_on_heartbeatself event_typer%r%r&matchs zShopWeeklyRotationPlugin.matchc Cslzt|j}tdt|||jtt|jWdSt y5}z t d|WYd}~dSd}~ww)Nu9积分商店已刷新（本周随机称号 %s 个）：%su积分商店周刷新失败: %s) rr2r inforxapisend_msgr r Exception exception)rrer%r%r&handles zShopWeeklyRotationPlugin.handleN) __name__ __module__ __qualname__r]rJRUN_AT RUN_WEEKDAYSrrr%r%r%r&rs rc@s$eZdZdZdZddZddZdS)ShopManualRefreshPluginshop_manual_refreshu3管理员指令：立刻刷新积分商店货架。cCs|dko |o |dS)Nmessageu /刷新商店) admin_user on_full_matchrr%r%r&rs zShopManualRefreshPlugin.matchc Cs|jjdurdS|jj}z!t|j}td||jt|t dt |d|WdSt yT}zt d||jt|t d|WYd}~dSd}~ww)Nu;管理员手动刷新积分商店（随机称号 id）：%su1商店已刷新。本周随机上架称号（共 u 个）：u管理员刷新商店失败: %su刷新失败：) r/r0rr2r rrrr r rxrr)rr5rrr%r%r&rs     *zShopManualRefreshPlugin.handleN)rrrr]rJrrr%r%r%r&rs  rc@s.eZdZdZdZddZd ddZd d Zd S) r redeem_shopu*使用积分兑换商店称号或权益。cCs|dko|dS)Nru/商店) on_commandrr%r%r&r szRedeemShopPlugin.matchrr cCsdgd}t|j}|s!|d|d|dd|S|||dd|S)N)u—— 积分商店 ——u用法：/商店 <商品id>ru6本周暂无上架商品（每周一 8:00 刷新）。ru#发送 /商店 <商品id> 兑换。r)rr2rrrr)rrrr%r%r& _format_lists      zRedeemShopPlugin._format_listc sjjdurdSjj}ddtdgpgD}tjt|dkr1jt gdS|d }|t vrLj t |td|ddSt |}t|d}|d durkj t |td dSj|}||krj t |td |d |d dS|}|drz t|ddd}Wn ttfyd}Ynw|durˆj||rˆj t |tddSdfdd } j|||| \} } | sj t |td| dSj|} |d} t| tr| rz| j| d}WnPtttfy| }YnBw|drPz"t|ddd}t|p,i}|dd}d|d| d }WnttfyOd| d }Ynwd| d }j t |t|dS)NcSsg|]}|r|qSr%r%).0r`r%r%r& sz+RedeemShopPlugin.handle..argsrrOu未知商品 id：u!，发送 /商店 查看列表。rKrMu!该商品未配置发放逻辑。u积分不足：需要 u ，当前 u。r[r\u-你已拥有该称号，无需重复兑换。rr+cs dSr_r%r%apply_fnrr%r&grantDs z&RedeemShopPlugin.handle..grantu兑换失败：rN)restr]r^u兑换成功，称号「u」已解锁。剩余积分 u兑换成功，剩余积分 rr+)r/r0getattrrpr2rxrsend_forward_msgr rr!rrr rget_user_pointrfrgrhrir3redeem_shop_itemrrr formatKeyErrorr)rr0r product_idrYrKpointsr5rmrokerrrtipmsgrr]r%rr&rsv            zRedeemShopPlugin.handleNrr )rrrr]rJrrrr%r%r%r&rs   )rrrr)r(r)r*rrr+)r(r)rr+)rrUr)rrq)rrr)*__doc__ __future__rryrrtypingrr core.baserrcore.cqr r core.loggerr core.utilsr plugins.titler r ShopApplyr__annotations__r{r#r'r6r=rBrErHrTrZrprrrrrrr%r%r%r&sv           %