3 _W@sddlmZddlmZddlmZeddddZed dd d Zed d ddZeddddZ eddddZ eddddZ dS))command)get_doc)partialcmdcharsz^!$name$cCsb|jdjdj}xH|D]@}t|}d|d|d<d|d|d<|j|dt|qWdS) z0 !cmdchars -> 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   rZbotscCsr|jdjdj}xX|D]P}t|}d|d|d<d|d|d<d|d|d<|j|d t|qWd S) z0 !bots -> 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|t|ddkr*t|j|n t|j|}|jdjd|fj}|s`|d|n6d|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_bots$    rz cmdchar.usedz ^(!|@)$namecCs|s|j|t|ddkr*t|j|n t|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 >sz#is_cmdchar_used..N)rrrr rr r join)rrrrrrrrrris_cmdchar_used2s$r$z cmdchar.addc Cs@|s|j|t|ddkr*t|j|n t|j|}|j}|jdjd|dj}|rx|d|krx|d|dS|jj |s|j|dS|jdjd|dj }|jdjd |d|d|f|jdj r$|r |d |d|dd j d d |Dfn|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"Yszadd_cmdchar..zCommand char %s added to %sz&Command char %s is already used by %s!N) rrrr splitrr rauthZ is_authedr Zrowcountr#) rrrrrrargsZboZalready_existsrrr add_cmdchar@s$$    ,r(z cmdchar.delcCs|s|j|t|ddkr*t|j|n t|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_s$   r)N) Zutil.decoratorsrZ util.datar functoolsrrrrr$r(r)rrrrs