ä¾TJ ã@sxdZddlmZmZGdd„deƒZGdd„deƒZGdd„deƒZGd d „d e ƒZ d S) z2 ServerState, Channel and ChannelHandler objects. é)ÚChannelÚUserDictc@seZdZdd„ZdS)ÚFancySetcCs|j|ƒ|j|ƒdS)N)ÚdiscardÚadd)ÚselfÚoÚn©r ú&/home/sam/code/Infobot/utils/sstate.pyÚreplaces zFancySet.replaceN)Ú__name__Ú __module__Ú __qualname__r r r r r rs rcs"eZdZ‡fdd†Z‡S)ÚDotDictcs6|dkr&||jƒkr&||Stƒj|ƒS)NÚkeys)rÚsuperÚ__getattribute__)rÚname)Ú __class__r r r s zDotDict.__getattribute__)r rrrr r )rr r s rcsXeZdZdZdd„Zdd„Zdd„Zdd „Z‡fd d †Z‡S) ÚChannelHandlerz A class that handles channels. cCsœt|ƒtkr.|j|t|ƒiƒnjt|ƒtkr‚||jƒkri|j||j|ƒ|j|j|iƒntdt|ƒƒ‚dS)Nz:Type of argument passed to %s.add should be Channel or str)ÚtypeÚstrÚupdaterÚvaluesÚupdate_channelrÚ TypeError)rÚchannelr r r rszChannelHandler.addcCsdS)Nrr )rr r r Ú__str__ szChannelHandler.__str__cCs^||jƒkst‚tƒ}x6|jƒD](}||jjƒkr.|j|ƒq.W|S)N)ÚlowerÚAssertionErrorrrÚusersrr)rÚuserZ channels_Úvr r r Úget_channels_for#s  zChannelHandler.get_channels_forcGs*|d}|d}|j|jO_dS)Nér)r!)rÚargsZto_mergeZ merge_withr r r r+s  zChannelHandler.update_channelcsm|dkr.d||jƒkr.|d|S|ttƒkrPtƒj|ƒStd|||fƒ‚dS)Nrú#z9'%s' object has no attribute '%s', have you joined '#%s'?)rÚdirrrrÚAttributeError)rr)rr r r0s  zChannelHandler.__getattribute__) r rrÚ__doc__rrr$rrr r )rr rs    rc@sReZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd S) Ú ServerStatezp A class that holds the active channels and admins and some more things about the bot that are server-specific. cCsg||_tƒ|_tƒ|_d|_g|_i|_tƒ|_t |ƒ|_ t ƒ|_ dS)NZJames) ÚbotrÚadminsÚmutedÚnickZnoticesÚmessagesrÚchannelsrr!rÚevents)rr,r r r Ú__init__;s       zServerState.__init__cCs|jj|jƒƒdS)z Add an user to the admin list. N)r-rr)rr/r r r Ú add_adminFszServerState.add_admincCs|jj|jƒƒdS)z% Remove an user from the admin list. N)r-Úremover)rÚdnickr r r Ú del_adminJszServerState.del_admincCs|jj|jƒƒdS)z Add an user to the mute list. N)r.rr)rr/r r r ÚmuteNszServerState.mutecCs|jj|jƒƒdS)z% Remove an user from the block list. N)r.r5r)rr6r r r ÚunmuteRszServerState.unmuteN) r rrr*r3r4r7r8r9r r r r r+9s    r+N) r*rrrÚsetrÚdictrrÚobjectr+r r r r Ús &