示例#1
0
    def __init__(self, context):
        self.context = context
        module = self.__class__.__module__
        self.config = config = context.config.get(module, {})
        self.log = logging.getLogger(module)
        self.log.debug('Config: %r', config)

        if 'cmd' in context.config:  # in case of
            config['cmd'] = context.config['cmd']
        context.config['cmd'] = self.cmd = config.get('cmd', '!')
        context.config['re_cmd'] = re.escape(self.cmd)

        self.use_shlex = self.config.get('use_shlex', True)
        self.antiflood = self.config.get('antiflood', False)
        self.case_sensitive = self.config.get('casesensitive',
                                              self.case_sensitive)

        guard = utils.maybedotted(config.get('guard', self.default_policy))
        self.log.debug('Guard: %s', guard.__name__)
        self.guard = guard(context)

        self.error_format = utils.maybedotted(
            config.get('error_format', "Invalid arguments.".format))
        self.handles = defaultdict(Done)
        self.tasks = defaultdict(Done)

        self.aliases = {}
示例#2
0
文件: command.py 项目: gawel/irc3
    def __init__(self, context):
        self.context = context
        module = self.__class__.__module__
        self.config = config = context.config.get(module, {})
        self.log = logging.getLogger(module)
        self.log.debug('Config: %r', config)

        if 'cmd' in context.config:  # in case of
            config['cmd'] = context.config['cmd']
        context.config['cmd'] = self.cmd = config.get('cmd', '!')
        context.config['re_cmd'] = re.escape(self.cmd)

        self.use_shlex = self.config.get('use_shlex', True)
        self.antiflood = self.config.get('antiflood', False)
        self.case_sensitive = self.config.get('casesensitive',
                                              self.case_sensitive)

        guard = utils.maybedotted(config.get('guard', self.default_policy))
        self.log.debug('Guard: %s', guard.__name__)
        self.guard = guard(context)

        self.error_format = utils.maybedotted(config.get('error_format',
                                              "Invalid arguments.".format))
        self.handles = defaultdict(Done)
        self.tasks = defaultdict(Done)

        self.aliases = {}
示例#3
0
 def test_maybedotted(self):
     self.assertTrue(maybedotted('irc3.plugins') is irc3.plugins)
     self.assertTrue(maybedotted('irc3.utils.IrcString') is IrcString)
     self.assertTrue(maybedotted(IrcString) is IrcString)
     self.assertRaises(LookupError, maybedotted, 'irc3.none.none')
     self.assertRaises(LookupError, maybedotted, 'irc3.none')
     self.assertRaises(LookupError, maybedotted, None)
     self.assertRaises(LookupError, maybedotted, '')
示例#4
0
 def test_maybedotted(self):
     self.assertTrue(
         maybedotted('irc3.plugins') is irc3.plugins)
     self.assertTrue(
         maybedotted('irc3.utils.IrcString') is IrcString)
     self.assertTrue(
         maybedotted(IrcString) is IrcString)
     self.assertRaises(LookupError, maybedotted, 'irc3.none.none')
     self.assertRaises(LookupError, maybedotted, 'irc3.none')
     self.assertRaises(LookupError, maybedotted, None)
     self.assertRaises(LookupError, maybedotted, '')
示例#5
0
 def __init__(self, bot):
     self.bot = bot
     self.loop = bot.loop
     self.config = cfg = bot.config.get('dcc', {})
     self.config.update(send_limit_rate=int(cfg.get('send_limit_rate', 0)),
                        send_block_size=int(
                            cfg.get('send_block_size', DCCSend.block_size)))
     self.connections = {}
     self.protocols = {}
     for klass in (DCCChat, DCCGet, DCCSend):
         n = klass.type
         self.config.update({
             n + '_limit':
             int(cfg.get(n + '_limit', 100)),
             n + '_user_limit':
             int(cfg.get(n + '_user_limit', 1)),
             n + '_accept_timeout':
             int(cfg.get(n + '_accept_timeout', 60)),
             n + '_idle_timeout':
             int(cfg.get(n + '_idle_timeout', 60 * 5)),
         })
         klass = maybedotted(self.config.get(n + '_protocol', klass))
         self.connections[n] = {'total': 0, 'masks': defaultdict(dict)}
         self.protocols[n] = klass
     self.seeks = {}
示例#6
0
    def __init__(self, context):
        self.context = context
        module = self.__class__.__module__
        self.config = config = context.config.get(module, {})
        self.log = logging.getLogger(module)
        self.log.debug('Config: %r', config)

        if 'cmd' in context.config:  # in case of
            config['cmd'] = context.config['cmd']
        context.config['cmd'] = self.cmd = config.get('cmd', '!')
        if self.cmd in WEIRD_CHARS:
            context.config['re_cmd'] = r'\%s' % self.cmd
        else:
            context.config['re_cmd'] = self.cmd

        self.antiflood = self.config.get('antiflood', False)
        self.case_sensitive = self.config.get('casesensitive',
                                              self.case_sensitive)

        guard = utils.maybedotted(config.get('guard', self.default_policy))
        self.log.debug('Guard: %s', guard.__name__)
        self.guard = guard(context)

        self.handles = defaultdict(Done)
        self.tasks = defaultdict(Done)
示例#7
0
def reconnect(bot, mask, target, args):
    """force reconnect

        %%reconnect
    """
    plugin = bot.get_plugin(utils.maybedotted('irc3.plugins.core.Core'))
    bot.loop.call_soon(plugin.reconnect)
示例#8
0
文件: command.py 项目: winny-/irc3
    def __init__(self, context):
        self.context = context
        module = self.__class__.__module__
        self.config = config = context.config.get(module, {})
        self.log = logging.getLogger(module)
        self.log.debug('Config: %r', config)

        if 'cmd' in context.config:  # in case of
            config['cmd'] = context.config['cmd']
        context.config['cmd'] = self.cmd = config.get('cmd', '!')
        if self.cmd in WEIRD_CHARS:
            context.config['re_cmd'] = r'\%s' % self.cmd
        else:
            context.config['re_cmd'] = self.cmd

        self.antiflood = self.config.get('antiflood', False)
        self.case_sensitive = self.config.get('casesensitive',
                                              self.case_sensitive)

        guard = utils.maybedotted(config.get('guard', self.default_policy))
        self.log.debug('Guard: %s', guard.__name__)
        self.guard = guard(context)

        self.handles = defaultdict(Done)
        self.tasks = defaultdict(Done)
示例#9
0
文件: command.py 项目: gawel/irc3
def reconnect(bot, mask, target, args):
    """force reconnect

        %%reconnect
    """
    plugin = bot.get_plugin(utils.maybedotted('irc3.plugins.core.Core'))
    bot.loop.call_soon(plugin.reconnect)
示例#10
0
文件: command.py 项目: jpcw/irc3
 def __init__(self, bot):
     self.bot = bot
     self.config = config = bot.config.get(__name__, {})
     self.log = logging.getLogger(__name__)
     self.log.debug('Config: %r', config)
     bot.config['cmd'] = self.cmd = config.get('cmd', '!')
     guard = utils.maybedotted(config.get('guard', free_policy))
     self.log.debug('Guard: %s', guard.__name__)
     self.guard = guard(bot)
示例#11
0
文件: command.py 项目: Mika64/irc3
 def __init__(self, bot):
     self.bot = bot
     self.config = config = bot.config.get(__name__, {})
     self.log = logging.getLogger(__name__)
     self.log.debug('Config: %r', config)
     bot.config['cmd'] = self.cmd = config.get('cmd', '!')
     guard = utils.maybedotted(config.get('guard', free_policy))
     self.log.debug('Guard: %s', guard.__name__)
     self.guard = guard(bot)
示例#12
0
 def callback(context, name, ob):
     obj = context.context
     if info.scope == 'class':
         callback = func.__get__(obj.get_plugin(ob), ob)
     else:
         callback = utils.wraps_with_context(func, obj)
     plugin = obj.get_plugin(utils.maybedotted(commands))
     predicates.update(module=func.__module__)
     cmd_name = predicates.get('name', func.__name__)
     plugin[cmd_name] = (predicates, callback)
     obj.log.debug('Register command %r', cmd_name)
示例#13
0
文件: command.py 项目: pypingou/irc3
 def callback(context, name, ob):
     obj = context.context
     if info.scope == 'class':
         callback = func.__get__(obj.get_plugin(ob), ob)
     else:
         @functools.wraps(func)
         def wrapper(*args, **kwargs):
             return func(obj, *args, **kwargs)
         callback = wrapper
     plugin = obj.get_plugin(utils.maybedotted(commands))
     predicates.update(module=func.__module__)
     plugin[func.__name__] = (predicates, callback)
     obj.log.debug('Register command %r', func.__name__)
示例#14
0
 def callback(context, name, ob):
     obj = context.context
     if info.scope == 'class':
         callback = func.__get__(obj.get_plugin(ob), ob)
     else:
         callback = utils.wraps_with_context(func, obj)
     plugin = obj.get_plugin(utils.maybedotted(commands))
     predicates.update(module=func.__module__)
     cmd_name = predicates.get('name', func.__name__)
     if not plugin.case_sensitive:
         cmd_name = cmd_name.lower()
     plugin[cmd_name] = (predicates, callback)
     obj.log.debug('Register command %r', cmd_name)
示例#15
0
    def callback(context, name, ob):
        obj = context.context
        if info.scope == 'class':
            callback = func.__get__(obj.get_plugin(ob), ob)
        else:

            @functools.wraps(func)
            def wrapper(*args, **kwargs):
                return func(obj, *args, **kwargs)

            callback = wrapper
        plugin = obj.get_plugin(utils.maybedotted(commands))
        predicates.update(module=func.__module__)
        plugin[func.__name__] = (predicates, callback)
        obj.log.debug('Register command %r', func.__name__)
示例#16
0
    def create(self, name_or_class, mask, filepath=None, **kwargs):
        """Create a new DCC connection. Return an ``asyncio.Protocol``"""
        if isinstance(name_or_class, type):
            name = name_or_class.type
            protocol = name_or_class
        else:
            name = name_or_class
            protocol = self.protocols[name]
        assert name in DCC_TYPES
        if filepath:
            kwargs.setdefault('limit_rate', self.config['send_limit_rate'])
            kwargs['filepath'] = filepath
            if protocol.type == DCCSend.type:
                kwargs.setdefault('offset', 0)
                kwargs.update(
                    filename_safe=slugify(os.path.basename(filepath)),
                    filesize=os.path.getsize(filepath),
                )
            elif protocol.type == DCCGet.type:
                try:
                    offset = os.path.getsize(filepath)
                except OSError:
                    offset = 0
                kwargs.setdefault('offset', offset)
                kwargs.setdefault('resume', False)
        kwargs.setdefault('port', None)
        f = protocol(mask=mask,
                     ip=int(self.bot.ip),
                     bot=self.bot,
                     loop=self.loop,
                     **kwargs)

        if kwargs['port']:
            if self.bot.config.get('dcc_sock_factory'):
                sock_factory = maybedotted(self.bot.config.dcc_sock_factory)
                args = dict(sock=sock_factory(self.bot, f.host, f.port))
            else:
                args = dict(host=f.host, port=f.port)
            task = self.bot.create_task(
                self.loop.create_connection(f.factory, **args))
            task.add_done_callback(partial(self.created, f))
        else:
            task = self.bot.create_task(
                self.loop.create_server(f.factory, '0.0.0.0', 0, backlog=1))
            task.add_done_callback(partial(self.created, f))
        return f
示例#17
0
文件: manager.py 项目: Harmon758/irc3
    def create(self, name_or_class, mask, filepath=None, **kwargs):
        """Create a new DCC connection. Return an ``asyncio.Protocol``"""
        if isinstance(name_or_class, type):
            name = name_or_class.type
            protocol = name_or_class
        else:
            name = name_or_class
            protocol = self.protocols[name]
        assert name in DCC_TYPES
        if filepath:
            kwargs.setdefault('limit_rate',
                              self.config['send_limit_rate'])
            kwargs['filepath'] = filepath
            if protocol.type == DCCSend.type:
                kwargs.setdefault('offset', 0)
                kwargs.update(
                    filename_safe=slugify(os.path.basename(filepath)),
                    filesize=os.path.getsize(filepath),
                )
            elif protocol.type == DCCGet.type:
                try:
                    offset = os.path.getsize(filepath)
                except OSError:
                    offset = 0
                kwargs.setdefault('offset', offset)
                kwargs.setdefault('resume', False)
        kwargs.setdefault('port', None)
        f = protocol(
            mask=mask, ip=int(self.bot.ip),
            bot=self.bot, loop=self.loop, **kwargs)

        if kwargs['port']:
            if self.bot.config.get('dcc_sock_factory'):
                sock_factory = maybedotted(self.bot.config.dcc_sock_factory)
                args = dict(sock=sock_factory(self.bot, f.host, f.port))
            else:
                args = dict(host=f.host, port=f.port)
            task = self.bot.create_task(
                self.loop.create_connection(f.factory, **args))
            task.add_done_callback(partial(self.created, f))
        else:
            task = self.bot.create_task(
                self.loop.create_server(
                    f.factory, '0.0.0.0', 0, backlog=1))
            task.add_done_callback(partial(self.created, f))
        return f
示例#18
0
    def __init__(self, bot):
        self.bot = bot
        self.config = config = bot.config.get(__name__, {})
        self.log = logging.getLogger(__name__)
        self.log.debug('Config: %r', config)

        if 'cmd' in bot.config:  # in case of
            config['cmd'] = bot.config['cmd']
        bot.config['cmd'] = self.cmd = config.get('cmd', '!')

        self.antiflood = self.config.get('antiflood', False)

        guard = utils.maybedotted(config.get('guard', free_policy))
        self.log.debug('Guard: %s', guard.__name__)
        self.guard = guard(bot)

        self.handles = defaultdict(Done)
示例#19
0
文件: command.py 项目: saimn/irc3
    def __init__(self, bot):
        self.bot = bot
        self.config = config = bot.config.get(__name__, {})
        self.log = logging.getLogger(__name__)
        self.log.debug('Config: %r', config)

        if 'cmd' in bot.config:  # in case of
            config['cmd'] = bot.config['cmd']
        bot.config['cmd'] = self.cmd = config.get('cmd', '!')

        self.antiflood = self.config.get('antiflood', False)

        guard = utils.maybedotted(config.get('guard', free_policy))
        self.log.debug('Guard: %s', guard.__name__)
        self.guard = guard(bot)

        self.handles = defaultdict(Done)
示例#20
0
文件: command.py 项目: gawel/irc3
 def callback(context, name, ob):
     obj = context.context
     if info.scope == 'class':
         callback = func.__get__(obj.get_plugin(ob), ob)
     else:
         callback = utils.wraps_with_context(func, obj)
     plugin = obj.get_plugin(utils.maybedotted(commands))
     predicates.update(module=func.__module__)
     cmd_name = predicates.get('name', func.__name__)
     if not plugin.case_sensitive:
         cmd_name = cmd_name.lower()
     plugin[cmd_name] = (predicates, callback)
     aliases = predicates.get('aliases', None)
     if aliases is not None:
         for alias in aliases:
             plugin.aliases[alias] = cmd_name
         obj.log.debug('Register command %r %r', cmd_name, aliases)
     else:
         obj.log.debug('Register command %r', cmd_name)
示例#21
0
 def __init__(self, bot):
     self.bot = bot
     self.loop = bot.loop
     self.config = cfg = bot.config.get('dcc', {})
     self.config.update(
         send_limit_rate=int(cfg.get('send_limit_rate', 0)),
         send_block_size=int(cfg.get('send_block_size', DCCSend.block_size))
     )
     self.connections = {}
     self.protocols = {}
     for klass in (DCCChat, DCCGet, DCCSend):
         n = klass.type
         self.config.update({
             n + '_limit': int(cfg.get(n + '_limit', 100)),
             n + '_user_limit': int(cfg.get(n + '_user_limit', 1)),
             n + '_accept_timeout': int(cfg.get(n + '_accept_timeout', 60)),
             n + '_idle_timeout': int(cfg.get(n + '_idle_timeout', 60 * 5)),
         })
         klass = maybedotted(self.config.get(n + '_protocol', klass))
         self.connections[n] = {'total': 0, 'masks': defaultdict(dict)}
         self.protocols[n] = klass
     self.seeks = {}