Replace for loops with comprehensions (#629)

This commit is contained in:
Almaz 2021-07-18 14:23:18 +03:00 committed by GitHub
parent ebd7f79326
commit 82d10844d7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 25 additions and 42 deletions

View file

@ -213,12 +213,5 @@ class MongoStorage(BaseStorage):
:return: list of tuples where first element is chat id and second is user id :return: list of tuples where first element is chat id and second is user id
""" """
db = await self.get_db() db = await self.get_db()
result = []
items = await db[STATE].find().to_list() items = await db[STATE].find().to_list()
for item in items: return [(int(item['chat']), int(item['user'])) for item in items]
result.append(
(int(item['chat']), int(item['user']))
)
return result

View file

@ -232,9 +232,7 @@ class Dispatcher(DataMixin, ContextInstanceMixin):
:return: :return:
""" """
if fast: if fast:
tasks = [] tasks = [self.updates_handler.notify(update) for update in updates]
for update in updates:
tasks.append(self.updates_handler.notify(update))
return await asyncio.gather(*tasks) return await asyncio.gather(*tasks)
results = [] results = []

View file

@ -48,9 +48,13 @@ class FiltersFactory:
:param full_config: :param full_config:
:return: :return:
""" """
filters_set = [] filters_set = list(
filters_set.extend(self._resolve_registered(event_handler, self._resolve_registered(
{k: v for k, v in full_config.items() if v is not None})) event_handler,
{k: v for k, v in full_config.items() if v is not None},
)
)
if custom_filters: if custom_filters:
filters_set.extend(custom_filters) filters_set.extend(custom_filters)

View file

@ -242,7 +242,7 @@ class NotFilter(_LogicFilter):
class AndFilter(_LogicFilter): class AndFilter(_LogicFilter):
def __init__(self, *targets): def __init__(self, *targets):
self.targets = list(wrap_async(target) for target in targets) self.targets = [wrap_async(target) for target in targets]
async def check(self, *args): async def check(self, *args):
""" """
@ -268,7 +268,7 @@ class AndFilter(_LogicFilter):
class OrFilter(_LogicFilter): class OrFilter(_LogicFilter):
def __init__(self, *targets): def __init__(self, *targets):
self.targets = list(wrap_async(target) for target in targets) self.targets = [wrap_async(target) for target in targets]
async def check(self, *args): async def check(self, *args):
""" """

View file

@ -131,22 +131,14 @@ class ListField(Field):
def serialize(self, value): def serialize(self, value):
if value is None: if value is None:
return None return None
result = []
serialize = super(ListField, self).serialize serialize = super(ListField, self).serialize
for item in value: return [serialize(item) for item in value]
result.append(serialize(item))
return result
def deserialize(self, value, parent=None): def deserialize(self, value, parent=None):
if value is None: if value is None:
return None return None
result = []
deserialize = super(ListField, self).deserialize deserialize = super(ListField, self).deserialize
for item in value: return [deserialize(item, parent=parent) for item in value]
result.append(deserialize(item, parent=parent))
return result
class ListOfLists(Field): class ListOfLists(Field):
@ -154,9 +146,7 @@ class ListOfLists(Field):
result = [] result = []
serialize = super(ListOfLists, self).serialize serialize = super(ListOfLists, self).serialize
for row in value: for row in value:
row_result = [] row_result = [serialize(item) for item in row]
for item in row:
row_result.append(serialize(item))
result.append(row_result) result.append(row_result)
return result return result
@ -165,9 +155,7 @@ class ListOfLists(Field):
deserialize = super(ListOfLists, self).deserialize deserialize = super(ListOfLists, self).deserialize
if hasattr(value, '__iter__'): if hasattr(value, '__iter__'):
for row in value: for row in value:
row_result = [] row_result = [deserialize(item, parent=parent) for item in row]
for item in row:
row_result.append(deserialize(item, parent=parent))
result.append(row_result) result.append(row_result)
return result return result

View file

@ -62,9 +62,7 @@ class InlineKeyboardMarkup(base.TelegramObject):
:return: self :return: self
:rtype: :obj:`types.InlineKeyboardMarkup` :rtype: :obj:`types.InlineKeyboardMarkup`
""" """
btn_array = [] btn_array = [button for button in args]
for button in args:
btn_array.append(button)
self.inline_keyboard.append(btn_array) self.inline_keyboard.append(btn_array)
return self return self

View file

@ -79,9 +79,7 @@ class ReplyKeyboardMarkup(base.TelegramObject):
:return: self :return: self
:rtype: :obj:`types.ReplyKeyboardMarkup` :rtype: :obj:`types.ReplyKeyboardMarkup`
""" """
btn_array = [] btn_array = [button for button in args]
for button in args:
btn_array.append(button)
self.keyboard.append(btn_array) self.keyboard.append(btn_array)
return self return self

View file

@ -198,10 +198,14 @@ class OrderedHelperMeta(type):
def __new__(mcs, name, bases, namespace, **kwargs): def __new__(mcs, name, bases, namespace, **kwargs):
cls = super().__new__(mcs, name, bases, namespace) cls = super().__new__(mcs, name, bases, namespace)
props_keys = [] props_keys = [
prop_name
for prop_name in (name for name, prop in namespace.items() if isinstance(prop, (Item, ListItem))): for prop_name in (
props_keys.append(prop_name) name
for name, prop in namespace.items()
if isinstance(prop, (Item, ListItem))
)
]
setattr(cls, PROPS_KEYS_ATTR_NAME, props_keys) setattr(cls, PROPS_KEYS_ATTR_NAME, props_keys)