_W@sddlmZddlmZddlmZeddddZed dd d Zed d ddZeddddZ eddddZ eddddZ dS))command)get_doc)partialcmdcharsz^!$name$cCs{|jdjdj}xX|D]P}t|}d|d|d get all cmdchars currently in use dbzqSELECT cmdchar, string_agg(bot, ', ' ORDER BY bot) FROM bots GROUP BY cmdchar;z02%srz10%sru %s — %sN)dataexecutefetchalllistnoticetuple)botnickchanarginfoitemr./plugins/cmdchar.py get_cmdcharss   rZbotscCs|jdjdj}xj|D]b}t|}d|d|d get all bots that are known to the db rz~SELECT bot, owner, string_agg(cmdchar, ' ' ORDER BY cmdchar) FROM bots GROUP BY bot, owner;z02%srz10%srz11%sz %s [%s] - %sN)rr r r r r )rrrrrrrrrget_botss  rrz ^(!|@)$name\sc Cs|s|j|tn|ddkr>t|j|nt|j|}|jdjd|fj}|s|d|nAd|d}d|d}d |d } |d ||| fd S) z= {!@}bot -> get information about a bot in the db r!rzSELECT bot, owner, string_agg(cmdchar, ' ' ORDER BY cmdchar) FROM bots WHERE bot = %s GROUP BY bot, owner;z%s is an unknown bot.z02%sz10%srz11%srz %s [%s] - %sN)msgrrr rr fetchone) rrrgrrmsg_fnrZbot_nickownerrrrrget_bots4"rz cmdchar.usedz ^(!|@)$namecCs|s|j|tn|ddkr>t|j|nt|j|}|jdjd|fj}|s|d|n*|d|djdd |Dfd S) z[ {!@}cmdchar.used -> tells you whether the cmdchar you passed in is used already rrrz(SELECT bot FROM bots WHERE cmdchar = %s;z %s is unused.z%s is used by %sz, cSsg|]}|dqS)rr).0irrr >s z#is_cmdchar_used..N)rrrr rr r join)rrrrrrrrrris_cmdchar_used2s4"r$z cmdchar.addc Cs|s|j|tn|ddkr>t|j|nt|j|}|j}|jdjd|dj}|r|d|kr|d|dSn|jj |s|j|dS|jdjd|dj }|jdjd |d|d|f|jdj r|re|d |d|dd j d d |Dfq|d|d|dfn|d|d|dfdS)z5 {!@}cmdchar.add -> add a bots entry rrrz;SELECT owner FROM bots WHERE lower(bot) = lower(%s) LIMIT 1rzYou don't own %s!z@You are not registered with NickServ or not properly identified.z'SELECT bot FROM bots WHERE cmdchar = %szBINSERT INTO bots (cmdchar, bot, owner) VALUES (%s, %s, %s);z.Command char %s added to %s (also used by: %s)z, cSsg|]}|dqS)rr)r r!rrrr"Ys zadd_cmdchar..zCommand char %s added to %sz&Command char %s is already used by %s!N) rrrr splitrr rauth is_authedr rowcountr#) rrrrrrargsZboZalready_existsrrr add_cmdchar@s$4 ## 8r*z cmdchar.delcCs|s|j|tn|ddkr>t|j|nt|j|}|j}|jdjd|d|dfj}|s|d|d|dfS|d|kr|d|dS|jdjd|d|df|d |d|dfd S) z8 {!@}cmdchar.del -> remove a bots entry rrrzMSELECT cmdchar, owner FROM bots WHERE lower(bot) = lower(%s) AND cmdchar = %srz"Command char %s is not used by %s!zYou don't own %s!z>DELETE FROM bots WHERE lower(bot) = lower(%s) AND cmdchar = %sz Removed command char %s from %s.N)rrrr r%rr r)rrrrrrr)rrrr del_cmdchar_s4 -r+N) Zutil.decoratorsrZ util.datar functoolsrrrrr$r*r+rrrrs