Automated pre-commit update

Author: caroljung-cg

.codegen/codemods/update_loggers/update_loggers.py

@@ -1,18 +1,18 @@
 import codegen
-from codegen.sdk.core.codebase import Codebase, PyCodebaseType
+from codegen.sdk.core.codebase import PyCodebaseType
 
 
 @codegen.function("update-loggers")
 def run(codebase: PyCodebaseType) -> None:
     """Updates all loggers in src/codegen to use the new get_logger function."""
     for file in codebase.files:
-        if not str(file.filepath).startswith('src/codegen/'):
+        if not str(file.filepath).startswith("src/codegen/"):
             continue
 
-        if (imp := file.get_import('logging')) is None:
+        if (imp := file.get_import("logging")) is None:
             continue
 
-        if logger := file.get_global_var('logger'):
+        if logger := file.get_global_var("logger"):
             print(f"Updating logger in {file.filepath}")
-            logger.value.edit(f"get_logger(__name__)")
-            imp.edit(f"from codegen.shared.logging.get_logger import get_logger")
+            logger.value.edit("get_logger(__name__)")
+            imp.edit("from codegen.shared.logging.get_logger import get_logger")

src/codegen/cli/commands/lsp/lsp.py

@@ -1,7 +1,7 @@
-from codegen.shared.logging.get_logger import get_logger
-
 import click
 
+from codegen.shared.logging.get_logger import get_logger
+
 logger = get_logger(__name__)
 
 

src/codegen/cli/commands/serve/main.py

@@ -1,5 +1,4 @@
 import importlib.util
-from codegen.shared.logging.get_logger import get_logger
 import socket
 import subprocess
 import sys
@@ -13,6 +12,7 @@
 from rich.panel import Panel
 
 from codegen.extensions.events.codegen_app import CodegenApp
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 

src/codegen/extensions/clients/linear.py

@@ -1,9 +1,10 @@
 import json
-from codegen.shared.logging.get_logger import get_logger
 
 import requests
 from pydantic import BaseModel
 
+from codegen.shared.logging.get_logger import get_logger
+
 logger = get_logger(__name__)
 
 

src/codegen/extensions/events/app.py

@@ -1,10 +1,9 @@
-from codegen.shared.logging.get_logger import get_logger
-
 import modal  # deptry: ignore
 
 from codegen.extensions.events.github import GitHub
 from codegen.extensions.events.linear import Linear
 from codegen.extensions.events.slack import Slack
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 

src/codegen/extensions/events/codegen_app.py

@@ -1,10 +1,10 @@
-from codegen.shared.logging.get_logger import get_logger
 from typing import Any, Optional
 
 from fastapi import FastAPI, Request
 from fastapi.responses import HTMLResponse
 
 from codegen.sdk.core.codebase import Codebase
+from codegen.shared.logging.get_logger import get_logger
 
 from .github import GitHub
 from .linear import Linear

src/codegen/extensions/events/github.py

@@ -1,4 +1,3 @@
-from codegen.shared.logging.get_logger import get_logger
 import os
 from typing import Any, Callable, TypeVar
 
@@ -8,6 +7,7 @@
 
 from codegen.extensions.events.interface import EventHandlerManagerProtocol
 from codegen.extensions.github.types.base import GitHubInstallation, GitHubWebhookPayload
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 logger.setLevel(logging.DEBUG)

src/codegen/extensions/events/linear.py

@@ -1,10 +1,10 @@
-from codegen.shared.logging.get_logger import get_logger
 from typing import Any, Callable, TypeVar
 
 from pydantic import BaseModel
 
 from codegen.extensions.events.interface import EventHandlerManagerProtocol
 from codegen.extensions.linear.types import LinearEvent
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 logger.setLevel(logging.DEBUG)

src/codegen/extensions/events/slack.py

@@ -1,10 +1,10 @@
-from codegen.shared.logging.get_logger import get_logger
 import os
 
 from slack_sdk import WebClient
 
 from codegen.extensions.events.interface import EventHandlerManagerProtocol
 from codegen.extensions.slack.types import SlackWebhookPayload
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 logger.setLevel(logging.DEBUG)

src/codegen/extensions/index/file_index.py

@@ -1,6 +1,5 @@
 """File-level semantic code search index."""
 
-from codegen.shared.logging.get_logger import get_logger
 import pickle
 from pathlib import Path
 
@@ -12,6 +11,7 @@
 from codegen.extensions.index.code_index import CodeIndex
 from codegen.sdk.core.codebase import Codebase
 from codegen.sdk.core.file import File
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 

src/codegen/extensions/index/symbol_index.py

@@ -1,6 +1,5 @@
 """Symbol-level semantic code search index."""
 
-from codegen.shared.logging.get_logger import get_logger
 import pickle
 from pathlib import Path
 
@@ -11,6 +10,7 @@
 from codegen.extensions.index.code_index import CodeIndex
 from codegen.sdk.core.codebase import Codebase
 from codegen.sdk.core.symbol import Symbol
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 

src/codegen/extensions/linear/linear_client.py

@@ -1,4 +1,3 @@
-from codegen.shared.logging.get_logger import get_logger
 import os
 from typing import Optional
 
@@ -7,6 +6,7 @@
 from urllib3.util.retry import Retry
 
 from codegen.extensions.linear.types import LinearComment, LinearIssue, LinearTeam, LinearUser
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 

src/codegen/extensions/lsp/definition.py

@@ -1,5 +1,3 @@
-from codegen.shared.logging.get_logger import get_logger
-
 from lsprotocol.types import Position
 
 from codegen.sdk.core.assignment import Assignment
@@ -9,6 +7,7 @@
 from codegen.sdk.core.expressions.name import Name
 from codegen.sdk.core.interfaces.editable import Editable
 from codegen.sdk.core.interfaces.has_name import HasName
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 

src/codegen/extensions/lsp/execute.py

@@ -1,10 +1,10 @@
-from codegen.shared.logging.get_logger import get_logger
 from typing import TYPE_CHECKING, Any, Callable
 
 from lsprotocol import types
 from lsprotocol.types import Position, Range
 
 from codegen.extensions.lsp.codemods.base import CodeAction
+from codegen.shared.logging.get_logger import get_logger
 
 if TYPE_CHECKING:
     from codegen.extensions.lsp.server import CodegenLanguageServer

src/codegen/extensions/lsp/io.py

@@ -1,4 +1,3 @@
-from codegen.shared.logging.get_logger import get_logger
 import pprint
 from dataclasses import dataclass
 from pathlib import Path
@@ -10,6 +9,7 @@
 
 from codegen.sdk.codebase.io.file_io import FileIO
 from codegen.sdk.codebase.io.io import IO
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 

src/codegen/extensions/lsp/lsp.py

@@ -1,5 +1,3 @@
-from codegen.shared.logging.get_logger import get_logger
-
 from lsprotocol import types
 
 import codegen
@@ -11,6 +9,7 @@
 from codegen.extensions.lsp.utils import get_path
 from codegen.sdk.codebase.diff_lite import ChangeType, DiffLite
 from codegen.sdk.core.file import SourceFile
+from codegen.shared.logging.get_logger import get_logger
 
 version = getattr(codegen, "__version__", "v0.1")
 server = CodegenLanguageServer("codegen", version, protocol_cls=CodegenLanguageServerProtocol)

src/codegen/extensions/lsp/server.py

@@ -1,4 +1,3 @@
-from codegen.shared.logging.get_logger import get_logger
 from typing import Any, Optional
 
 from lsprotocol import types
@@ -16,6 +15,7 @@
 from codegen.sdk.core.file import File, SourceFile
 from codegen.sdk.core.interfaces.editable import Editable
 from codegen.sdk.core.symbol import Symbol
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 

src/codegen/git/clients/git_repo_client.py

@@ -1,4 +1,3 @@
-from codegen.shared.logging.get_logger import get_logger
 import time
 from datetime import datetime
 
@@ -19,6 +18,7 @@
 from codegen.git.clients.github_client import GithubClient
 from codegen.git.schemas.repo_config import RepoConfig
 from codegen.git.utils.format import format_comparison
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 

src/codegen/git/clients/github_client.py

@@ -1,11 +1,11 @@
-from codegen.shared.logging.get_logger import get_logger
-
 from github import Consts
 from github.GithubException import UnknownObjectException
 from github.MainClass import Github
 from github.Organization import Organization
 from github.Repository import Repository
 
+from codegen.shared.logging.get_logger import get_logger
+
 logger = get_logger(__name__)
 
 

src/codegen/git/models/codemod_context.py

@@ -1,11 +1,11 @@
-from codegen.shared.logging.get_logger import get_logger
 from importlib.metadata import version
 from typing import Any
 
 from pydantic import BaseModel
 from pydantic.fields import Field
 
 from codegen.git.models.pull_request_context import PullRequestContext
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 

src/codegen/git/repo_operator/repo_operator.py

@@ -1,7 +1,6 @@
 import codecs
 import fnmatch
 import glob
-from codegen.shared.logging.get_logger import get_logger
 import os
 from collections.abc import Generator
 from datetime import UTC, datetime
@@ -28,6 +27,7 @@
 from codegen.git.utils.codeowner_utils import create_codeowners_parser_for_repo
 from codegen.git.utils.file_utils import create_files
 from codegen.git.utils.remote_progress import CustomRemoteProgress
+from codegen.shared.logging.get_logger import get_logger
 from codegen.shared.performance.stopwatch_utils import stopwatch
 from codegen.shared.performance.time_utils import humanize_duration
 

src/codegen/git/schemas/repo_config.py

@@ -1,4 +1,3 @@
-from codegen.shared.logging.get_logger import get_logger
 import os.path
 from pathlib import Path
 
@@ -7,6 +6,7 @@
 from codegen.configs.models.repository import RepositoryConfig
 from codegen.git.schemas.enums import RepoVisibility
 from codegen.shared.enums.programming_language import ProgrammingLanguage
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 

src/codegen/git/utils/clone.py

@@ -1,10 +1,10 @@
-from codegen.shared.logging.get_logger import get_logger
 import os
 import subprocess
 
 from git import Repo as GitRepo
 
 from codegen.git.utils.remote_progress import CustomRemoteProgress
+from codegen.shared.logging.get_logger import get_logger
 from codegen.shared.performance.stopwatch_utils import subprocess_with_stopwatch
 
 logger = get_logger(__name__)

src/codegen/git/utils/codeowner_utils.py

@@ -1,10 +1,9 @@
-from codegen.shared.logging.get_logger import get_logger
-
 from codeowners import CodeOwners
 from github.PullRequest import PullRequest
 
 from codegen.git.clients.git_repo_client import GitRepoClient
 from codegen.git.configs.constants import CODEOWNERS_FILEPATHS
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 

src/codegen/git/utils/language.py

@@ -1,10 +1,10 @@
-from codegen.shared.logging.get_logger import get_logger
 from collections import Counter
 from pathlib import Path
 from typing import Literal
 
 from codegen.git.utils.file_utils import split_git_path
 from codegen.shared.enums.programming_language import ProgrammingLanguage
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 

src/codegen/git/utils/remote_progress.py

@@ -1,9 +1,9 @@
-from codegen.shared.logging.get_logger import get_logger
 import time
 
 from git import RemoteProgress
 
 from codegen.git.schemas.enums import FetchResult
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 

src/codegen/gscli/backend/typestub_utils.py

@@ -1,12 +1,13 @@
 import ast
-from codegen.shared.logging.get_logger import get_logger
 import os
 import re
 from collections.abc import Callable
 from concurrent.futures import ThreadPoolExecutor
 
 import astor
 
+from codegen.shared.logging.get_logger import get_logger
+
 logger = get_logger(__name__)
 
 

src/codegen/gscli/generate/commands.py

@@ -1,5 +1,4 @@
 import json
-from codegen.shared.logging.get_logger import get_logger
 import os
 import re
 import shutil
@@ -16,6 +15,7 @@
 from codegen.sdk.code_generation.codegen_sdk_codebase import get_codegen_sdk_codebase
 from codegen.sdk.code_generation.doc_utils.generate_docs_json import generate_docs_json
 from codegen.sdk.code_generation.mdx_docs_generation import render_mdx_page_for_class
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)
 

src/codegen/runner/clients/client.py

@@ -1,11 +1,10 @@
 """Client used to abstract the weird stdin/stdout communication we have with the sandbox"""
 
-from codegen.shared.logging.get_logger import get_logger
-
 import requests
 from fastapi import params
 
 from codegen.runner.models.apis import ServerInfo
+from codegen.shared.logging.get_logger import get_logger
 
 logger = get_logger(__name__)