[Codegen] update-loggers

Author: codegen-bot

.codegen/codemods/update_loggers/update_loggers.py

@@ -12,7 +12,7 @@ def run(codebase: PyCodebaseType) -> None:
         if file.get_import("logging") is None:
             continue
 
-        if logger := file.get_global_var("logger") and logger.value.source == "logging.getLogger(__name__)":
+        if (logger := file.get_global_var("logger")) and logger.value.source == "logging.getLogger(__name__)":
             print(f"Updating logger in {file.filepath}")
             logger.set_value("get_logger(__name__)")
             file.add_import_from_import_string("\nfrom codegen.shared.logging.get_logger import get_logger")

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

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

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

@@ -1,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import importlib.util
 import logging
 import socket
@@ -14,7 +16,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,12 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import json
 import logging
 
 import requests
 from pydantic import BaseModel
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 # --- TYPES

src/codegen/extensions/events/app.py

@@ -1,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 
 import modal  # deptry: ignore
@@ -6,7 +8,7 @@
 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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 from typing import Any, Optional
 
@@ -10,7 +12,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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 import os
 from typing import Any, Callable, TypeVar
@@ -9,7 +11,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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 from typing import Any, Callable, TypeVar
 
@@ -6,7 +8,7 @@
 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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 import os
 
@@ -6,7 +8,7 @@
 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,5 +1,7 @@
 """File-level semantic code search index."""
 
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 import pickle
 from pathlib import Path
@@ -13,7 +15,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,5 +1,7 @@
 """Symbol-level semantic code search index."""
 
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 import pickle
 from pathlib import Path
@@ -12,7 +14,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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 import os
 from typing import Optional
@@ -8,7 +10,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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 
 from lsprotocol.types import Position
@@ -10,7 +12,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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 from typing import TYPE_CHECKING, Any, Callable
 
@@ -9,7 +11,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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 import pprint
 from dataclasses import dataclass
@@ -11,7 +13,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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 
 from lsprotocol import types
@@ -14,7 +16,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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 from typing import Any, Optional
 
@@ -17,7 +19,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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 import time
 from datetime import datetime
@@ -20,7 +22,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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 
 from github import Consts
@@ -6,7 +8,7 @@
 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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 from importlib.metadata import version
 from typing import Any
@@ -7,7 +9,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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import codecs
 import fnmatch
 import glob
@@ -31,7 +33,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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 import os.path
 from pathlib import Path
@@ -8,7 +10,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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 import os
 import subprocess
@@ -7,7 +9,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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 
 from codeowners import CodeOwners
@@ -6,7 +8,7 @@
 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,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 from collections import Counter
 from pathlib import Path
@@ -6,7 +8,7 @@
 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,13 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import logging
 import time
 
 from git import RemoteProgress
 
 from codegen.git.schemas.enums import FetchResult
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class CustomRemoteProgress(RemoteProgress):

src/codegen/gscli/backend/typestub_utils.py

@@ -1,3 +1,5 @@
+
+from codegen.shared.logging.get_logger import get_logger
 import ast
 import logging
 import os
@@ -7,7 +9,7 @@
 
 import astor
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class MethodRemover(ast.NodeTransformer):