o jje @s@UdZddlmZddlZddlmZmZddlmZmZm Z ddl m Z m Z er/ddl mZe d gdfZiad ed <d Zd dddZd?ddZd@ddZdAddZdAdd ZdAd!d"ZdAd#d$Zd%dd&ed'd(d)d d&ed*d(d+d d&ed,d(d-dd&ed.d(d/Zd ed0<dBd2d3ZdCd4d5ZdDd7d8ZdEd:d;ZdFd=d>ZdS)Guq 积分商店货架与刷新逻辑：数据库 shop_stock 为唯一货架；内存 SHOP_ITEMS 由库表同步。 ) annotationsN)datetime timedelta) TYPE_CHECKINGAnyCallable) TITLE_DEFS get_title_def)RedeemShopPluginr zdict[str, dict[str, Any]] SHOP_ITEMS commonrare legendarytdefdict[str, Any] | NonereturnintcCs<|pidpd}t|tr|}tt|tdS)Nrarityr)get isinstancestrstriplowerrTITLE_PRICE_BY_RARITY)rrr!6/home/dore/onebot/Bot_Ero/plugins/redeem_shop/logic.pytitle_price_from_defs  r#plugin'RedeemShopPlugin'title_idNonecCsP|jj}|dur td|jj||rtd|jjj||dds&tddS)N无法识别用户u你已拥有该称号Fcommitu称号发放失败) bot_eventuser_id RuntimeError dbmanagertitleshasunlock)r$r&uidr!r!r" _grant_title$sr3cCsH|jj}|dur tdttddd}|jjj ||dddS)Nr(r)days%Y-%m-%dFr)) r+r,r-rnowrstrftimer.shop set_draw_pack)r$r2untilr!r!r"_grant_extra_draw_pack.s r;cC0|jj}|dur td|jjj|ddddSNr(rFr))r+r,r-r.r8add_luckr$r2r!r!r"_grant_checkin_boost7r@cCr<r=)r+r,r-r.r8 add_waiverr?r!r!r"_grant_lottery_waiver>rArCcCs>|jj}|dur tdtd}|jjj||dddS)Nr(r5Fr)) r+r,r-rr6r7r.r8clear_draw_count)r$r2todayr!r!r"_grant_lottery_refreshEs rFuB额外抽卡补充包（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)NrJ)rQitemsr)outpidmetar!r!r"_fixed_shop_stock_mappingmsrWc Cs2|j}t|D]\}}t|}|druz t|ddd}Wn tt fy/Yq wt |p5i}| dd}| dpBd}t |trN| }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__r name?rru普通u稀有u传奇ru解锁称号「u」（u）cSs t||S)N)r3)pttr!r!r"s z2refresh_shop_items_from_database..)rHrI initial_stockrKrHrIrKrL)rHrIr_rKrLN)r8 all_stockr clearr startswithrsplit ValueError IndexErrorr rrrrr#rQ) dbrowsrU_stocktidrnmrrcrIrVr!r!r" refresh_shop_items_from_databasetsB            rl list[int]cCsjtt}tt}g}|r)tdt|}tj||d}|D] }t |d|<q|j |t ||S)uK清空商店：随机 4 个称号 + 固定功能商品，并同步内存。)krX) listrkeysdictrWminlenrandomsampleWEEKLY_TITLE_STOCKr8 replace_shelfrl)rfall_idsmappingpickedrorir!r!r"weekly_refresh_shop_shelfs   r| list[str]cCst|g}ttD]A}t|}|d}|d}|j|}|dur'd}n |dkr.d}nt|}|d|d||d |d ||d q |S) uB返回本周货架每条商品的文案行（含空行分隔）。rIrHNu 未上架rGu不限[z] u 售价 u 积分｜剩余 )rlsortedr rqr8rJrappend)rflinesrUrVrIdescstock_nrJr!r!r"format_shop_shelf_liness   rrcCs8t|}gd}|r||n|dd|S)u-每周刷新后发到群里的公告文案。)u积分商店已刷新~ud发送 /商店 查看完整列表，或 /商店 <商品id> 兑换。（方括号里的是商品id）ru—— 本周货架 ——u本周暂无上架商品。 )rextendrjoinrstrip)rfshelfrr!r!r"format_shop_weekly_announcements   r)rrrr)r$r%r&rrr')r$r%rr')rrR)rr')rrm)rr})rr) __doc__ __future__rrurrtypingrrr plugins.titlerr rr ShopApplyr __annotations__rwrr#r3r;r@rCrFrQrWrlr|rrr!r!r!r"sf          %