[Codegen] update-loggers

Author: codegen-bot

.codegen/codemods/update_loggers/update_loggers.py

@@ -0,0 +1,18 @@
+import codegen
+from codegen.sdk.core.codebase import Codebase, 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/'):
+            continue
+
+        if (imp := file.get_import('logging')) is None:
+            continue
+
+        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")

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

@@ -1,8 +1,8 @@
-import logging
+from codegen.shared.logging.get_logger import get_logger
 
 import click
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 @click.command(name="lsp")

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

@@ -1,5 +1,5 @@
 import importlib.util
-import logging
+from codegen.shared.logging.get_logger import get_logger
 import socket
 import subprocess
 import sys
@@ -14,7 +14,7 @@
 
 from codegen.extensions.events.codegen_app import CodegenApp
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 def setup_logging(debug: bool):

src/codegen/extensions/clients/linear.py

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

src/codegen/extensions/events/app.py

@@ -1,12 +1,12 @@
-import logging
+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
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class CodegenApp(modal.App):

src/codegen/extensions/events/codegen_app.py

@@ -1,4 +1,4 @@
-import logging
+from codegen.shared.logging.get_logger import get_logger
 from typing import Any, Optional
 
 from fastapi import FastAPI, Request
@@ -10,7 +10,7 @@
 from .linear import Linear
 from .slack import Slack
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class CodegenApp:

src/codegen/extensions/events/github.py

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

src/codegen/extensions/events/linear.py

@@ -1,12 +1,12 @@
-import logging
+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
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 logger.setLevel(logging.DEBUG)
 
 # Type variable for event types

src/codegen/extensions/events/slack.py

@@ -1,12 +1,12 @@
-import logging
+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
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 logger.setLevel(logging.DEBUG)
 
 

src/codegen/extensions/index/file_index.py

@@ -1,6 +1,6 @@
 """File-level semantic code search index."""
 
-import logging
+from codegen.shared.logging.get_logger import get_logger
 import pickle
 from pathlib import Path
 
@@ -13,7 +13,7 @@
 from codegen.sdk.core.codebase import Codebase
 from codegen.sdk.core.file import File
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class FileIndex(CodeIndex):

src/codegen/extensions/index/symbol_index.py

@@ -1,6 +1,6 @@
 """Symbol-level semantic code search index."""
 
-import logging
+from codegen.shared.logging.get_logger import get_logger
 import pickle
 from pathlib import Path
 
@@ -12,7 +12,7 @@
 from codegen.sdk.core.codebase import Codebase
 from codegen.sdk.core.symbol import Symbol
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 # TODO: WIP!

src/codegen/extensions/linear/linear_client.py

@@ -1,4 +1,4 @@
-import logging
+from codegen.shared.logging.get_logger import get_logger
 import os
 from typing import Optional
 
@@ -8,7 +8,7 @@
 
 from codegen.extensions.linear.types import LinearComment, LinearIssue, LinearTeam, LinearUser
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class LinearClient:

src/codegen/extensions/lsp/definition.py

@@ -1,4 +1,4 @@
-import logging
+from codegen.shared.logging.get_logger import get_logger
 
 from lsprotocol.types import Position
 
@@ -10,7 +10,7 @@
 from codegen.sdk.core.interfaces.editable import Editable
 from codegen.sdk.core.interfaces.has_name import HasName
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 def go_to_definition(node: Editable | None, uri: str, position: Position) -> Editable | None:

src/codegen/extensions/lsp/execute.py

@@ -1,4 +1,4 @@
-import logging
+from codegen.shared.logging.get_logger import get_logger
 from typing import TYPE_CHECKING, Any, Callable
 
 from lsprotocol import types
@@ -9,7 +9,7 @@
 if TYPE_CHECKING:
     from codegen.extensions.lsp.server import CodegenLanguageServer
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 def process_args(args: Any) -> tuple[str, Range]:

src/codegen/extensions/lsp/io.py

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

src/codegen/extensions/lsp/lsp.py

@@ -1,4 +1,4 @@
-import logging
+from codegen.shared.logging.get_logger import get_logger
 
 from lsprotocol import types
 
@@ -14,7 +14,7 @@
 
 version = getattr(codegen, "__version__", "v0.1")
 server = CodegenLanguageServer("codegen", version, protocol_cls=CodegenLanguageServerProtocol)
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 @server.feature(types.TEXT_DOCUMENT_DID_OPEN)

src/codegen/extensions/lsp/server.py

@@ -1,4 +1,4 @@
-import logging
+from codegen.shared.logging.get_logger import get_logger
 from typing import Any, Optional
 
 from lsprotocol import types
@@ -17,7 +17,7 @@
 from codegen.sdk.core.interfaces.editable import Editable
 from codegen.sdk.core.symbol import Symbol
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class CodegenLanguageServer(LanguageServer):

src/codegen/git/clients/git_repo_client.py

@@ -1,4 +1,4 @@
-import logging
+from codegen.shared.logging.get_logger import get_logger
 import time
 from datetime import datetime
 
@@ -20,7 +20,7 @@
 from codegen.git.schemas.repo_config import RepoConfig
 from codegen.git.utils.format import format_comparison
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class GitRepoClient:

src/codegen/git/clients/github_client.py

@@ -1,12 +1,12 @@
-import logging
+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
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class GithubClient:

src/codegen/git/models/codemod_context.py

@@ -1,4 +1,4 @@
-import logging
+from codegen.shared.logging.get_logger import get_logger
 from importlib.metadata import version
 from typing import Any
 
@@ -7,7 +7,7 @@
 
 from codegen.git.models.pull_request_context import PullRequestContext
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class CodemodContext(BaseModel):

src/codegen/git/repo_operator/repo_operator.py

@@ -1,7 +1,7 @@
 import codecs
 import fnmatch
 import glob
-import logging
+from codegen.shared.logging.get_logger import get_logger
 import os
 from collections.abc import Generator
 from datetime import UTC, datetime
@@ -31,7 +31,7 @@
 from codegen.shared.performance.stopwatch_utils import stopwatch
 from codegen.shared.performance.time_utils import humanize_duration
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class RepoOperator:

src/codegen/git/schemas/repo_config.py

@@ -1,4 +1,4 @@
-import logging
+from codegen.shared.logging.get_logger import get_logger
 import os.path
 from pathlib import Path
 
@@ -8,7 +8,7 @@
 from codegen.git.schemas.enums import RepoVisibility
 from codegen.shared.enums.programming_language import ProgrammingLanguage
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class RepoConfig(BaseModel):

src/codegen/git/utils/clone.py

@@ -1,4 +1,4 @@
-import logging
+from codegen.shared.logging.get_logger import get_logger
 import os
 import subprocess
 
@@ -7,7 +7,7 @@
 from codegen.git.utils.remote_progress import CustomRemoteProgress
 from codegen.shared.performance.stopwatch_utils import subprocess_with_stopwatch
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 # TODO: move into RepoOperator

src/codegen/git/utils/codeowner_utils.py

@@ -1,12 +1,12 @@
-import logging
+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
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 def get_filepath_owners(codeowners: CodeOwners, filepath: str) -> set[str]:

src/codegen/git/utils/language.py

@@ -1,12 +1,12 @@
-import logging
+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
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 # Minimum ratio of files that must match the dominant language
 MIN_LANGUAGE_RATIO = 0.1

src/codegen/git/utils/remote_progress.py

@@ -1,11 +1,11 @@
-import logging
+from codegen.shared.logging.get_logger import get_logger
 import time
 
 from git import RemoteProgress
 
 from codegen.git.schemas.enums import FetchResult
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class CustomRemoteProgress(RemoteProgress):