o h[^@sbddlZddlmZddlZddlZddlZddlmZmZddlZGdddej Z ddZ dS)N)commands)datetime timedeltac@seZdZddZddZdJddZdd Zd d Zd d ZddZ ddZ ddZ e j ddZddZddZe dJdejfddZe dKdefdd Ze e jd!d"dejd#efd$d%Ze e jd!d"dejd#efd&d'Ze jd!d(e jd!d"d)d*Zee jd!d"d+d,Zee jd!d"d-d.Zee jd!d"d#efd/d0Zee jd!d"d1efd2d3Z ee jd!d"dJd4ej!fd5d6Z"ee jd!d"d7ed8ej#fd9d:Z$ee jd!d"d7efd;d<Z%e j d=d>Z&ee jd!d"d4ej!fd?d@Z'ee jd!d"d4ej!dAe(fdBdCZ)ee jd!d"dejfdDdEZ*ee jd!d"dFdGZ+ee jd!d"dHdIZ,dS)LLevelingcCs,||_d|_d|_||i|_dS)Nzleveling_config.jsonzleveling_data.json)bot config_file data_file load_config load_data last_message)selfrr GC:\Users\Buzz\Documents\TitanTS\Discord\TitanBot\files\cogs\leveling.py__init__ s  zLeveling.__init__c Csttj|jr#t|jd }t|}Wdn1swYnddddigidddd }||||_dS) u$Načte konfiguraci leveling systémurNT<Fu<🎉 Gratulujeme {user}! Dosáhl jsi **Level {level}**! 🎊) enabledxp_per_message xp_cooldownlevel_up_channel level_rolesblacklisted_channelsmultiplier_channelsreset_on_leavelevel_up_messageshow_level_card) ospathexistsropenjsonload save_configconfig)r fr$r r rr s"  zLeveling.load_configNcCsP|dur|j}t|jd}tj||ddWddS1s!wYdS)uUloží konfiguraciNwindent)r$r rr!dump)r r$r%r r rr#(s "zLeveling.save_configcCsVtj|jr&t|jd}t||_WddS1swYdSi|_dS)uNačte uživatelská datarN)rrrrr r!r" user_datar r%r r rr /s " zLeveling.load_datacCsDt|jd}tj|j|ddWddS1swYdS)uUloží uživatelská datar&r'r(N)r rr!r*r+r,r r r save_data7s"zLeveling.save_datacCstdt|S)u!Vypočítá level na základě XP皙?N)mathfloorsqrt)r xpr r rcalculate_level<szLeveling.calculate_levelcCst|ddS)u)Vypočítá XP potřebné pro daný levelr.r'N)r/ceil)r levelr r rcalculate_xp_needed@szLeveling.calculate_xp_neededcCs0t|}||jvrddddd|j|<|j|S)uZíská data uživatelerNr2r5messagesr )strr+)r user_idr r r get_user_dataDs  zLeveling.get_user_datacCst||}|d}|d|7<||d|d<|dd7<t|d<||d|kr8|dSdS)uPřidá XP uživatelir5r2r8r N)r;r3rnow isoformatr-)r r:amountr+ old_levelr r radd_xpPs  zLeveling.add_xpc s|jdsdS|jjrdS|jj|jdvrdSt|jj}t}||jvr:||j| }||jdkr:dS||j|<|jd}|jd t|jjd}t ||}| |jj|}|ro| |j|j|IdHdSdS)u!Event pro zprávy - přidává XPrNrrrr?)r$authorrchannelidr9rr=r total_secondsgetintrAhandle_level_up) r messager: current_timeZ time_diffZbase_xp multiplierZfinal_xp new_levelr r r on_messagebs*      zLeveling.on_messagec s"|jdr|j|jd}n|}|r|jdj|j||jd}|jdr~tjd|dt d}|j |j j d|j d d |d d d ||j}||d }|d} || } |j dd | dd d d |j dd | ddd d |j|dIdHn||IdH|||IdHdS)zZpracuje level uprr)userr5usernamer🎉 Level Up!title descriptioncolor timestampurlu📊 Nový Level**Tnamevalueinliner<r2⭐ Celkové XP,u🎯 Do dalšího leveluz XP**embedN)r$r get_channelformatmentionr\discordEmbedrr= set_thumbnaildisplay_avatarrY add_fieldr;rEr6sendassign_level_role) r rOrDr5Ztarget_channelrJrbr+ next_level_xp current_xp needed_xpr r rrIs8    zLeveling.handle_level_upcst|tjs dSd}|jdD] \}}|t|kr|}q|rU|jt|}|rW||jvrYz|j |d|ddIdHWdStj yTt d|j YdSwdSdSdS)u#Přiřadí roli na základě leveluNrzLevel u dosažen)reasonu%❌ Nemám oprávnění přidat roli ) isinstancerfMemberr$itemsrHguildget_roleroles add_roles Forbiddenprintr\)r rOr5Zrole_to_assignZ level_reqrole_idroler r rrls$  "zLeveling.assign_level_rolerOc sdur|j|j}tjdjdtd}|jj j d|d}|d}|dkr5| |nd}| |d }||} ||} | | d } d } t | | d } d | d | | }|j dd|ddd|j dd|dddd|j dd|ddddd|j dd|d| dd| dd| dd dddd|jD}|jd d!dd"z tfd#d$t|d D}|j d%d&|d't|ddWn tyYnw|j|d(IdHdS))uZobrazí rank kartu uživateleNu 📊 Rank - eX)rTrVrWrXr5r2rr<du█u░u 📈 LevelrZTr[r_r`u 💬 Zprávyr8u!🎯 Progress do dalšího levelu`z` z.1fz% **z** / **z** XPFcS$g|]\}}|ddkr||fqSr2rr .0uiddatar r r $z!Leveling.rank..cS |ddSNr<r2r xr r r zLeveling.rank..keyreversec3s*|]\}\}}|tjkr|VqdSN)r9rE)rir_rOr r s(z Leveling.rank..u🏆 Rank na serveruz**#z** z ra)rCr;rErfrg display_namerr=rhrirYr6rHrjr+rssortnext enumeratelen StopIterationrk)r ctxrOr+rbZ current_levelrnZcurrent_level_xprmZ progress_xproZprogress_percentZ bar_lengthZ filled_lengthZbar all_usersrankr rrrsj     " z Leveling.rankr<pagec sdd|jD}|jdddd|s |dIdHdSd }tt||}td t||}|d |}||}|||}t j d d |d |dt d} d} t ||d D]O\} \} } zD|jt| }|r| d krtdn | dkrzdn| dkrdnd}| |d| d|jd7} | d| dd| ddd| d dd!7} Wq]Yq]d |d |d"| | _|d kr| jd#|jd$|d d%d&|j| d'IdHdS)(u.Zobrazí žebříček nejlepších uživatelůcSrrr rr r rrrz(Leveling.leaderboard..cSrrr rr r rrrz&Leveling.leaderboard..Tru📭 Žebříček je prázdný!N r<u🏆 Leaderboardu Stránka /irSu🥇r'u🥈u🥉u🔸z **#z**  z Level r5u • r2r`u XP • r8u zpráv z uPoužij z leaderboard u pro další stránku)textra)r+rsrrkr/r4rmaxminrfrgrr=rrget_userrHrrU set_footerprefix)r rrrZusers_per_pageZ total_pages start_idxZend_idxZ page_usersrbZleaderboard_textrr:rrOemojir r r leaderboardsB  (, zLeveling.leaderboardT) administratorr?cs||j}||j|}tjddd}|jd|jdd|jdd|d ddd|jd d||jd d ddd|rN|jd d |ddd|j|dIdHdS)uPřidá XP uživateli (admin)u⭐ XP PřidánorRrTrV👤 UživatelTr[u➕ Přidáno XPrZr` 📊 Nové XPr2rQLevel **FraN) r;rEcopyrArfrgrjrerk)r rrOr?old_datarMrbr r raddxp-s&zLeveling.addxpcs||j}|d}|d}td|d||d<||d|d<|tjddd}|jd|jdd |jd d |d d dd |jd d |dd d dd |d|krj|jdd|dd|ddd |j |dIdHdS)uOdebere XP uživateli (admin)r2r5ru➖ XP OdebránorrTr[u➖ Odebráno XPrZr`ru📉 Level Downrz** (z )FraN) r;rErr3r-rfrgrjrerk)r rrOr?r+Zold_xpr@rbr r rremovexpAs"  "zLeveling.removexp)invoke_without_commandc stjddd}|j}|jdd|d|d|d|d |d |d |d |d |ddd|jd}|r` - Nastaví XP za zprávu `u5levelconfig cooldown ` - Nastaví cooldown `u@levelconfig channel <#channel>` - Nastaví kanál pro level up `u;levelconfig addrole <@role>` - Přidá level roli `z6levelconfig removerole ` - Odebere level roli `u;levelconfig blacklist <#channel>` - Zakáže XP v kanálu `uKlevelconfig multiplier <#channel> ` - Nastaví multiplikátor `u0levelconfig reset <@user>` - Resetuje uživateleFr[rz<#> Nenastavenoru✅u❌rrrru📊 Aktuální nastaveníz Zapnuto: u XP za zprávu: z Cooldown: us Level kanál: z Level role: u Zakázané kanály: raN)rfrgrrjr$rrk) r rrbrrZchannel_displayZenabled_statusZ xp_amountZ cooldown_timeZ roles_countZblacklisted_countr r r levelconfigZsd     zLeveling.levelconfigc(d|jd<||dIdHdS)uZapne leveling systémTru✅ Leveling systém zapnut!Nr$r#rkr rr r renable zLeveling.enablecr)uVypne leveling systémFru❌ Leveling systém vypnut!Nrrr r rdisablerzLeveling.disablecT|dks |dkr|dIdHdS||jd<||d|dIdHdS)uNastaví XP za zprávur<r}u❌ XP musí být mezi 1-100!Nru✅ XP za zprávu nastaveno na !rkr$r#)r rr?r r rr2 z Leveling.xpsecondscr)u#Nastaví cooldown pro získání XPriu+❌ Cooldown musí být mezi 0-3600 sekund!Nru✅ XP cooldown nastaven na z sekund!r)r rrr r rcooldownrzLeveling.cooldownrDcsX|durd|jd<|dIdHn|j|jd<|d|jdIdH|dS)u$Nastaví kanál pro level up zprávyNru✅ Level up kanál resetován!u ✅ Level up kanál nastaven na r)r$rkrErer#r rrDr r rrDs   zLeveling.channelr5r{cs^|dkr|dIdHdS|j|jdt|<||d|jd|dIdHdS)uPřidá roli pro určitý levelr<u ❌ Level musí být alespoň 1!Nru ✅ Role u přidána pro level r)rkrEr$r9r#re)r rr5r{r r raddroles$zLeveling.addrolecsdt||jdvr$|jdt|=||d|dIdHdS|d|dIdHdS)uOdebere roli pro určitý levelru✅ Role pro level u odebrána!Nu❌ Žádná role pro level u není nastavena!)r9r$r#rk)r rr5r r r removeroles zLeveling.removerolecs@|jddrt|j}||jvr|j|=|dSdSdS)u6Reset dat při opuštění serveru (pokud je povoleno)rFN)r$rGr9rEr+r-)r memberr:r r ron_member_removes   zLeveling.on_member_removecsx|j|jdvr |jd|j|d|jdIdHn|jd|j|d|jdIdH|dS)u&Přidá/odebere kanál ze zakázanýchru ✅ Kanál u odebrán ze zakázaných!Nu přidán do zakázaných!)rEr$removerkreappendr#rr r r blacklists zLeveling.blacklistrLcs|dks |dkr|dIdHdS|dkr8t|j|jdvr7|jdt|j=|d|jdIdHn||jdt|j<|d|jd |d IdH|dS) u%Nastaví XP multiplikátor pro kanálr.g$@u.❌ Multiplikátor musí být mezi 0.1 - 10.0!NrBru✅ Multiplikátor pro u odstraněn!z nastaven na zx!)rkr9rEr$rer#)r rrDrLr r rrLs  zLeveling.multiplierc st|j}||jvrY|j|}ddddd|j|<|tjdd|jddd}|jd d |d d |d dd|dddd|jdddd|j |dIdHdS| d|jdIdHdS)uResetuje XP a level uživatelerNr7u🔄 Reset UživatelezXP a level pro u byl resetován!r)rTrUrVu📊 Předchozí datazLevel: r5z XP: r2r`u Zprávy: r8Tr[u📊 Nová datauLevel: 0 XP: 0 Zprávy: 0rau❌ Uživatel u$ nemá žádná data k resetování!) r9rEr+rr-rfrgrerjrk)r rrOr:rrbr r rresets&    2zLeveling.resetcs|jds|dIdHdStjddd}d}t|jddd d D]&\}}|jt|}|rA|d |d |j d 7}q&|d |d|d7}q&|pPd|_ |j|dIdHdS)uZobrazí seznam level rolíru*📭 Žádné level role nejsou nastaveny!Nu🎭 Level Roler|rrcSs t|dS)Nr)rHrr r rr*rz$Leveling.listroles..)rz**Level z:** ru:** Neplatná role (ID: ) uŽádné role nenalezenyra) r$rkrfrgsortedrsrtrurHrerU)r rrbZ roles_textr5rzr{r r r listroless  " zLeveling.listrolescstjddd}|jdr9d}|jdD]}|j|}|r(|d|jd7}q|d|d 7}q|jd |d d |jd rvd}|jd D]&\}}|jt|}|rb|d|jd|d7}qG|d|d|d7}qG|jd|d d |jds|jd sd|_ |j |dIdHdS)u.Zobrazí blacklisted kanály a multiplikátoryu📋 Kanály Konfiguracer|rrru• ru• Neplatný kanál (ID: ru🚫 Zakázané kanályFr[rz: **zx** z): **u⚡ Multiplikátoryu5Žádné speciální kanály nejsou nakonfigurovány.raN) rfrgr$rtrcrerjrsrHrUrk)r rrbZblacklist_text channel_idrDZmultiplier_textrLr r r listchannels4s0   zLeveling.listchannelsr)r<)-__name__ __module__ __qualname__rr r#r r-r3r6r;rArCoglistenerrNrIrlcommandrfrrrrHrhas_permissionsrrgrouprrrr2r TextChannelrDRolerrrrfloatrLrrrr r r rr s   "(A*     /                rcs|t|IdHdSr)add_cogr)rr r rsetupXsr) rf discord.extrr!rr/rrasynciorrrr r r rs  S