pythonic findWhere

litestar-org · Aug 18, 2023 · 0cb1251 · 0cb1251
1 parent f7f3da2
commit 0cb1251
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/litestar/contrib/sqlalchemy/dto.py b/litestar/contrib/sqlalchemy/dto.py
@@ -260,9 +260,7 @@ def get_sqlalchemy_dto_config_from_annotated_type(
         Returns:
             The type and config object extracted from the annotation.
         """
-return next((item for item in field_definition.metadata if isinstance(item, SQLAlchemyDTOConfig)), None)
-            return configs[0]
-        return None
+        return next((item for item in field_definition.metadata if isinstance(item, SQLAlchemyDTOConfig)), None)
 
 
 def _detect_defaults(elem: ElementType) -> tuple[Any, Any]:

diff --git a/litestar/dto/base_dto.py b/litestar/dto/base_dto.py
@@ -268,9 +268,7 @@ def get_dto_config_from_annotated_type(field_definition: FieldDefinition) -> DTO
         Returns:
             The type and config object extracted from the annotation.
         """
-        if configs := [item for item in field_definition.metadata if isinstance(item, DTOConfig)]:
-            return configs[0]
-        return None
+        return next((item for item in field_definition.metadata if isinstance(item, DTOConfig)), None)
 
     @classmethod
     def resolve_model_type(cls, field_definition: FieldDefinition) -> FieldDefinition: