reformat

Author: bagel897

ruff.toml

@@ -216,3 +216,5 @@ extend-generics = [
 ]
 [lint.isort]
 known-first-party = ["codegen"]
+[format]
+docstring-code-format = true

src/codegen/sdk/core/detached_symbols/function_call.py

@@ -42,7 +42,7 @@ class FunctionCall(Expression[Parent], HasName, Resolvable, Generic[Parent]):
  """Abstract representation of a function invocation, e.g. in Python:
  ```
  def f():
- g() # FunctionCall
+ g()  # FunctionCall
  ```
  """
 

src/codegen/sdk/core/file.py

@@ -812,11 +812,8 @@ def get_node_by_name(self, name: str) -> Symbol | TImport | None:
  def valid_symbol_names(self) -> dict[str, Symbol | TImport | WildcardImport[TImport]]:
  """Returns a dict mapping name => Symbol (or import) in this file."""
  valid_symbol_names = {}
- for s in self.symbols:
- valid_symbol_names[s.full_name] = s
- for imp in self.imports:
- for name, dest in imp.names:
- valid_symbol_names[name] = dest
+ valid_symbol_names.update({s.full_name: s for s in self.symbols})
+ valid_symbol_names.update({name: dest for imp in self.imports for name, dest in imp.names})
  return valid_symbol_names
 
  @noapidoc

src/codegen/sdk/core/import_resolution.py

@@ -403,6 +403,7 @@ def is_dynamic(self) -> bool:
  def my_function():
  import foo # Dynamic - only imported when function runs
 
+
  if condition:
  from bar import baz # Dynamic - only imported if condition is True
 

src/codegen/sdk/core/interfaces/editable.py

@@ -1123,10 +1123,7 @@ def parent_class(self) -> Class | None:
 
  def _get_ast_children(self) -> list[tuple[str | None, AST]]:
  children = []
- names = {}
- for name, val in self._list_members(include_methods=True).items():
- if isinstance(val, Editable):
- names[val] = name
+ names = {val: name for name, val in self._list_members(include_methods=True).items()}
  for child in self.file._range_index.get_children(self):
  if self.G.config.feature_flags.debug:
  assert child != self, child

src/codegen/sdk/core/symbol_groups/expression_group.py

@@ -59,8 +59,4 @@ def function_calls(self) -> list[FunctionCall]:
  list[FunctionCall]: A list of all function calls found in the expressions
  of this group.
  """
- fcalls = []
- for expr in self.expressions:
- for call in expr.function_calls:
- fcalls.append(call)
- return fcalls
+ return [call for expr in self.expressions for call in expr.function_calls]

src/codegen/sdk/python/statements/comment.py

@@ -77,9 +77,7 @@ def _parse_comment(self) -> str:
  # Remove the triple quotes and extract the text content
  text_block = self.source[3:-3]
  # Parse the text block into lines
- text_lines = []
- for line in text_block.lstrip("\n").split("\n"):
- text_lines.append(line)
+ text_lines = list(text_block.lstrip("\n").split("\n"))
  # Get indentation level
  padding = lowest_indentation(text_lines, skip_lines=skip_lines)
  # Remove indentation

src/codegen/sdk/typescript/file.py

@@ -391,9 +391,7 @@ def valid_import_names(self) -> dict[str, Symbol | TSImport]:
  valid_export_names = {}
  if len(self.default_exports) == 1:
  valid_export_names["default"] = self.default_exports[0]
- for export in self.exports:
- for name, dest in export.names:
- valid_export_names[name] = dest
+ valid_export_names.update({name: dest for export in self.exports for name, dest in export.names})
  return valid_export_names
 
  ####################################################################################################################

src/codegen/sdk/typescript/statements/switch_case.py

@@ -9,9 +9,10 @@
 from codegen.shared.decorators.docs import ts_apidoc
 
 if TYPE_CHECKING:
- from codegen.sdk.codebase.codebase_graph import CodebaseGraph
  from src.codegen.sdk.typescript.statements.switch_statement import TSSwitchStatement
 
+ from codegen.sdk.codebase.codebase_graph import CodebaseGraph
+
 
 @ts_apidoc
 class TSSwitchCase(SwitchCase[TSCodeBlock["TSSwitchStatement"]], TSBlockStatement):