content_type -> content_types

This commit is contained in:
Alex Root Junior 2017-06-02 19:41:35 +03:00
parent 765741e122
commit f7c2bb47c1
2 changed files with 21 additions and 21 deletions

View file

@ -104,18 +104,18 @@ class Dispatcher:
def stop_pooling(self):
self._pooling = False
def message_handler(self, commands=None, regexp=None, content_type=None, func=None, custom_filters=None, **kwargs):
def message_handler(self, commands=None, regexp=None, content_types=None, func=None, custom_filters=None, **kwargs):
if commands is None:
commands = []
if content_type is None:
content_type = [ContentType.TEXT]
if content_types is None:
content_types = [ContentType.TEXT]
if custom_filters is None:
custom_filters = []
filters_set = generate_default_filters(*custom_filters,
commands=commands,
regexp=regexp,
content_type=content_type,
content_types=content_types,
func=func,
**kwargs)
@ -125,19 +125,19 @@ class Dispatcher:
return decorator
def edited_message_handler(self, commands=None, regexp=None, content_type=None, func=None, custom_filters=None,
def edited_message_handler(self, commands=None, regexp=None, content_types=None, func=None, custom_filters=None,
**kwargs):
if commands is None:
commands = []
if content_type is None:
content_type = [ContentType.TEXT]
if content_types is None:
content_types = [ContentType.TEXT]
if custom_filters is None:
custom_filters = []
filters_set = generate_default_filters(*custom_filters,
commands=commands,
regexp=regexp,
content_type=content_type,
content_types=content_types,
func=func,
**kwargs)
@ -147,18 +147,18 @@ class Dispatcher:
return decorator
def channel_post_handler(self, commands=None, regexp=None, content_type=None, func=None, *custom_filters, **kwargs):
def channel_post_handler(self, commands=None, regexp=None, content_types=None, func=None, *custom_filters, **kwargs):
if commands is None:
commands = []
if content_type is None:
content_type = [ContentType.TEXT]
if content_types is None:
content_types = [ContentType.TEXT]
if custom_filters is None:
custom_filters = []
filters_set = generate_default_filters(*custom_filters,
commands=commands,
regexp=regexp,
content_type=content_type,
content_types=content_types,
func=func,
**kwargs)
@ -168,19 +168,19 @@ class Dispatcher:
return decorator
def edited_channel_post_handler(self, commands=None, regexp=None, content_type=None, func=None, *custom_filters,
def edited_channel_post_handler(self, commands=None, regexp=None, content_types=None, func=None, *custom_filters,
**kwargs):
if commands is None:
commands = []
if content_type is None:
content_type = [ContentType.TEXT]
if content_types is None:
content_types = [ContentType.TEXT]
if custom_filters is None:
custom_filters = []
filters_set = generate_default_filters(*custom_filters,
commands=commands,
regexp=regexp,
content_type=content_type,
content_types=content_types,
func=func,
**kwargs)
@ -256,15 +256,15 @@ class Dispatcher:
return decorator
async def next_message(self, message: types.Message, otherwise=None, once=False,
regexp=None, content_type=None, func=None, custom_filters=None, **kwargs):
if content_type is None:
content_type = []
regexp=None, content_types=None, func=None, custom_filters=None, **kwargs):
if content_types is None:
content_types = []
if custom_filters is None:
custom_filters = []
filters_set = generate_default_filters(*custom_filters,
regexp=regexp,
content_type=content_type,
content_types=content_types,
func=func,
**kwargs)
self.next_step_message_handlers.register(message, otherwise, once, filters_set)

View file

@ -93,7 +93,7 @@ def generate_default_filters(*args, **kwargs):
filters_set.append(CommandsFilter(filter_))
elif name == 'regexp':
filters_set.append(RegexpFilter(filter_))
elif name == 'content_type':
elif name == 'content_types':
filters_set.append(ContentTypeFilter(filter_))
elif name == 'func':
filters_set.append(filter_)