diff --git a/README.md b/README.md index f448a7d61..b270105a0 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,4 @@ -# Graph sitter +# Codegen [![Documentation](https://img.shields.io/badge/docs-docs.codegen.com-blue)](https://docs.codegen.com) [![Unit Tests](https://github.com/codegen-sh/codegen-sdk/actions/workflows/unit-tests.yml/badge.svg)](https://github.com/codegen-sh/codegen-sdk/actions/workflows/unit-tests.yml) diff --git a/hatch.toml b/hatch.toml index 12054f94a..8d99a1ea6 100644 --- a/hatch.toml +++ b/hatch.toml @@ -72,6 +72,8 @@ macos-max-compat = false [build] packages = [ "src/codegen/sdk", + "src/codegen/git", + "src/codegen/utils", "src/codegen/gscli", "src/graph_visualization", ] diff --git a/pyproject.toml b/pyproject.toml index afd6ab8e6..d585b81e9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -62,6 +62,7 @@ dependencies = [ "toml>=0.10.2", "PyGithub==2.5.0", "GitPython==3.1.44", + "psutil>=5.8.0", ] license = {file = "LICENSE"} classifiers = [ diff --git a/src/codegen/git/repo_operator/remote_repo_operator.py b/src/codegen/git/repo_operator/remote_repo_operator.py index 604dc72db..27feea8af 100644 --- a/src/codegen/git/repo_operator/remote_repo_operator.py +++ b/src/codegen/git/repo_operator/remote_repo_operator.py @@ -16,7 +16,7 @@ from codegen.git.utils.clone_url import get_clone_url_for_repo_config, url_to_github from codegen.git.utils.codeowner_utils import create_codeowners_parser_for_repo from codegen.git.utils.remote_progress import CustomRemoteProgress -from codegen.git.utils.stopwatch_utils import stopwatch +from codegen.utils.performance.stopwatch_utils import stopwatch logger = logging.getLogger(__name__) diff --git a/src/codegen/git/repo_operator/repo_operator.py b/src/codegen/git/repo_operator/repo_operator.py index 961546bf4..25e8c8b23 100644 --- a/src/codegen/git/repo_operator/repo_operator.py +++ b/src/codegen/git/repo_operator/repo_operator.py @@ -17,8 +17,8 @@ from codegen.git.configs.constants import CODEGEN_BOT_EMAIL, CODEGEN_BOT_NAME from codegen.git.schemas.enums import CheckoutResult, FetchResult from codegen.git.schemas.repo_config import BaseRepoConfig -from codegen.git.utils.stopwatch_utils import stopwatch -from codegen.git.utils.time_utils import humanize_duration +from codegen.utils.performance.stopwatch_utils import stopwatch +from codegen.utils.time_utils import humanize_duration logger = logging.getLogger(__name__) diff --git a/src/codegen/git/utils/clone.py b/src/codegen/git/utils/clone.py index 9a8f22ac4..fc7075951 100644 --- a/src/codegen/git/utils/clone.py +++ b/src/codegen/git/utils/clone.py @@ -5,7 +5,7 @@ from codegen.git.schemas.github import GithubType from codegen.git.schemas.repo_config import RepoConfig from codegen.git.utils.clone_url import get_authenticated_clone_url_for_repo_config -from codegen.git.utils.stopwatch_utils import subprocess_with_stopwatch +from codegen.utils.performance.stopwatch_utils import subprocess_with_stopwatch logger = logging.getLogger(__name__) diff --git a/src/codegen/gscli/generate/runner_imports.py b/src/codegen/gscli/generate/runner_imports.py index df4863bcc..12ceeac41 100644 --- a/src/codegen/gscli/generate/runner_imports.py +++ b/src/codegen/gscli/generate/runner_imports.py @@ -19,7 +19,7 @@ """ # TODO: these should also be made public (i.e. included in the docs site) GS_PRIVATE_IMPORTS = """ -from codegen.sdk.codebase.control_flow import StopCodemodException +from codegen.utils.exceptions.control_flow import StopCodemodException """.strip() IMPORT_STRING_TEMPLATE = """ diff --git a/src/codegen/sdk/code_generation/current_code_codebase.py b/src/codegen/sdk/code_generation/current_code_codebase.py index d4501402c..cbb64c348 100644 --- a/src/codegen/sdk/code_generation/current_code_codebase.py +++ b/src/codegen/sdk/code_generation/current_code_codebase.py @@ -9,7 +9,7 @@ from codegen.sdk.codebase.config import CodebaseConfig, DefaultConfig, ProjectConfig from codegen.sdk.core.codebase import Codebase, CodebaseType from codegen.sdk.enums import ProgrammingLanguage -from codegen.sdk.writer_decorators import DocumentedObject, apidoc_objects, no_apidoc_objects, py_apidoc_objects, ts_apidoc_objects +from codegen.utils.decorators.docs import DocumentedObject, apidoc_objects, no_apidoc_objects, py_apidoc_objects, ts_apidoc_objects logger = logging.getLogger(__name__) diff --git a/src/codegen/sdk/codebase/codebase_graph.py b/src/codegen/sdk/codebase/codebase_graph.py index c9a75b5ba..a8fdc0eca 100644 --- a/src/codegen/sdk/codebase/codebase_graph.py +++ b/src/codegen/sdk/codebase/codebase_graph.py @@ -16,10 +16,8 @@ from rustworkx import PyDiGraph, WeightedEdgeList from codegen.git.repo_operator.repo_operator import RepoOperator -from codegen.git.utils.stopwatch_utils import stopwatch, stopwatch_with_sentry from codegen.sdk.codebase.config import CodebaseConfig, DefaultConfig, ProjectConfig, SessionOptions from codegen.sdk.codebase.config_parser import ConfigParser, get_config_parser_for_language -from codegen.sdk.codebase.control_flow import StopCodemodException from codegen.sdk.codebase.diff_lite import ChangeType, DiffLite from codegen.sdk.codebase.flagging.flags import Flags from codegen.sdk.codebase.transaction_manager import TransactionManager @@ -35,6 +33,8 @@ from codegen.sdk.extensions.sort import sort_editables from codegen.sdk.extensions.utils import uncache_all from codegen.sdk.typescript.external.ts_declassify.ts_declassify import TSDeclassify +from codegen.utils.exceptions.control_flow import StopCodemodException +from codegen.utils.performance.stopwatch_utils import stopwatch, stopwatch_with_sentry if TYPE_CHECKING: from codegen.sdk.codebase.node_classes.node_classes import NodeClasses diff --git a/src/codegen/sdk/codebase/flagging/enums.py b/src/codegen/sdk/codebase/flagging/enums.py index 6375484b0..c080adefa 100644 --- a/src/codegen/sdk/codebase/flagging/enums.py +++ b/src/codegen/sdk/codebase/flagging/enums.py @@ -3,7 +3,7 @@ from typing_extensions import ReadOnly -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc @apidoc diff --git a/src/codegen/sdk/codebase/flagging/flags.py b/src/codegen/sdk/codebase/flagging/flags.py index 1c6332d89..817d0c25b 100644 --- a/src/codegen/sdk/codebase/flagging/flags.py +++ b/src/codegen/sdk/codebase/flagging/flags.py @@ -4,7 +4,7 @@ from codegen.sdk.codebase.flagging.code_flag import CodeFlag from codegen.sdk.codebase.flagging.enums import MessageType from codegen.sdk.core.interfaces.editable import Editable -from codegen.sdk.writer_decorators import noapidoc +from codegen.utils.decorators.docs import noapidoc if TYPE_CHECKING: from app.codemod.types import Group diff --git a/src/codegen/sdk/codebase/span.py b/src/codegen/sdk/codebase/span.py index 59aa175b2..62d667fe1 100644 --- a/src/codegen/sdk/codebase/span.py +++ b/src/codegen/sdk/codebase/span.py @@ -4,7 +4,7 @@ from pydantic_core.core_schema import ValidationInfo from tree_sitter import Point, Range -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc def validate_range(value: Any, info: ValidationInfo) -> Range: diff --git a/src/codegen/sdk/codebase/transaction_manager.py b/src/codegen/sdk/codebase/transaction_manager.py index 1ea184a1e..f3a74465c 100644 --- a/src/codegen/sdk/codebase/transaction_manager.py +++ b/src/codegen/sdk/codebase/transaction_manager.py @@ -3,7 +3,6 @@ from pathlib import Path from typing import TYPE_CHECKING -from codegen.sdk.codebase.control_flow import MaxPreviewTimeExceeded, MaxTransactionsExceeded from codegen.sdk.codebase.diff_lite import DiffLite from codegen.sdk.codebase.transactions import ( EditTransaction, @@ -14,6 +13,7 @@ Transaction, TransactionPriority, ) +from codegen.utils.exceptions.control_flow import MaxPreviewTimeExceeded, MaxTransactionsExceeded if TYPE_CHECKING: from codegen.sdk.core.file import File diff --git a/src/codegen/sdk/core/assignment.py b/src/codegen/sdk/core/assignment.py index ea792ce5c..0cd1c1181 100644 --- a/src/codegen/sdk/core/assignment.py +++ b/src/codegen/sdk/core/assignment.py @@ -30,7 +30,7 @@ from codegen.sdk.extensions.sort import sort_editables from codegen.sdk.typescript.expressions.object_type import TSObjectType from codegen.sdk.utils import find_index -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.expressions.type import Type diff --git a/src/codegen/sdk/core/class_definition.py b/src/codegen/sdk/core/class_definition.py index 522b28192..5304f56d7 100644 --- a/src/codegen/sdk/core/class_definition.py +++ b/src/codegen/sdk/core/class_definition.py @@ -30,7 +30,7 @@ from codegen.sdk.core.symbol_groups.parents import Parents from codegen.sdk.enums import SymbolType from codegen.sdk.extensions.utils import cached_property -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc from codegen.visualizations.enums import VizNode if TYPE_CHECKING: diff --git a/src/codegen/sdk/core/codebase.py b/src/codegen/sdk/core/codebase.py index c88b9ed86..0c018bc3a 100644 --- a/src/codegen/sdk/core/codebase.py +++ b/src/codegen/sdk/core/codebase.py @@ -23,13 +23,11 @@ from codegen.git.repo_operator.repo_operator import RepoOperator from codegen.git.schemas.enums import CheckoutResult from codegen.git.schemas.repo_config import BaseRepoConfig -from codegen.git.utils.stopwatch_utils import stopwatch from codegen.sdk._proxy import proxy_property from codegen.sdk.ai.helpers import AbstractAIHelper, MultiProviderAIHelper from codegen.sdk.codebase.codebase_ai import generate_system_prompt, generate_tools from codegen.sdk.codebase.codebase_graph import GLOBAL_FILE_IGNORE_LIST, CodebaseGraph from codegen.sdk.codebase.config import CodebaseConfig, DefaultConfig, ProjectConfig, SessionOptions -from codegen.sdk.codebase.control_flow import MaxAIRequestsError from codegen.sdk.codebase.diff_lite import DiffLite from codegen.sdk.codebase.flagging.code_flag import CodeFlag from codegen.sdk.codebase.flagging.enums import FlagKwargs @@ -71,7 +69,9 @@ from codegen.sdk.typescript.symbol import TSSymbol from codegen.sdk.typescript.type_alias import TSTypeAlias from codegen.sdk.utils import determine_project_language -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc +from codegen.utils.exceptions.control_flow import MaxAIRequestsError +from codegen.utils.performance.stopwatch_utils import stopwatch from codegen.visualizations.visualization_manager import VisualizationManager if TYPE_CHECKING: diff --git a/src/codegen/sdk/core/dataclasses/usage.py b/src/codegen/sdk/core/dataclasses/usage.py index d217186b8..996cae6ec 100644 --- a/src/codegen/sdk/core/dataclasses/usage.py +++ b/src/codegen/sdk/core/dataclasses/usage.py @@ -6,7 +6,7 @@ from dataclasses_json import dataclass_json -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc if TYPE_CHECKING: from codegen.sdk.core.detached_symbols.function_call import FunctionCall diff --git a/src/codegen/sdk/core/detached_symbols/argument.py b/src/codegen/sdk/core/detached_symbols/argument.py index bf72dc559..bbf4e3da6 100644 --- a/src/codegen/sdk/core/detached_symbols/argument.py +++ b/src/codegen/sdk/core/detached_symbols/argument.py @@ -15,7 +15,7 @@ from codegen.sdk.core.interfaces.importable import Importable from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/detached_symbols/code_block.py b/src/codegen/sdk/core/detached_symbols/code_block.py index d13720f59..910f10b7f 100644 --- a/src/codegen/sdk/core/detached_symbols/code_block.py +++ b/src/codegen/sdk/core/detached_symbols/code_block.py @@ -25,7 +25,7 @@ from codegen.sdk.extensions.sort import sort_editables from codegen.sdk.extensions.utils import find_line_start_and_end_nodes from codegen.sdk.output.ast import AST -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.interfaces.has_block import HasBlock diff --git a/src/codegen/sdk/core/detached_symbols/decorator.py b/src/codegen/sdk/core/detached_symbols/decorator.py index 0c561cc76..065f4d44a 100644 --- a/src/codegen/sdk/core/detached_symbols/decorator.py +++ b/src/codegen/sdk/core/detached_symbols/decorator.py @@ -11,7 +11,7 @@ from codegen.sdk.core.expressions import Expression from codegen.sdk.core.expressions.name import Name from codegen.sdk.core.interfaces.has_name import HasName -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.class_definition import Class diff --git a/src/codegen/sdk/core/detached_symbols/function_call.py b/src/codegen/sdk/core/detached_symbols/function_call.py index 1ad84f040..dc3b385e7 100644 --- a/src/codegen/sdk/core/detached_symbols/function_call.py +++ b/src/codegen/sdk/core/detached_symbols/function_call.py @@ -25,7 +25,7 @@ from codegen.sdk.extensions.utils import cached_property, is_descendant_of from codegen.sdk.typescript.enums import TSFunctionTypeNames from codegen.sdk.utils import find_first_ancestor -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/detached_symbols/parameter.py b/src/codegen/sdk/core/detached_symbols/parameter.py index 52431695a..494f91955 100644 --- a/src/codegen/sdk/core/detached_symbols/parameter.py +++ b/src/codegen/sdk/core/detached_symbols/parameter.py @@ -21,7 +21,7 @@ from codegen.sdk.extensions.autocommit import commiter from codegen.sdk.extensions.resolution import UsageKind from codegen.sdk.utils import find_first_descendant -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.expressions.type import Type diff --git a/src/codegen/sdk/core/directory.py b/src/codegen/sdk/core/directory.py index a81d06f48..ab842a341 100644 --- a/src/codegen/sdk/core/directory.py +++ b/src/codegen/sdk/core/directory.py @@ -3,7 +3,7 @@ from pathlib import Path from typing import TYPE_CHECKING, Generic, Self, TypeVar -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc if TYPE_CHECKING: from codegen.sdk.core.assignment import Assignment diff --git a/src/codegen/sdk/core/export.py b/src/codegen/sdk/core/export.py index 21d7a88a1..987d29318 100644 --- a/src/codegen/sdk/core/export.py +++ b/src/codegen/sdk/core/export.py @@ -10,7 +10,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.symbol_groups.collection import Collection from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.statements.export_statement import ExportStatement diff --git a/src/codegen/sdk/core/expressions/await_expression.py b/src/codegen/sdk/core/expressions/await_expression.py index 85ee3a1fe..01b420269 100644 --- a/src/codegen/sdk/core/expressions/await_expression.py +++ b/src/codegen/sdk/core/expressions/await_expression.py @@ -6,7 +6,7 @@ from codegen.sdk.core.interfaces.has_value import HasValue from codegen.sdk.core.interfaces.wrapper_expression import IWrapper from codegen.sdk.extensions.autocommit import reader -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/core/expressions/binary_expression.py b/src/codegen/sdk/core/expressions/binary_expression.py index ead1f48f7..e3637b9a1 100644 --- a/src/codegen/sdk/core/expressions/binary_expression.py +++ b/src/codegen/sdk/core/expressions/binary_expression.py @@ -16,7 +16,7 @@ from codegen.sdk.core.symbol_groups.expression_group import ExpressionGroup from codegen.sdk.extensions.autocommit import commiter, reader from codegen.sdk.extensions.sort import sort_editables -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/core/expressions/boolean.py b/src/codegen/sdk/core/expressions/boolean.py index fc4ad74fe..17a90e6ab 100644 --- a/src/codegen/sdk/core/expressions/boolean.py +++ b/src/codegen/sdk/core/expressions/boolean.py @@ -5,7 +5,7 @@ from codegen.sdk.core.expressions.builtin import Builtin from codegen.sdk.core.interfaces.has_name import HasName from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc Parent = TypeVar("Parent", bound="Expression") diff --git a/src/codegen/sdk/core/expressions/builtin.py b/src/codegen/sdk/core/expressions/builtin.py index fd4feddae..de2658738 100644 --- a/src/codegen/sdk/core/expressions/builtin.py +++ b/src/codegen/sdk/core/expressions/builtin.py @@ -5,7 +5,7 @@ from codegen.sdk.core.interfaces.chainable import Chainable from codegen.sdk.core.interfaces.has_attribute import HasAttribute from codegen.sdk.extensions.autocommit import reader -from codegen.sdk.writer_decorators import noapidoc +from codegen.utils.decorators.docs import noapidoc if TYPE_CHECKING: from codegen.sdk.core.external_module import ExternalModule diff --git a/src/codegen/sdk/core/expressions/chained_attribute.py b/src/codegen/sdk/core/expressions/chained_attribute.py index f071db786..6415aa5f0 100644 --- a/src/codegen/sdk/core/expressions/chained_attribute.py +++ b/src/codegen/sdk/core/expressions/chained_attribute.py @@ -12,7 +12,7 @@ from codegen.sdk.core.interfaces.has_attribute import HasAttribute from codegen.sdk.core.interfaces.resolvable import Resolvable from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.interfaces.has_name import HasName diff --git a/src/codegen/sdk/core/expressions/comparison_expression.py b/src/codegen/sdk/core/expressions/comparison_expression.py index dfed264c4..4af29c22b 100644 --- a/src/codegen/sdk/core/expressions/comparison_expression.py +++ b/src/codegen/sdk/core/expressions/comparison_expression.py @@ -4,7 +4,7 @@ from codegen.sdk.core.expressions import Expression from codegen.sdk.core.expressions.binary_expression import BinaryExpression from codegen.sdk.core.symbol_groups.expression_group import ExpressionGroup -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc Parent = TypeVar("Parent") diff --git a/src/codegen/sdk/core/expressions/defined_name.py b/src/codegen/sdk/core/expressions/defined_name.py index 47d21684f..ab8c9f703 100644 --- a/src/codegen/sdk/core/expressions/defined_name.py +++ b/src/codegen/sdk/core/expressions/defined_name.py @@ -4,7 +4,7 @@ from codegen.sdk.codebase.resolution_stack import ResolutionStack from codegen.sdk.core.expressions import Name from codegen.sdk.extensions.autocommit import reader -from codegen.sdk.writer_decorators import noapidoc +from codegen.utils.decorators.docs import noapidoc if TYPE_CHECKING: from codegen.sdk.core.symbol import Symbol diff --git a/src/codegen/sdk/core/expressions/expression.py b/src/codegen/sdk/core/expressions/expression.py index 8e5f1ad0a..230a875cf 100644 --- a/src/codegen/sdk/core/expressions/expression.py +++ b/src/codegen/sdk/core/expressions/expression.py @@ -6,7 +6,7 @@ from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.enums import NodeType from codegen.sdk.extensions.autocommit import reader -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/core/expressions/generic_type.py b/src/codegen/sdk/core/expressions/generic_type.py index 93fe5e487..9c3811176 100644 --- a/src/codegen/sdk/core/expressions/generic_type.py +++ b/src/codegen/sdk/core/expressions/generic_type.py @@ -13,7 +13,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.symbol_groups.collection import Collection from codegen.sdk.extensions.resolution import ResolutionStack -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/expressions/multi_expression.py b/src/codegen/sdk/core/expressions/multi_expression.py index bad2fa5e2..bd8bd522d 100644 --- a/src/codegen/sdk/core/expressions/multi_expression.py +++ b/src/codegen/sdk/core/expressions/multi_expression.py @@ -9,7 +9,7 @@ from codegen.sdk.core.interfaces.has_name import HasName from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/expressions/name.py b/src/codegen/sdk/core/expressions/name.py index 4e825cc44..917a6de88 100644 --- a/src/codegen/sdk/core/expressions/name.py +++ b/src/codegen/sdk/core/expressions/name.py @@ -7,7 +7,7 @@ from codegen.sdk.core.expressions.expression import Expression from codegen.sdk.core.interfaces.resolvable import Resolvable from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.interfaces.has_name import HasName diff --git a/src/codegen/sdk/core/expressions/named_type.py b/src/codegen/sdk/core/expressions/named_type.py index 17461d8a3..8af8db419 100644 --- a/src/codegen/sdk/core/expressions/named_type.py +++ b/src/codegen/sdk/core/expressions/named_type.py @@ -14,7 +14,7 @@ from codegen.sdk.core.interfaces.importable import Importable from codegen.sdk.core.interfaces.resolvable import Resolvable from codegen.sdk.core.node_id_factory import NodeId -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/expressions/none_type.py b/src/codegen/sdk/core/expressions/none_type.py index 30e4b6d42..b122a27e6 100644 --- a/src/codegen/sdk/core/expressions/none_type.py +++ b/src/codegen/sdk/core/expressions/none_type.py @@ -7,7 +7,7 @@ from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.core.interfaces.importable import Importable from codegen.sdk.extensions.autocommit import reader -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/core/expressions/number.py b/src/codegen/sdk/core/expressions/number.py index d6440a4ee..7b34fec48 100644 --- a/src/codegen/sdk/core/expressions/number.py +++ b/src/codegen/sdk/core/expressions/number.py @@ -5,7 +5,7 @@ from codegen.sdk.core.expressions.builtin import Builtin from codegen.sdk.core.interfaces.has_name import HasName from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc Parent = TypeVar("Parent", bound="Expression") diff --git a/src/codegen/sdk/core/expressions/parenthesized_expression.py b/src/codegen/sdk/core/expressions/parenthesized_expression.py index 56bbac7f5..530e3e2c8 100644 --- a/src/codegen/sdk/core/expressions/parenthesized_expression.py +++ b/src/codegen/sdk/core/expressions/parenthesized_expression.py @@ -10,7 +10,7 @@ from codegen.sdk.core.interfaces.wrapper_expression import IWrapper from codegen.sdk.extensions.autocommit import reader from codegen.sdk.typescript.statements.if_block_statement import TSIfBlockStatement -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/core/expressions/placeholder_type.py b/src/codegen/sdk/core/expressions/placeholder_type.py index e932a247c..789ec4563 100644 --- a/src/codegen/sdk/core/expressions/placeholder_type.py +++ b/src/codegen/sdk/core/expressions/placeholder_type.py @@ -8,7 +8,7 @@ from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.core.interfaces.importable import Importable from codegen.sdk.extensions.autocommit import reader -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/core/expressions/string.py b/src/codegen/sdk/core/expressions/string.py index b1225dd07..a3737424c 100644 --- a/src/codegen/sdk/core/expressions/string.py +++ b/src/codegen/sdk/core/expressions/string.py @@ -11,7 +11,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.symbol_groups.collection import Collection from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/expressions/subscript_expression.py b/src/codegen/sdk/core/expressions/subscript_expression.py index 41ff7f5fc..55e1e647d 100644 --- a/src/codegen/sdk/core/expressions/subscript_expression.py +++ b/src/codegen/sdk/core/expressions/subscript_expression.py @@ -9,7 +9,7 @@ from codegen.sdk.core.interfaces.chainable import Chainable from codegen.sdk.core.interfaces.resolvable import Resolvable from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.interfaces.has_name import HasName diff --git a/src/codegen/sdk/core/expressions/ternary_expression.py b/src/codegen/sdk/core/expressions/ternary_expression.py index 5918f6650..66cadc02c 100644 --- a/src/codegen/sdk/core/expressions/ternary_expression.py +++ b/src/codegen/sdk/core/expressions/ternary_expression.py @@ -12,7 +12,7 @@ from codegen.sdk.core.interfaces.importable import Importable from codegen.sdk.core.interfaces.unwrappable import Unwrappable from codegen.sdk.extensions.autocommit import commiter, reader -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/core/expressions/tuple_type.py b/src/codegen/sdk/core/expressions/tuple_type.py index 5eddf7127..2964b4b2c 100644 --- a/src/codegen/sdk/core/expressions/tuple_type.py +++ b/src/codegen/sdk/core/expressions/tuple_type.py @@ -9,7 +9,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.symbol_groups.collection import Collection from codegen.sdk.extensions.autocommit import reader -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/expressions/type.py b/src/codegen/sdk/core/expressions/type.py index d5bea5a62..6be77c392 100644 --- a/src/codegen/sdk/core/expressions/type.py +++ b/src/codegen/sdk/core/expressions/type.py @@ -9,7 +9,7 @@ from codegen.sdk.core.interfaces.chainable import Chainable from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.extensions.autocommit import reader -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.interfaces.importable import Importable diff --git a/src/codegen/sdk/core/expressions/unary_expression.py b/src/codegen/sdk/core/expressions/unary_expression.py index 238639976..fcbd28171 100644 --- a/src/codegen/sdk/core/expressions/unary_expression.py +++ b/src/codegen/sdk/core/expressions/unary_expression.py @@ -12,7 +12,7 @@ from codegen.sdk.extensions.autocommit import commiter, reader from codegen.sdk.extensions.resolution import ResolutionStack from codegen.sdk.extensions.utils import TSNode -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/core/expressions/union_type.py b/src/codegen/sdk/core/expressions/union_type.py index 0523d08c1..8f53ebab4 100644 --- a/src/codegen/sdk/core/expressions/union_type.py +++ b/src/codegen/sdk/core/expressions/union_type.py @@ -9,7 +9,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.symbol_groups.collection import Collection from codegen.sdk.extensions.autocommit import reader -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/expressions/unpack.py b/src/codegen/sdk/core/expressions/unpack.py index 7791cb18d..e93c973a8 100644 --- a/src/codegen/sdk/core/expressions/unpack.py +++ b/src/codegen/sdk/core/expressions/unpack.py @@ -5,7 +5,7 @@ from codegen.sdk.core.interfaces.has_value import HasValue from codegen.sdk.core.interfaces.unwrappable import Unwrappable from codegen.sdk.core.interfaces.wrapper_expression import IWrapper -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/core/expressions/value.py b/src/codegen/sdk/core/expressions/value.py index 79057acdd..decd4954f 100644 --- a/src/codegen/sdk/core/expressions/value.py +++ b/src/codegen/sdk/core/expressions/value.py @@ -7,7 +7,7 @@ from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.core.interfaces.has_name import HasName from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/core/external_module.py b/src/codegen/sdk/core/external_module.py index 3343c7031..7b08f3a7b 100644 --- a/src/codegen/sdk/core/external_module.py +++ b/src/codegen/sdk/core/external_module.py @@ -14,7 +14,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.placeholder.placeholder_stub import StubPlaceholder from codegen.sdk.enums import ImportType, NodeType -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc from codegen.visualizations.enums import VizNode if TYPE_CHECKING: diff --git a/src/codegen/sdk/core/file.py b/src/codegen/sdk/core/file.py index 0cd9bd006..9023718ee 100644 --- a/src/codegen/sdk/core/file.py +++ b/src/codegen/sdk/core/file.py @@ -37,7 +37,7 @@ from codegen.sdk.topological_sort import pseudo_topological_sort from codegen.sdk.tree_sitter_parser import get_parser_by_filepath_or_extension, parse_file from codegen.sdk.typescript.function import TSFunction -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc from codegen.visualizations.enums import VizNode logger = logging.getLogger(__name__) diff --git a/src/codegen/sdk/core/function.py b/src/codegen/sdk/core/function.py index 625cf7031..c56e3b2b1 100644 --- a/src/codegen/sdk/core/function.py +++ b/src/codegen/sdk/core/function.py @@ -22,7 +22,7 @@ from codegen.sdk.enums import SymbolType from codegen.sdk.extensions.sort import sort_editables from codegen.sdk.extensions.utils import cached_property -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc from codegen.visualizations.enums import VizNode if TYPE_CHECKING: diff --git a/src/codegen/sdk/core/import_resolution.py b/src/codegen/sdk/core/import_resolution.py index aa0fd37c7..4ac16d799 100644 --- a/src/codegen/sdk/core/import_resolution.py +++ b/src/codegen/sdk/core/import_resolution.py @@ -26,7 +26,7 @@ from codegen.sdk.enums import EdgeType, ImportType, NodeType from codegen.sdk.extensions.utils import cached_property from codegen.sdk.output.constants import ANGULAR_STYLE -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc from codegen.visualizations.enums import VizNode if TYPE_CHECKING: diff --git a/src/codegen/sdk/core/interface.py b/src/codegen/sdk/core/interface.py index 4b45e294b..1d7351cdd 100644 --- a/src/codegen/sdk/core/interface.py +++ b/src/codegen/sdk/core/interface.py @@ -14,7 +14,7 @@ from codegen.sdk.core.symbol_groups.parents import Parents from codegen.sdk.enums import SymbolType from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import noapidoc +from codegen.utils.decorators.docs import noapidoc if TYPE_CHECKING: from codegen.sdk.core.class_definition import Class diff --git a/src/codegen/sdk/core/interfaces/callable.py b/src/codegen/sdk/core/interfaces/callable.py index 8c59e7bce..adb6090cb 100644 --- a/src/codegen/sdk/core/interfaces/callable.py +++ b/src/codegen/sdk/core/interfaces/callable.py @@ -7,7 +7,7 @@ from codegen.sdk.core.interfaces.usable import Usable from codegen.sdk.core.placeholder.placeholder import Placeholder from codegen.sdk.core.symbol_group import SymbolGroup -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc if TYPE_CHECKING: from codegen.sdk.core.class_definition import Class diff --git a/src/codegen/sdk/core/interfaces/chainable.py b/src/codegen/sdk/core/interfaces/chainable.py index 29c48f18f..2dfc07ea2 100644 --- a/src/codegen/sdk/core/interfaces/chainable.py +++ b/src/codegen/sdk/core/interfaces/chainable.py @@ -5,7 +5,7 @@ from codegen.sdk.codebase.resolution_stack import ResolutionStack from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.extensions.utils import cached_property -from codegen.sdk.writer_decorators import noapidoc +from codegen.utils.decorators.docs import noapidoc if TYPE_CHECKING: from codegen.sdk.core.interfaces.has_attribute import HasAttribute diff --git a/src/codegen/sdk/core/interfaces/editable.py b/src/codegen/sdk/core/interfaces/editable.py index 4c2f7a328..b37302462 100644 --- a/src/codegen/sdk/core/interfaces/editable.py +++ b/src/codegen/sdk/core/interfaces/editable.py @@ -30,7 +30,7 @@ from codegen.sdk.output.jsonable import JSONable from codegen.sdk.output.utils import style_editable from codegen.sdk.utils import descendant_for_byte_range, find_all_descendants, find_first_ancestor, find_index, truncate_line -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/interfaces/exportable.py b/src/codegen/sdk/core/interfaces/exportable.py index 8f58f73a3..5dcf23585 100644 --- a/src/codegen/sdk/core/interfaces/exportable.py +++ b/src/codegen/sdk/core/interfaces/exportable.py @@ -7,7 +7,7 @@ from codegen.sdk.core.autocommit import reader from codegen.sdk.core.interfaces.usable import Usable from codegen.sdk.enums import EdgeType, ImportType, NodeType -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc if TYPE_CHECKING: from codegen.sdk.core.export import Export diff --git a/src/codegen/sdk/core/interfaces/has_block.py b/src/codegen/sdk/core/interfaces/has_block.py index c186e13a5..bc9f1828f 100644 --- a/src/codegen/sdk/core/interfaces/has_block.py +++ b/src/codegen/sdk/core/interfaces/has_block.py @@ -13,7 +13,7 @@ from codegen.sdk.core.statements.comment import Comment from codegen.sdk.core.symbol_groups.comment_group import CommentGroup from codegen.sdk.extensions.sort import sort_editables -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc TCodeBlock = TypeVar("TCodeBlock", bound="CodeBlock") TDecorator = TypeVar("TDecorator", bound="Decorator") diff --git a/src/codegen/sdk/core/interfaces/has_name.py b/src/codegen/sdk/core/interfaces/has_name.py index ea14501fc..7ac05ba0f 100644 --- a/src/codegen/sdk/core/interfaces/has_name.py +++ b/src/codegen/sdk/core/interfaces/has_name.py @@ -5,7 +5,7 @@ from codegen.sdk.core.expressions.chained_attribute import ChainedAttribute from codegen.sdk.core.expressions.defined_name import DefinedName from codegen.sdk.core.expressions.name import Name -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc @apidoc diff --git a/src/codegen/sdk/core/interfaces/has_value.py b/src/codegen/sdk/core/interfaces/has_value.py index 89d82be79..299313c78 100644 --- a/src/codegen/sdk/core/interfaces/has_value.py +++ b/src/codegen/sdk/core/interfaces/has_value.py @@ -1,6 +1,6 @@ from codegen.sdk.core.autocommit import reader, writer from codegen.sdk.core.expressions.expression import Expression -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc @apidoc diff --git a/src/codegen/sdk/core/interfaces/importable.py b/src/codegen/sdk/core/interfaces/importable.py index 67f685002..9a01b7f58 100644 --- a/src/codegen/sdk/core/interfaces/importable.py +++ b/src/codegen/sdk/core/interfaces/importable.py @@ -11,7 +11,7 @@ from codegen.sdk.enums import EdgeType from codegen.sdk.extensions.autocommit import commiter from codegen.sdk.extensions.sort import sort_editables -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/interfaces/resolvable.py b/src/codegen/sdk/core/interfaces/resolvable.py index 58133dfa8..514253e88 100644 --- a/src/codegen/sdk/core/interfaces/resolvable.py +++ b/src/codegen/sdk/core/interfaces/resolvable.py @@ -6,7 +6,7 @@ from codegen.sdk.core.autocommit import writer from codegen.sdk.core.interfaces.chainable import Chainable from codegen.sdk.core.interfaces.editable import Editable -from codegen.sdk.writer_decorators import noapidoc +from codegen.utils.decorators.docs import noapidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/core/interfaces/supports_generic.py b/src/codegen/sdk/core/interfaces/supports_generic.py index 73f9342f9..9d5c0b4a3 100644 --- a/src/codegen/sdk/core/interfaces/supports_generic.py +++ b/src/codegen/sdk/core/interfaces/supports_generic.py @@ -6,7 +6,7 @@ from codegen.sdk.core.symbol import Symbol from codegen.sdk.core.symbol_groups.type_parameters import TypeParameters from codegen.sdk.extensions.utils import cached_property -from codegen.sdk.writer_decorators import noapidoc +from codegen.utils.decorators.docs import noapidoc if TYPE_CHECKING: from codegen.sdk.core.expressions import Type diff --git a/src/codegen/sdk/core/interfaces/typeable.py b/src/codegen/sdk/core/interfaces/typeable.py index b5315d8f3..bdad7472b 100644 --- a/src/codegen/sdk/core/interfaces/typeable.py +++ b/src/codegen/sdk/core/interfaces/typeable.py @@ -8,7 +8,7 @@ from codegen.sdk.core.interfaces.chainable import Chainable from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.core.placeholder.placeholder_type import TypePlaceholder -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.expressions.type import Type diff --git a/src/codegen/sdk/core/interfaces/unwrappable.py b/src/codegen/sdk/core/interfaces/unwrappable.py index 35f6c9245..00165a142 100644 --- a/src/codegen/sdk/core/interfaces/unwrappable.py +++ b/src/codegen/sdk/core/interfaces/unwrappable.py @@ -5,7 +5,7 @@ from codegen.sdk.core.expressions import Expression from codegen.sdk.core.interfaces.editable import Editable -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc Parent = TypeVar("Parent", bound=Editable) diff --git a/src/codegen/sdk/core/interfaces/usable.py b/src/codegen/sdk/core/interfaces/usable.py index 42e2298e2..a009c7ddf 100644 --- a/src/codegen/sdk/core/interfaces/usable.py +++ b/src/codegen/sdk/core/interfaces/usable.py @@ -8,7 +8,7 @@ from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.core.interfaces.importable import Importable from codegen.sdk.enums import EdgeType -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc if TYPE_CHECKING: from codegen.sdk.core.export import Export diff --git a/src/codegen/sdk/core/interfaces/wrapper_expression.py b/src/codegen/sdk/core/interfaces/wrapper_expression.py index 33586da15..49f556fbc 100644 --- a/src/codegen/sdk/core/interfaces/wrapper_expression.py +++ b/src/codegen/sdk/core/interfaces/wrapper_expression.py @@ -10,7 +10,7 @@ from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.core.interfaces.has_name import HasName from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import noapidoc +from codegen.utils.decorators.docs import noapidoc if TYPE_CHECKING: from codegen.sdk.core.interfaces.importable import Importable diff --git a/src/codegen/sdk/core/placeholder/placeholder.py b/src/codegen/sdk/core/placeholder/placeholder.py index 06426af82..d5d132e3f 100644 --- a/src/codegen/sdk/core/placeholder/placeholder.py +++ b/src/codegen/sdk/core/placeholder/placeholder.py @@ -2,7 +2,7 @@ from typing import TYPE_CHECKING, Generic, Literal, TypeVar from codegen.sdk.core.autocommit import repr_func -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc if TYPE_CHECKING: from codegen.sdk.core.interfaces.editable import Editable diff --git a/src/codegen/sdk/core/placeholder/placeholder_stub.py b/src/codegen/sdk/core/placeholder/placeholder_stub.py index 9f436d421..37cedaa9b 100644 --- a/src/codegen/sdk/core/placeholder/placeholder_stub.py +++ b/src/codegen/sdk/core/placeholder/placeholder_stub.py @@ -2,7 +2,7 @@ from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.core.placeholder.placeholder import Placeholder -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/core/placeholder/placeholder_type.py b/src/codegen/sdk/core/placeholder/placeholder_type.py index 74b8c3597..855161766 100644 --- a/src/codegen/sdk/core/placeholder/placeholder_type.py +++ b/src/codegen/sdk/core/placeholder/placeholder_type.py @@ -2,7 +2,7 @@ from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.core.placeholder.placeholder import Placeholder -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/core/statements/assignment_statement.py b/src/codegen/sdk/core/statements/assignment_statement.py index 7c4e84b4a..8287bd962 100644 --- a/src/codegen/sdk/core/statements/assignment_statement.py +++ b/src/codegen/sdk/core/statements/assignment_statement.py @@ -17,7 +17,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.statements.statement import Statement, StatementType from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.detached_symbols.code_block import CodeBlock diff --git a/src/codegen/sdk/core/statements/attribute.py b/src/codegen/sdk/core/statements/attribute.py index d81d91a0b..d899f797c 100644 --- a/src/codegen/sdk/core/statements/attribute.py +++ b/src/codegen/sdk/core/statements/attribute.py @@ -18,7 +18,7 @@ from codegen.sdk.core.statements.statement import StatementType from codegen.sdk.extensions.autocommit import reader from codegen.sdk.extensions.resolution import ResolutionStack -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.detached_symbols.code_block import CodeBlock diff --git a/src/codegen/sdk/core/statements/block_statement.py b/src/codegen/sdk/core/statements/block_statement.py index 872c9a2de..9e024f6cf 100644 --- a/src/codegen/sdk/core/statements/block_statement.py +++ b/src/codegen/sdk/core/statements/block_statement.py @@ -13,7 +13,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.statements.statement import Statement from codegen.sdk.extensions.autocommit import commiter, reader -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/statements/catch_statement.py b/src/codegen/sdk/core/statements/catch_statement.py index 28ea0314e..2481881f8 100644 --- a/src/codegen/sdk/core/statements/catch_statement.py +++ b/src/codegen/sdk/core/statements/catch_statement.py @@ -7,7 +7,7 @@ from codegen.sdk.core.interfaces.has_name import HasName from codegen.sdk.core.statements.block_statement import BlockStatement from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.detached_symbols.code_block import CodeBlock diff --git a/src/codegen/sdk/core/statements/comment.py b/src/codegen/sdk/core/statements/comment.py index 6c24f33de..3b1771141 100644 --- a/src/codegen/sdk/core/statements/comment.py +++ b/src/codegen/sdk/core/statements/comment.py @@ -10,7 +10,7 @@ from codegen.sdk.core.interfaces.has_name import HasName from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.statements.statement import Statement, StatementType -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.detached_symbols.code_block import CodeBlock diff --git a/src/codegen/sdk/core/statements/export_statement.py b/src/codegen/sdk/core/statements/export_statement.py index 014286c12..d9dc1c995 100644 --- a/src/codegen/sdk/core/statements/export_statement.py +++ b/src/codegen/sdk/core/statements/export_statement.py @@ -16,7 +16,7 @@ from codegen.sdk.typescript.export import TSExport from codegen.sdk.typescript.statements.import_statement import TSImportStatement from codegen.sdk.utils import find_first_ancestor -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/core/statements/expression_statement.py b/src/codegen/sdk/core/statements/expression_statement.py index eec83ce13..8b4ec56cb 100644 --- a/src/codegen/sdk/core/statements/expression_statement.py +++ b/src/codegen/sdk/core/statements/expression_statement.py @@ -13,7 +13,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.statements.statement import Statement, StatementType from codegen.sdk.extensions.autocommit import commiter, reader -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.detached_symbols.code_block import CodeBlock diff --git a/src/codegen/sdk/core/statements/for_loop_statement.py b/src/codegen/sdk/core/statements/for_loop_statement.py index 87736bd6c..4781c9e9e 100644 --- a/src/codegen/sdk/core/statements/for_loop_statement.py +++ b/src/codegen/sdk/core/statements/for_loop_statement.py @@ -11,7 +11,7 @@ from codegen.sdk.core.statements.statement import StatementType from codegen.sdk.core.symbol import Symbol from codegen.sdk.core.symbol_groups.collection import Collection -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc if TYPE_CHECKING: from codegen.sdk.core.detached_symbols.code_block import CodeBlock diff --git a/src/codegen/sdk/core/statements/if_block_statement.py b/src/codegen/sdk/core/statements/if_block_statement.py index e21564601..6ebbb9b98 100644 --- a/src/codegen/sdk/core/statements/if_block_statement.py +++ b/src/codegen/sdk/core/statements/if_block_statement.py @@ -15,7 +15,7 @@ from codegen.sdk.core.interfaces.importable import Importable from codegen.sdk.core.statements.statement import Statement, StatementType from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.detached_symbols.code_block import CodeBlock diff --git a/src/codegen/sdk/core/statements/import_statement.py b/src/codegen/sdk/core/statements/import_statement.py index ba7a4d072..315e42043 100644 --- a/src/codegen/sdk/core/statements/import_statement.py +++ b/src/codegen/sdk/core/statements/import_statement.py @@ -11,7 +11,7 @@ from codegen.sdk.core.statements.statement import Statement from codegen.sdk.core.symbol_groups.collection import Collection from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.detached_symbols.code_block import CodeBlock diff --git a/src/codegen/sdk/core/statements/raise_statement.py b/src/codegen/sdk/core/statements/raise_statement.py index 2b736a5db..a7d014b37 100644 --- a/src/codegen/sdk/core/statements/raise_statement.py +++ b/src/codegen/sdk/core/statements/raise_statement.py @@ -11,7 +11,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.statements.statement import Statement, StatementType from codegen.sdk.extensions.autocommit import commiter, reader -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/statements/return_statement.py b/src/codegen/sdk/core/statements/return_statement.py index 8351752a1..7e9a51705 100644 --- a/src/codegen/sdk/core/statements/return_statement.py +++ b/src/codegen/sdk/core/statements/return_statement.py @@ -11,7 +11,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.statements.statement import Statement, StatementType from codegen.sdk.extensions.autocommit import commiter, reader -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/statements/statement.py b/src/codegen/sdk/core/statements/statement.py index 3602f35e9..f9faa0e15 100644 --- a/src/codegen/sdk/core/statements/statement.py +++ b/src/codegen/sdk/core/statements/statement.py @@ -15,7 +15,7 @@ from codegen.sdk.extensions.autocommit import commiter from codegen.sdk.output.constants import ANGULAR_STYLE from codegen.sdk.utils import find_all_descendants -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/statements/switch_case.py b/src/codegen/sdk/core/statements/switch_case.py index 12c140726..57d00ca48 100644 --- a/src/codegen/sdk/core/statements/switch_case.py +++ b/src/codegen/sdk/core/statements/switch_case.py @@ -7,7 +7,7 @@ from codegen.sdk.core.interfaces.has_name import HasName from codegen.sdk.core.statements.block_statement import BlockStatement from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.assignment import Assignment diff --git a/src/codegen/sdk/core/statements/switch_statement.py b/src/codegen/sdk/core/statements/switch_statement.py index 091a3b60c..e35123fa2 100644 --- a/src/codegen/sdk/core/statements/switch_statement.py +++ b/src/codegen/sdk/core/statements/switch_statement.py @@ -10,7 +10,7 @@ from codegen.sdk.core.statements.statement import Statement, StatementType from codegen.sdk.core.statements.switch_case import SwitchCase from codegen.sdk.extensions.autocommit import commiter, reader -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.detached_symbols.code_block import CodeBlock diff --git a/src/codegen/sdk/core/statements/symbol_statement.py b/src/codegen/sdk/core/statements/symbol_statement.py index 91cae684e..ca14647b8 100644 --- a/src/codegen/sdk/core/statements/symbol_statement.py +++ b/src/codegen/sdk/core/statements/symbol_statement.py @@ -10,7 +10,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.statements.statement import Statement, StatementType from codegen.sdk.extensions.autocommit import reader -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/statements/try_catch_statement.py b/src/codegen/sdk/core/statements/try_catch_statement.py index 153dc9fda..f62e27208 100644 --- a/src/codegen/sdk/core/statements/try_catch_statement.py +++ b/src/codegen/sdk/core/statements/try_catch_statement.py @@ -6,7 +6,7 @@ from codegen.sdk.core.interfaces.has_block import HasBlock from codegen.sdk.core.statements.block_statement import BlockStatement from codegen.sdk.core.statements.statement import StatementType -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc if TYPE_CHECKING: from codegen.sdk.core.detached_symbols.code_block import CodeBlock diff --git a/src/codegen/sdk/core/statements/while_statement.py b/src/codegen/sdk/core/statements/while_statement.py index 7023ac692..d131c5b3e 100644 --- a/src/codegen/sdk/core/statements/while_statement.py +++ b/src/codegen/sdk/core/statements/while_statement.py @@ -14,7 +14,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.statements.statement import Statement, StatementType from codegen.sdk.extensions.autocommit import commiter, reader -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/symbol.py b/src/codegen/sdk/core/symbol.py index 636e1cf09..6c320ee64 100644 --- a/src/codegen/sdk/core/symbol.py +++ b/src/codegen/sdk/core/symbol.py @@ -24,7 +24,7 @@ from codegen.sdk.enums import ImportType, NodeType, SymbolType from codegen.sdk.extensions.sort import sort_editables from codegen.sdk.output.constants import ANGULAR_STYLE -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.core.detached_symbols.code_block import CodeBlock diff --git a/src/codegen/sdk/core/symbol_group.py b/src/codegen/sdk/core/symbol_group.py index 7572c6983..df1f5f3dd 100644 --- a/src/codegen/sdk/core/symbol_group.py +++ b/src/codegen/sdk/core/symbol_group.py @@ -12,7 +12,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.extensions.autocommit import commiter from codegen.sdk.output.ast import AST -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/symbol_groups/collection.py b/src/codegen/sdk/core/symbol_groups/collection.py index b08f2de28..a09d793f1 100644 --- a/src/codegen/sdk/core/symbol_groups/collection.py +++ b/src/codegen/sdk/core/symbol_groups/collection.py @@ -9,7 +9,7 @@ from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.symbol_group import SymbolGroup -from codegen.sdk.writer_decorators import noapidoc +from codegen.utils.decorators.docs import noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/symbol_groups/comment_group.py b/src/codegen/sdk/core/symbol_groups/comment_group.py index 1c11cf999..eb026923f 100644 --- a/src/codegen/sdk/core/symbol_groups/comment_group.py +++ b/src/codegen/sdk/core/symbol_groups/comment_group.py @@ -7,7 +7,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.statements.comment import Comment from codegen.sdk.core.symbol_group import SymbolGroup -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc Parent = TypeVar("Parent") diff --git a/src/codegen/sdk/core/symbol_groups/dict.py b/src/codegen/sdk/core/symbol_groups/dict.py index 9d55f8667..d8c5f84d9 100644 --- a/src/codegen/sdk/core/symbol_groups/dict.py +++ b/src/codegen/sdk/core/symbol_groups/dict.py @@ -14,7 +14,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.symbol_groups.collection import Collection from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/symbol_groups/expression_group.py b/src/codegen/sdk/core/symbol_groups/expression_group.py index d8b7d7d57..8e3b52dd4 100644 --- a/src/codegen/sdk/core/symbol_groups/expression_group.py +++ b/src/codegen/sdk/core/symbol_groups/expression_group.py @@ -6,7 +6,7 @@ from codegen.sdk.core.expressions import Expression from codegen.sdk.core.symbol_group import SymbolGroup from codegen.sdk.extensions.autocommit import reader -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc Parent = TypeVar("Parent") diff --git a/src/codegen/sdk/core/symbol_groups/list.py b/src/codegen/sdk/core/symbol_groups/list.py index 4fc6c3729..eb079afd8 100644 --- a/src/codegen/sdk/core/symbol_groups/list.py +++ b/src/codegen/sdk/core/symbol_groups/list.py @@ -7,7 +7,7 @@ from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.symbol_groups.collection import Collection -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/symbol_groups/multi_line_collection.py b/src/codegen/sdk/core/symbol_groups/multi_line_collection.py index 1bc58fec8..26e0f5047 100644 --- a/src/codegen/sdk/core/symbol_groups/multi_line_collection.py +++ b/src/codegen/sdk/core/symbol_groups/multi_line_collection.py @@ -7,7 +7,7 @@ from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.symbol_groups.collection import Collection -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/symbol_groups/tuple.py b/src/codegen/sdk/core/symbol_groups/tuple.py index ad981982d..7af7cf2d3 100644 --- a/src/codegen/sdk/core/symbol_groups/tuple.py +++ b/src/codegen/sdk/core/symbol_groups/tuple.py @@ -7,7 +7,7 @@ from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.symbol_groups.collection import Collection -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/core/type_alias.py b/src/codegen/sdk/core/type_alias.py index 702b1b165..9369241c4 100644 --- a/src/codegen/sdk/core/type_alias.py +++ b/src/codegen/sdk/core/type_alias.py @@ -16,7 +16,7 @@ from codegen.sdk.core.statements.attribute import Attribute from codegen.sdk.core.statements.statement import Statement from codegen.sdk.enums import SymbolType -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/output/inspect.py b/src/codegen/sdk/output/inspect.py index d9f4a83cf..e33692f66 100644 --- a/src/codegen/sdk/output/inspect.py +++ b/src/codegen/sdk/output/inspect.py @@ -2,7 +2,7 @@ from collections.abc import Callable from inspect import Parameter -from codegen.sdk.writer_decorators import DocumentedObject, no_apidoc_signatures +from codegen.utils.decorators.docs import DocumentedObject, no_apidoc_signatures def only_default_args(method: Callable) -> bool: diff --git a/src/codegen/sdk/output/jsonable.py b/src/codegen/sdk/output/jsonable.py index 2d82ba8cc..33656895e 100644 --- a/src/codegen/sdk/output/jsonable.py +++ b/src/codegen/sdk/output/jsonable.py @@ -9,7 +9,7 @@ from codegen.sdk.output.placeholder import Placeholder from codegen.sdk.output.utils import safe_getattr from codegen.sdk.types import JSON -from codegen.sdk.writer_decorators import noapidoc +from codegen.utils.decorators.docs import noapidoc BLACKLIST = ["json", "G", "viz", "autocommit_cache", "ts_node", "symbol_usages", "usages"] diff --git a/src/codegen/sdk/python/assignment.py b/src/codegen/sdk/python/assignment.py index aff2142a7..2358a6455 100644 --- a/src/codegen/sdk/python/assignment.py +++ b/src/codegen/sdk/python/assignment.py @@ -11,7 +11,7 @@ from codegen.sdk.extensions.autocommit import reader from codegen.sdk.python.symbol import PySymbol from codegen.sdk.python.symbol_groups.comment_group import PyCommentGroup -from codegen.sdk.writer_decorators import noapidoc, py_apidoc +from codegen.utils.decorators.docs import noapidoc, py_apidoc if TYPE_CHECKING: from codegen.sdk.python.statements.assignment_statement import PyAssignmentStatement diff --git a/src/codegen/sdk/python/class_definition.py b/src/codegen/sdk/python/class_definition.py index e16a2cfe9..fc7ae1cdd 100644 --- a/src/codegen/sdk/python/class_definition.py +++ b/src/codegen/sdk/python/class_definition.py @@ -20,7 +20,7 @@ from codegen.sdk.python.function import PyFunction from codegen.sdk.python.interfaces.has_block import PyHasBlock from codegen.sdk.python.symbol import PySymbol -from codegen.sdk.writer_decorators import noapidoc, py_apidoc +from codegen.utils.decorators.docs import noapidoc, py_apidoc @py_apidoc diff --git a/src/codegen/sdk/python/detached_symbols/code_block.py b/src/codegen/sdk/python/detached_symbols/code_block.py index ac2c14263..fb33f79c0 100644 --- a/src/codegen/sdk/python/detached_symbols/code_block.py +++ b/src/codegen/sdk/python/detached_symbols/code_block.py @@ -8,7 +8,7 @@ from codegen.sdk.core.statements.import_statement import ImportStatement from codegen.sdk.core.statements.statement import Statement, StatementType from codegen.sdk.core.symbol_groups.multi_line_collection import MultiLineCollection -from codegen.sdk.writer_decorators import noapidoc, py_apidoc +from codegen.utils.decorators.docs import noapidoc, py_apidoc if TYPE_CHECKING: from codegen.sdk.python.interfaces.has_block import PyHasBlock diff --git a/src/codegen/sdk/python/detached_symbols/decorator.py b/src/codegen/sdk/python/detached_symbols/decorator.py index 5931b7010..72bf7695f 100644 --- a/src/codegen/sdk/python/detached_symbols/decorator.py +++ b/src/codegen/sdk/python/detached_symbols/decorator.py @@ -5,7 +5,7 @@ from codegen.sdk.core.autocommit import reader from codegen.sdk.core.detached_symbols.decorator import Decorator from codegen.sdk.core.detached_symbols.function_call import FunctionCall -from codegen.sdk.writer_decorators import py_apidoc +from codegen.utils.decorators.docs import py_apidoc @py_apidoc diff --git a/src/codegen/sdk/python/detached_symbols/parameter.py b/src/codegen/sdk/python/detached_symbols/parameter.py index e6196e582..14c29fc5b 100644 --- a/src/codegen/sdk/python/detached_symbols/parameter.py +++ b/src/codegen/sdk/python/detached_symbols/parameter.py @@ -4,7 +4,7 @@ from codegen.sdk.core.detached_symbols.parameter import Parameter from codegen.sdk.core.symbol_groups.collection import Collection from codegen.sdk.python.expressions.type import PyType -from codegen.sdk.writer_decorators import py_apidoc +from codegen.utils.decorators.docs import py_apidoc @py_apidoc diff --git a/src/codegen/sdk/python/expressions/chained_attribute.py b/src/codegen/sdk/python/expressions/chained_attribute.py index b615be6c2..adf8a3391 100644 --- a/src/codegen/sdk/python/expressions/chained_attribute.py +++ b/src/codegen/sdk/python/expressions/chained_attribute.py @@ -3,7 +3,7 @@ from codegen.sdk.core.expressions import Expression, Name from codegen.sdk.core.expressions.chained_attribute import ChainedAttribute from codegen.sdk.core.interfaces.editable import Editable -from codegen.sdk.writer_decorators import py_apidoc +from codegen.utils.decorators.docs import py_apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/python/expressions/conditional_expression.py b/src/codegen/sdk/python/expressions/conditional_expression.py index bb9e013d3..c7e63bb77 100644 --- a/src/codegen/sdk/python/expressions/conditional_expression.py +++ b/src/codegen/sdk/python/expressions/conditional_expression.py @@ -2,7 +2,7 @@ from codegen.sdk.core.expressions.ternary_expression import TernaryExpression from codegen.sdk.core.interfaces.editable import Editable -from codegen.sdk.writer_decorators import py_apidoc +from codegen.utils.decorators.docs import py_apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/python/expressions/generic_type.py b/src/codegen/sdk/python/expressions/generic_type.py index 661fb7327..74235c062 100644 --- a/src/codegen/sdk/python/expressions/generic_type.py +++ b/src/codegen/sdk/python/expressions/generic_type.py @@ -5,7 +5,7 @@ from codegen.sdk.core.expressions.generic_type import GenericType from codegen.sdk.core.symbol_groups.collection import Collection from codegen.sdk.python.expressions.named_type import PyNamedType -from codegen.sdk.writer_decorators import py_apidoc +from codegen.utils.decorators.docs import py_apidoc if TYPE_CHECKING: from codegen.sdk.python.expressions.type import PyType diff --git a/src/codegen/sdk/python/expressions/named_type.py b/src/codegen/sdk/python/expressions/named_type.py index ca7bbecd3..97a597c7a 100644 --- a/src/codegen/sdk/python/expressions/named_type.py +++ b/src/codegen/sdk/python/expressions/named_type.py @@ -3,7 +3,7 @@ from tree_sitter import Node as TSNode from codegen.sdk.core.expressions.named_type import NamedType -from codegen.sdk.writer_decorators import py_apidoc +from codegen.utils.decorators.docs import py_apidoc Parent = TypeVar("Parent") diff --git a/src/codegen/sdk/python/expressions/string.py b/src/codegen/sdk/python/expressions/string.py index 5f781a7b0..4869bad8b 100644 --- a/src/codegen/sdk/python/expressions/string.py +++ b/src/codegen/sdk/python/expressions/string.py @@ -4,7 +4,7 @@ from codegen.sdk.core.expressions import Expression, String from codegen.sdk.core.node_id_factory import NodeId -from codegen.sdk.writer_decorators import py_apidoc +from codegen.utils.decorators.docs import py_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/python/expressions/union_type.py b/src/codegen/sdk/python/expressions/union_type.py index f5869117e..ad19308cb 100644 --- a/src/codegen/sdk/python/expressions/union_type.py +++ b/src/codegen/sdk/python/expressions/union_type.py @@ -1,7 +1,7 @@ from typing import TYPE_CHECKING, Generic, TypeVar from codegen.sdk.core.expressions.union_type import UnionType -from codegen.sdk.writer_decorators import py_apidoc +from codegen.utils.decorators.docs import py_apidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/python/file.py b/src/codegen/sdk/python/file.py index 6f739c947..c2b71e6c4 100644 --- a/src/codegen/sdk/python/file.py +++ b/src/codegen/sdk/python/file.py @@ -16,7 +16,7 @@ from codegen.sdk.python.statements.attribute import PyAttribute from codegen.sdk.python.statements.import_statement import PyImportStatement from codegen.sdk.python.symbol import PySymbol -from codegen.sdk.writer_decorators import noapidoc, py_apidoc +from codegen.utils.decorators.docs import noapidoc, py_apidoc @py_apidoc diff --git a/src/codegen/sdk/python/function.py b/src/codegen/sdk/python/function.py index 4c81a9705..422586112 100644 --- a/src/codegen/sdk/python/function.py +++ b/src/codegen/sdk/python/function.py @@ -23,7 +23,7 @@ from codegen.sdk.python.interfaces.has_block import PyHasBlock from codegen.sdk.python.placeholder.placeholder_return_type import PyReturnTypePlaceholder from codegen.sdk.python.symbol import PySymbol -from codegen.sdk.writer_decorators import noapidoc, py_apidoc +from codegen.utils.decorators.docs import noapidoc, py_apidoc logger = logging.getLogger(__name__) diff --git a/src/codegen/sdk/python/import_resolution.py b/src/codegen/sdk/python/import_resolution.py index f2110ba9d..4e1c6dc49 100644 --- a/src/codegen/sdk/python/import_resolution.py +++ b/src/codegen/sdk/python/import_resolution.py @@ -14,7 +14,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.statements.import_statement import ImportStatement from codegen.sdk.enums import ImportType, NodeType -from codegen.sdk.writer_decorators import noapidoc, py_apidoc +from codegen.utils.decorators.docs import noapidoc, py_apidoc if TYPE_CHECKING: from codegen.sdk.python.file import PyFile diff --git a/src/codegen/sdk/python/interfaces/has_block.py b/src/codegen/sdk/python/interfaces/has_block.py index bc44a6f21..aafb5714c 100644 --- a/src/codegen/sdk/python/interfaces/has_block.py +++ b/src/codegen/sdk/python/interfaces/has_block.py @@ -6,7 +6,7 @@ from codegen.sdk.python.detached_symbols.decorator import PyDecorator from codegen.sdk.python.statements.comment import PyComment, PyCommentType from codegen.sdk.python.symbol_groups.comment_group import PyCommentGroup -from codegen.sdk.writer_decorators import py_apidoc +from codegen.utils.decorators.docs import py_apidoc @py_apidoc diff --git a/src/codegen/sdk/python/placeholder/placeholder_return_type.py b/src/codegen/sdk/python/placeholder/placeholder_return_type.py index c7bd1da43..d09d4f2d6 100644 --- a/src/codegen/sdk/python/placeholder/placeholder_return_type.py +++ b/src/codegen/sdk/python/placeholder/placeholder_return_type.py @@ -2,7 +2,7 @@ from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.core.placeholder.placeholder import Placeholder -from codegen.sdk.writer_decorators import py_apidoc +from codegen.utils.decorators.docs import py_apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/python/statements/assignment_statement.py b/src/codegen/sdk/python/statements/assignment_statement.py index 7f277a9e3..430cd97f6 100644 --- a/src/codegen/sdk/python/statements/assignment_statement.py +++ b/src/codegen/sdk/python/statements/assignment_statement.py @@ -10,7 +10,7 @@ from codegen.sdk.core.statements.assignment_statement import AssignmentStatement from codegen.sdk.extensions.utils import find_all_descendants from codegen.sdk.python.assignment import PyAssignment -from codegen.sdk.writer_decorators import py_apidoc +from codegen.utils.decorators.docs import py_apidoc if TYPE_CHECKING: from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock diff --git a/src/codegen/sdk/python/statements/attribute.py b/src/codegen/sdk/python/statements/attribute.py index 4e3fd52d4..91bff3e25 100644 --- a/src/codegen/sdk/python/statements/attribute.py +++ b/src/codegen/sdk/python/statements/attribute.py @@ -9,7 +9,7 @@ from codegen.sdk.exceptions import APINotApplicableForLanguageError from codegen.sdk.python.assignment import PyAssignment from codegen.sdk.python.statements.assignment_statement import PyAssignmentStatement -from codegen.sdk.writer_decorators import noapidoc, py_apidoc +from codegen.utils.decorators.docs import noapidoc, py_apidoc if TYPE_CHECKING: from codegen.sdk.python.class_definition import PyClass diff --git a/src/codegen/sdk/python/statements/block_statement.py b/src/codegen/sdk/python/statements/block_statement.py index 0684be0ce..2dc0d21f7 100644 --- a/src/codegen/sdk/python/statements/block_statement.py +++ b/src/codegen/sdk/python/statements/block_statement.py @@ -6,7 +6,7 @@ from codegen.sdk.extensions.autocommit import reader from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock from codegen.sdk.python.interfaces.has_block import PyHasBlock -from codegen.sdk.writer_decorators import py_apidoc +from codegen.utils.decorators.docs import py_apidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/python/statements/break_statement.py b/src/codegen/sdk/python/statements/break_statement.py index abb4ec853..56fbd46f3 100644 --- a/src/codegen/sdk/python/statements/break_statement.py +++ b/src/codegen/sdk/python/statements/break_statement.py @@ -6,7 +6,7 @@ from codegen.sdk.core.interfaces.has_name import HasName from codegen.sdk.core.statements.statement import Statement, StatementType from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import noapidoc, py_apidoc +from codegen.utils.decorators.docs import noapidoc, py_apidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/python/statements/catch_statement.py b/src/codegen/sdk/python/statements/catch_statement.py index e089d7365..ba998c155 100644 --- a/src/codegen/sdk/python/statements/catch_statement.py +++ b/src/codegen/sdk/python/statements/catch_statement.py @@ -8,7 +8,7 @@ from codegen.sdk.core.statements.catch_statement import CatchStatement from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock from codegen.sdk.python.statements.block_statement import PyBlockStatement -from codegen.sdk.writer_decorators import py_apidoc +from codegen.utils.decorators.docs import py_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/python/statements/comment.py b/src/codegen/sdk/python/statements/comment.py index a044db8ac..06fb8c8d6 100644 --- a/src/codegen/sdk/python/statements/comment.py +++ b/src/codegen/sdk/python/statements/comment.py @@ -5,7 +5,7 @@ from codegen.sdk.core.autocommit import commiter, reader from codegen.sdk.core.statements.comment import Comment, lowest_indentation -from codegen.sdk.writer_decorators import noapidoc, py_apidoc +from codegen.utils.decorators.docs import noapidoc, py_apidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/python/statements/for_loop_statement.py b/src/codegen/sdk/python/statements/for_loop_statement.py index 7331c2771..c25404524 100644 --- a/src/codegen/sdk/python/statements/for_loop_statement.py +++ b/src/codegen/sdk/python/statements/for_loop_statement.py @@ -13,7 +13,7 @@ from codegen.sdk.core.statements.for_loop_statement import ForLoopStatement from codegen.sdk.extensions.autocommit import commiter, reader from codegen.sdk.python.statements.block_statement import PyBlockStatement -from codegen.sdk.writer_decorators import noapidoc, py_apidoc +from codegen.utils.decorators.docs import noapidoc, py_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/python/statements/if_block_statement.py b/src/codegen/sdk/python/statements/if_block_statement.py index b7cb62614..04efae6a5 100644 --- a/src/codegen/sdk/python/statements/if_block_statement.py +++ b/src/codegen/sdk/python/statements/if_block_statement.py @@ -9,7 +9,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.statements.if_block_statement import IfBlockStatement from codegen.sdk.core.statements.statement import StatementType -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc if TYPE_CHECKING: from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock diff --git a/src/codegen/sdk/python/statements/import_statement.py b/src/codegen/sdk/python/statements/import_statement.py index bcd9e3f28..d3ff601ee 100644 --- a/src/codegen/sdk/python/statements/import_statement.py +++ b/src/codegen/sdk/python/statements/import_statement.py @@ -10,7 +10,7 @@ from codegen.sdk.core.symbol_groups.collection import Collection from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock from codegen.sdk.python.import_resolution import PyImport -from codegen.sdk.writer_decorators import py_apidoc +from codegen.utils.decorators.docs import py_apidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/python/statements/match_case.py b/src/codegen/sdk/python/statements/match_case.py index 1a4861d2a..b2253f630 100644 --- a/src/codegen/sdk/python/statements/match_case.py +++ b/src/codegen/sdk/python/statements/match_case.py @@ -6,7 +6,7 @@ from codegen.sdk.core.statements.switch_case import SwitchCase from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock from codegen.sdk.python.statements.block_statement import PyBlockStatement -from codegen.sdk.writer_decorators import py_apidoc +from codegen.utils.decorators.docs import py_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/python/statements/match_statement.py b/src/codegen/sdk/python/statements/match_statement.py index b777f93ac..ff6dd70fd 100644 --- a/src/codegen/sdk/python/statements/match_statement.py +++ b/src/codegen/sdk/python/statements/match_statement.py @@ -8,7 +8,7 @@ from codegen.sdk.core.statements.switch_statement import SwitchStatement from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock from codegen.sdk.python.statements.match_case import PyMatchCase -from codegen.sdk.writer_decorators import py_apidoc +from codegen.utils.decorators.docs import py_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/python/statements/pass_statement.py b/src/codegen/sdk/python/statements/pass_statement.py index 8a31e650b..8e937ddf6 100644 --- a/src/codegen/sdk/python/statements/pass_statement.py +++ b/src/codegen/sdk/python/statements/pass_statement.py @@ -6,7 +6,7 @@ from codegen.sdk.core.interfaces.has_name import HasName from codegen.sdk.core.statements.statement import Statement, StatementType from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import noapidoc, py_apidoc +from codegen.utils.decorators.docs import noapidoc, py_apidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/python/statements/try_catch_statement.py b/src/codegen/sdk/python/statements/try_catch_statement.py index bc7459663..fe490b0fa 100644 --- a/src/codegen/sdk/python/statements/try_catch_statement.py +++ b/src/codegen/sdk/python/statements/try_catch_statement.py @@ -13,7 +13,7 @@ from codegen.sdk.python.detached_symbols.code_block import PyCodeBlock from codegen.sdk.python.statements.block_statement import PyBlockStatement from codegen.sdk.python.statements.catch_statement import PyCatchStatement -from codegen.sdk.writer_decorators import noapidoc, py_apidoc +from codegen.utils.decorators.docs import noapidoc, py_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/python/statements/while_statement.py b/src/codegen/sdk/python/statements/while_statement.py index d9e7ca9af..06a402d3c 100644 --- a/src/codegen/sdk/python/statements/while_statement.py +++ b/src/codegen/sdk/python/statements/while_statement.py @@ -14,7 +14,7 @@ from codegen.sdk.extensions.sort import sort_editables from codegen.sdk.python.interfaces.has_block import PyHasBlock from codegen.sdk.python.statements.if_block_statement import PyIfBlockStatement -from codegen.sdk.writer_decorators import noapidoc, py_apidoc +from codegen.utils.decorators.docs import noapidoc, py_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/python/statements/with_statement.py b/src/codegen/sdk/python/statements/with_statement.py index 6fb3d04c7..9c3303386 100644 --- a/src/codegen/sdk/python/statements/with_statement.py +++ b/src/codegen/sdk/python/statements/with_statement.py @@ -15,7 +15,7 @@ from codegen.sdk.extensions.autocommit import commiter from codegen.sdk.extensions.sort import sort_editables from codegen.sdk.python.interfaces.has_block import PyHasBlock -from codegen.sdk.writer_decorators import apidoc, noapidoc +from codegen.utils.decorators.docs import apidoc, noapidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/python/symbol.py b/src/codegen/sdk/python/symbol.py index a07317576..ce45f94b3 100644 --- a/src/codegen/sdk/python/symbol.py +++ b/src/codegen/sdk/python/symbol.py @@ -12,7 +12,7 @@ from codegen.sdk.enums import ImportType from codegen.sdk.python.statements.comment import PyComment, PyCommentType from codegen.sdk.python.symbol_groups.comment_group import PyCommentGroup -from codegen.sdk.writer_decorators import noapidoc, py_apidoc +from codegen.utils.decorators.docs import noapidoc, py_apidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/python/symbol_groups/comment_group.py b/src/codegen/sdk/python/symbol_groups/comment_group.py index 7be70b6a6..89aa605d2 100644 --- a/src/codegen/sdk/python/symbol_groups/comment_group.py +++ b/src/codegen/sdk/python/symbol_groups/comment_group.py @@ -11,7 +11,7 @@ from codegen.sdk.core.symbol_groups.comment_group import CommentGroup from codegen.sdk.enums import SymbolType from codegen.sdk.python.statements.comment import PyComment -from codegen.sdk.writer_decorators import noapidoc, py_apidoc +from codegen.utils.decorators.docs import noapidoc, py_apidoc if TYPE_CHECKING: from codegen.sdk.python.function import PyFunction diff --git a/src/codegen/sdk/typescript/assignment.py b/src/codegen/sdk/typescript/assignment.py index ca7b4bd69..533b0c2db 100644 --- a/src/codegen/sdk/typescript/assignment.py +++ b/src/codegen/sdk/typescript/assignment.py @@ -10,7 +10,7 @@ from codegen.sdk.core.expressions.multi_expression import MultiExpression from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.typescript.symbol import TSSymbol -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.typescript.statements.assignment_statement import TSAssignmentStatement diff --git a/src/codegen/sdk/typescript/class_definition.py b/src/codegen/sdk/typescript/class_definition.py index 88832178e..c2f7f5f22 100644 --- a/src/codegen/sdk/typescript/class_definition.py +++ b/src/codegen/sdk/typescript/class_definition.py @@ -24,7 +24,7 @@ from codegen.sdk.typescript.function import TSFunction from codegen.sdk.typescript.interfaces.has_block import TSHasBlock from codegen.sdk.typescript.symbol import TSSymbol -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc @ts_apidoc diff --git a/src/codegen/sdk/typescript/detached_symbols/code_block.py b/src/codegen/sdk/typescript/detached_symbols/code_block.py index 5d1592b33..1f56cf058 100644 --- a/src/codegen/sdk/typescript/detached_symbols/code_block.py +++ b/src/codegen/sdk/typescript/detached_symbols/code_block.py @@ -6,7 +6,7 @@ from codegen.sdk.core.statements.statement import Statement from codegen.sdk.core.symbol_groups.multi_line_collection import MultiLineCollection from codegen.sdk.extensions.utils import find_line_start_and_end_nodes -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.typescript.interfaces.has_block import TSHasBlock diff --git a/src/codegen/sdk/typescript/detached_symbols/decorator.py b/src/codegen/sdk/typescript/detached_symbols/decorator.py index 194eb1a45..4b4899513 100644 --- a/src/codegen/sdk/typescript/detached_symbols/decorator.py +++ b/src/codegen/sdk/typescript/detached_symbols/decorator.py @@ -5,7 +5,7 @@ from codegen.sdk.core.autocommit import reader from codegen.sdk.core.detached_symbols.decorator import Decorator from codegen.sdk.core.detached_symbols.function_call import FunctionCall -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc @ts_apidoc diff --git a/src/codegen/sdk/typescript/detached_symbols/jsx/element.py b/src/codegen/sdk/typescript/detached_symbols/jsx/element.py index 9619d67d6..5920b711e 100644 --- a/src/codegen/sdk/typescript/detached_symbols/jsx/element.py +++ b/src/codegen/sdk/typescript/detached_symbols/jsx/element.py @@ -17,7 +17,7 @@ from codegen.sdk.typescript.detached_symbols.jsx.expression import JSXExpression from codegen.sdk.typescript.detached_symbols.jsx.prop import JSXProp from codegen.sdk.utils import find_all_descendants -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/typescript/detached_symbols/jsx/expression.py b/src/codegen/sdk/typescript/detached_symbols/jsx/expression.py index fe10a3c2d..4b6ef6907 100644 --- a/src/codegen/sdk/typescript/detached_symbols/jsx/expression.py +++ b/src/codegen/sdk/typescript/detached_symbols/jsx/expression.py @@ -8,7 +8,7 @@ from codegen.sdk.core.interfaces.has_name import HasName from codegen.sdk.core.interfaces.unwrappable import Unwrappable from codegen.sdk.extensions.autocommit import commiter -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/typescript/detached_symbols/jsx/prop.py b/src/codegen/sdk/typescript/detached_symbols/jsx/prop.py index 48b91ca37..9386d4afc 100644 --- a/src/codegen/sdk/typescript/detached_symbols/jsx/prop.py +++ b/src/codegen/sdk/typescript/detached_symbols/jsx/prop.py @@ -10,7 +10,7 @@ from codegen.sdk.core.interfaces.has_value import HasValue from codegen.sdk.extensions.autocommit import commiter from codegen.sdk.typescript.detached_symbols.jsx.expression import JSXExpression -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.core.function import Function diff --git a/src/codegen/sdk/typescript/detached_symbols/parameter.py b/src/codegen/sdk/typescript/detached_symbols/parameter.py index b142d220b..880eab64d 100644 --- a/src/codegen/sdk/typescript/detached_symbols/parameter.py +++ b/src/codegen/sdk/typescript/detached_symbols/parameter.py @@ -16,7 +16,7 @@ from codegen.sdk.typescript.expressions.object_type import TSObjectType from codegen.sdk.typescript.expressions.type import TSType from codegen.sdk.typescript.symbol_groups.dict import TSPair -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.typescript.function import TSFunction diff --git a/src/codegen/sdk/typescript/enum_definition.py b/src/codegen/sdk/typescript/enum_definition.py index 26c332c92..c03de436f 100644 --- a/src/codegen/sdk/typescript/enum_definition.py +++ b/src/codegen/sdk/typescript/enum_definition.py @@ -18,7 +18,7 @@ from codegen.sdk.typescript.interfaces.has_block import TSHasBlock from codegen.sdk.typescript.statements.attribute import TSAttribute from codegen.sdk.typescript.symbol import TSSymbol -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock diff --git a/src/codegen/sdk/typescript/export.py b/src/codegen/sdk/typescript/export.py index 25e0b29d7..092a2a6fa 100644 --- a/src/codegen/sdk/typescript/export.py +++ b/src/codegen/sdk/typescript/export.py @@ -35,7 +35,7 @@ from codegen.sdk.typescript.symbol import TSSymbol from codegen.sdk.typescript.type_alias import TSTypeAlias from codegen.sdk.utils import find_all_descendants -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.core.statements.export_statement import ExportStatement diff --git a/src/codegen/sdk/typescript/expressions/array_type.py b/src/codegen/sdk/typescript/expressions/array_type.py index 953e4a476..01d9abca8 100644 --- a/src/codegen/sdk/typescript/expressions/array_type.py +++ b/src/codegen/sdk/typescript/expressions/array_type.py @@ -3,7 +3,7 @@ from tree_sitter import Node as TSNode from codegen.sdk.typescript.expressions.named_type import TSNamedType -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/typescript/expressions/chained_attribute.py b/src/codegen/sdk/typescript/expressions/chained_attribute.py index 3deb659c6..f5a5a7626 100644 --- a/src/codegen/sdk/typescript/expressions/chained_attribute.py +++ b/src/codegen/sdk/typescript/expressions/chained_attribute.py @@ -5,7 +5,7 @@ from codegen.sdk.core.expressions.chained_attribute import ChainedAttribute from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.extensions.autocommit import reader -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/typescript/expressions/conditional_type.py b/src/codegen/sdk/typescript/expressions/conditional_type.py index 0f98dce32..a7f649396 100644 --- a/src/codegen/sdk/typescript/expressions/conditional_type.py +++ b/src/codegen/sdk/typescript/expressions/conditional_type.py @@ -9,7 +9,7 @@ from codegen.sdk.core.expressions.type import Type from codegen.sdk.core.interfaces.importable import Importable from codegen.sdk.core.node_id_factory import NodeId -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/typescript/expressions/expression_type.py b/src/codegen/sdk/typescript/expressions/expression_type.py index 11a68711e..423d21ddc 100644 --- a/src/codegen/sdk/typescript/expressions/expression_type.py +++ b/src/codegen/sdk/typescript/expressions/expression_type.py @@ -7,7 +7,7 @@ from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.typescript.expressions.named_type import TSNamedType -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/typescript/expressions/function_type.py b/src/codegen/sdk/typescript/expressions/function_type.py index 262636c26..b9d17ec1d 100644 --- a/src/codegen/sdk/typescript/expressions/function_type.py +++ b/src/codegen/sdk/typescript/expressions/function_type.py @@ -12,7 +12,7 @@ from codegen.sdk.core.symbol_groups.collection import Collection from codegen.sdk.typescript.detached_symbols.parameter import TSParameter from codegen.sdk.typescript.placeholder.placeholder_return_type import TSReturnTypePlaceholder -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/typescript/expressions/generic_type.py b/src/codegen/sdk/typescript/expressions/generic_type.py index ae423227b..a9b6d52cc 100644 --- a/src/codegen/sdk/typescript/expressions/generic_type.py +++ b/src/codegen/sdk/typescript/expressions/generic_type.py @@ -5,7 +5,7 @@ from codegen.sdk.core.expressions.generic_type import GenericType from codegen.sdk.core.symbol_groups.collection import Collection from codegen.sdk.core.symbol_groups.dict import Dict -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/typescript/expressions/lookup_type.py b/src/codegen/sdk/typescript/expressions/lookup_type.py index 825f4ee95..d4161cc5c 100644 --- a/src/codegen/sdk/typescript/expressions/lookup_type.py +++ b/src/codegen/sdk/typescript/expressions/lookup_type.py @@ -10,7 +10,7 @@ from codegen.sdk.core.expressions.type import Type from codegen.sdk.core.interfaces.importable import Importable from codegen.sdk.core.node_id_factory import NodeId -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/typescript/expressions/named_type.py b/src/codegen/sdk/typescript/expressions/named_type.py index ce955d268..8559223f0 100644 --- a/src/codegen/sdk/typescript/expressions/named_type.py +++ b/src/codegen/sdk/typescript/expressions/named_type.py @@ -3,7 +3,7 @@ from tree_sitter import Node as TSNode from codegen.sdk.core.expressions.named_type import NamedType -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/typescript/expressions/object_type.py b/src/codegen/sdk/typescript/expressions/object_type.py index 188bc581f..b81de92a4 100644 --- a/src/codegen/sdk/typescript/expressions/object_type.py +++ b/src/codegen/sdk/typescript/expressions/object_type.py @@ -9,7 +9,7 @@ from codegen.sdk.core.interfaces.importable import Importable from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.typescript.symbol_groups.dict import TSDict, TSPair -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/typescript/expressions/query_type.py b/src/codegen/sdk/typescript/expressions/query_type.py index e9352d4ec..f1fd96bfe 100644 --- a/src/codegen/sdk/typescript/expressions/query_type.py +++ b/src/codegen/sdk/typescript/expressions/query_type.py @@ -9,7 +9,7 @@ from codegen.sdk.core.expressions.type import Type from codegen.sdk.core.interfaces.importable import Importable from codegen.sdk.core.node_id_factory import NodeId -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/typescript/expressions/readonly_type.py b/src/codegen/sdk/typescript/expressions/readonly_type.py index 0e3d8f134..8ed84bc77 100644 --- a/src/codegen/sdk/typescript/expressions/readonly_type.py +++ b/src/codegen/sdk/typescript/expressions/readonly_type.py @@ -9,7 +9,7 @@ from codegen.sdk.core.expressions.type import Type from codegen.sdk.core.interfaces.importable import Importable from codegen.sdk.core.node_id_factory import NodeId -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/typescript/expressions/string.py b/src/codegen/sdk/typescript/expressions/string.py index 7e05e9cf3..08ebe0bf9 100644 --- a/src/codegen/sdk/typescript/expressions/string.py +++ b/src/codegen/sdk/typescript/expressions/string.py @@ -4,7 +4,7 @@ from codegen.sdk.core.expressions import Expression, String from codegen.sdk.core.node_id_factory import NodeId -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/typescript/expressions/ternary_expression.py b/src/codegen/sdk/typescript/expressions/ternary_expression.py index ab00561a5..9743e98c5 100644 --- a/src/codegen/sdk/typescript/expressions/ternary_expression.py +++ b/src/codegen/sdk/typescript/expressions/ternary_expression.py @@ -2,7 +2,7 @@ from codegen.sdk.core.expressions.ternary_expression import TernaryExpression from codegen.sdk.core.interfaces.editable import Editable -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/typescript/expressions/undefined_type.py b/src/codegen/sdk/typescript/expressions/undefined_type.py index 281c41702..fa6f86ea8 100644 --- a/src/codegen/sdk/typescript/expressions/undefined_type.py +++ b/src/codegen/sdk/typescript/expressions/undefined_type.py @@ -6,7 +6,7 @@ from codegen.sdk.core.expressions.type import Type from codegen.sdk.core.interfaces.importable import Importable from codegen.sdk.extensions.autocommit import reader -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/typescript/expressions/union_type.py b/src/codegen/sdk/typescript/expressions/union_type.py index 6084f6c02..5c9d2322d 100644 --- a/src/codegen/sdk/typescript/expressions/union_type.py +++ b/src/codegen/sdk/typescript/expressions/union_type.py @@ -1,7 +1,7 @@ from typing import TYPE_CHECKING, Generic, TypeVar from codegen.sdk.core.expressions.union_type import UnionType -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/typescript/file.py b/src/codegen/sdk/typescript/file.py index 93edcd24a..bdb1f33f4 100644 --- a/src/codegen/sdk/typescript/file.py +++ b/src/codegen/sdk/typescript/file.py @@ -26,7 +26,7 @@ from codegen.sdk.typescript.ts_config import TSConfig from codegen.sdk.typescript.type_alias import TSTypeAlias from codegen.sdk.utils import calculate_base_path, find_all_descendants -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/typescript/function.py b/src/codegen/sdk/typescript/function.py index 736231fd8..f4ed86306 100644 --- a/src/codegen/sdk/typescript/function.py +++ b/src/codegen/sdk/typescript/function.py @@ -23,7 +23,7 @@ from codegen.sdk.typescript.placeholder.placeholder_return_type import TSReturnTypePlaceholder from codegen.sdk.typescript.symbol import TSSymbol from codegen.sdk.utils import find_all_descendants -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.core.statements.export_statement import ExportStatement diff --git a/src/codegen/sdk/typescript/import_resolution.py b/src/codegen/sdk/typescript/import_resolution.py index 8d83f6c64..10cfa1a91 100644 --- a/src/codegen/sdk/typescript/import_resolution.py +++ b/src/codegen/sdk/typescript/import_resolution.py @@ -19,7 +19,7 @@ from codegen.sdk.core.symbol import Symbol from codegen.sdk.enums import ImportType, NodeType from codegen.sdk.utils import find_all_descendants, find_first_ancestor, find_first_descendant -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.typescript.file import TSFile diff --git a/src/codegen/sdk/typescript/interface.py b/src/codegen/sdk/typescript/interface.py index 120480df2..86824e931 100644 --- a/src/codegen/sdk/typescript/interface.py +++ b/src/codegen/sdk/typescript/interface.py @@ -19,7 +19,7 @@ from codegen.sdk.typescript.interfaces.has_block import TSHasBlock from codegen.sdk.typescript.statements.attribute import TSAttribute from codegen.sdk.typescript.symbol import TSSymbol -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc Parent = TypeVar("Parent", bound="TSHasBlock") diff --git a/src/codegen/sdk/typescript/interfaces/has_block.py b/src/codegen/sdk/typescript/interfaces/has_block.py index 82cd20291..5086b80bd 100644 --- a/src/codegen/sdk/typescript/interfaces/has_block.py +++ b/src/codegen/sdk/typescript/interfaces/has_block.py @@ -12,7 +12,7 @@ from codegen.sdk.typescript.statements.comment import TSComment, TSCommentType from codegen.sdk.typescript.symbol_groups.comment_group import TSCommentGroup from codegen.sdk.utils import find_index -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc if TYPE_CHECKING: from codegen.sdk.typescript.detached_symbols.jsx.element import JSXElement diff --git a/src/codegen/sdk/typescript/namespace.py b/src/codegen/sdk/typescript/namespace.py index c9df745eb..f4f1fbcbf 100644 --- a/src/codegen/sdk/typescript/namespace.py +++ b/src/codegen/sdk/typescript/namespace.py @@ -21,7 +21,7 @@ from codegen.sdk.typescript.interfaces.has_block import TSHasBlock from codegen.sdk.typescript.symbol import TSSymbol from codegen.sdk.typescript.type_alias import TSTypeAlias -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock diff --git a/src/codegen/sdk/typescript/placeholder/placeholder_return_type.py b/src/codegen/sdk/typescript/placeholder/placeholder_return_type.py index 346c6f31d..2f553397a 100644 --- a/src/codegen/sdk/typescript/placeholder/placeholder_return_type.py +++ b/src/codegen/sdk/typescript/placeholder/placeholder_return_type.py @@ -2,7 +2,7 @@ from codegen.sdk.core.interfaces.editable import Editable from codegen.sdk.core.placeholder.placeholder import Placeholder -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc Parent = TypeVar("Parent", bound="Editable") diff --git a/src/codegen/sdk/typescript/statements/assignment_statement.py b/src/codegen/sdk/typescript/statements/assignment_statement.py index 5a48e451e..7cb415102 100644 --- a/src/codegen/sdk/typescript/statements/assignment_statement.py +++ b/src/codegen/sdk/typescript/statements/assignment_statement.py @@ -11,7 +11,7 @@ from codegen.sdk.core.statements.assignment_statement import AssignmentStatement from codegen.sdk.extensions.autocommit import reader from codegen.sdk.typescript.assignment import TSAssignment -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock diff --git a/src/codegen/sdk/typescript/statements/attribute.py b/src/codegen/sdk/typescript/statements/attribute.py index 93ae3c4e9..6a8978130 100644 --- a/src/codegen/sdk/typescript/statements/attribute.py +++ b/src/codegen/sdk/typescript/statements/attribute.py @@ -13,7 +13,7 @@ from codegen.sdk.typescript.assignment import TSAssignment from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock from codegen.sdk.typescript.statements.assignment_statement import TSAssignmentStatement -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc if TYPE_CHECKING: from codegen.sdk.typescript.interfaces.has_block import TSHasBlock diff --git a/src/codegen/sdk/typescript/statements/block_statement.py b/src/codegen/sdk/typescript/statements/block_statement.py index 96cd02f94..a3f70e497 100644 --- a/src/codegen/sdk/typescript/statements/block_statement.py +++ b/src/codegen/sdk/typescript/statements/block_statement.py @@ -5,7 +5,7 @@ from codegen.sdk.core.statements.block_statement import BlockStatement from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock from codegen.sdk.typescript.interfaces.has_block import TSHasBlock -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/typescript/statements/catch_statement.py b/src/codegen/sdk/typescript/statements/catch_statement.py index 10e021fdc..cd02df4e0 100644 --- a/src/codegen/sdk/typescript/statements/catch_statement.py +++ b/src/codegen/sdk/typescript/statements/catch_statement.py @@ -8,7 +8,7 @@ from codegen.sdk.core.statements.catch_statement import CatchStatement from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock from codegen.sdk.typescript.statements.block_statement import TSBlockStatement -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/typescript/statements/comment.py b/src/codegen/sdk/typescript/statements/comment.py index 7559e3d0c..405e94c8b 100644 --- a/src/codegen/sdk/typescript/statements/comment.py +++ b/src/codegen/sdk/typescript/statements/comment.py @@ -5,7 +5,7 @@ from codegen.sdk.core.autocommit import commiter, reader from codegen.sdk.core.statements.comment import Comment, lowest_indentation -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: pass diff --git a/src/codegen/sdk/typescript/statements/for_loop_statement.py b/src/codegen/sdk/typescript/statements/for_loop_statement.py index 819ba15f5..f77a0e8a6 100644 --- a/src/codegen/sdk/typescript/statements/for_loop_statement.py +++ b/src/codegen/sdk/typescript/statements/for_loop_statement.py @@ -13,7 +13,7 @@ from codegen.sdk.core.statements.for_loop_statement import ForLoopStatement from codegen.sdk.extensions.autocommit import commiter, reader from codegen.sdk.typescript.statements.block_statement import TSBlockStatement -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/typescript/statements/if_block_statement.py b/src/codegen/sdk/typescript/statements/if_block_statement.py index 8c0db8a0f..01c432cdd 100644 --- a/src/codegen/sdk/typescript/statements/if_block_statement.py +++ b/src/codegen/sdk/typescript/statements/if_block_statement.py @@ -9,7 +9,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.statements.if_block_statement import IfBlockStatement from codegen.sdk.core.statements.statement import StatementType -from codegen.sdk.writer_decorators import apidoc +from codegen.utils.decorators.docs import apidoc if TYPE_CHECKING: from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock diff --git a/src/codegen/sdk/typescript/statements/import_statement.py b/src/codegen/sdk/typescript/statements/import_statement.py index ea643ea48..cebb069a3 100644 --- a/src/codegen/sdk/typescript/statements/import_statement.py +++ b/src/codegen/sdk/typescript/statements/import_statement.py @@ -10,7 +10,7 @@ from codegen.sdk.core.statements.import_statement import ImportStatement from codegen.sdk.core.symbol_groups.collection import Collection from codegen.sdk.typescript.import_resolution import TSImport -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc if TYPE_CHECKING: from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock diff --git a/src/codegen/sdk/typescript/statements/labeled_statement.py b/src/codegen/sdk/typescript/statements/labeled_statement.py index bd9ae110a..e9f36e38e 100644 --- a/src/codegen/sdk/typescript/statements/labeled_statement.py +++ b/src/codegen/sdk/typescript/statements/labeled_statement.py @@ -9,7 +9,7 @@ from codegen.sdk.core.interfaces.has_name import HasName from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.statements.statement import Statement, StatementType -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc if TYPE_CHECKING: from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock diff --git a/src/codegen/sdk/typescript/statements/switch_case.py b/src/codegen/sdk/typescript/statements/switch_case.py index e0f08fa8a..2cd55870c 100644 --- a/src/codegen/sdk/typescript/statements/switch_case.py +++ b/src/codegen/sdk/typescript/statements/switch_case.py @@ -6,7 +6,7 @@ from codegen.sdk.core.statements.switch_case import SwitchCase from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock from codegen.sdk.typescript.statements.block_statement import TSBlockStatement -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/typescript/statements/switch_statement.py b/src/codegen/sdk/typescript/statements/switch_statement.py index ae2bf9293..292fcb692 100644 --- a/src/codegen/sdk/typescript/statements/switch_statement.py +++ b/src/codegen/sdk/typescript/statements/switch_statement.py @@ -8,7 +8,7 @@ from codegen.sdk.core.statements.switch_statement import SwitchStatement from codegen.sdk.typescript.detached_symbols.code_block import TSCodeBlock from codegen.sdk.typescript.statements.switch_case import TSSwitchCase -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/typescript/statements/try_catch_statement.py b/src/codegen/sdk/typescript/statements/try_catch_statement.py index 3d2e3e3dc..bd100ada8 100644 --- a/src/codegen/sdk/typescript/statements/try_catch_statement.py +++ b/src/codegen/sdk/typescript/statements/try_catch_statement.py @@ -12,7 +12,7 @@ from codegen.sdk.extensions.autocommit import commiter, reader from codegen.sdk.typescript.statements.block_statement import TSBlockStatement from codegen.sdk.typescript.statements.catch_statement import TSCatchStatement -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/typescript/statements/while_statement.py b/src/codegen/sdk/typescript/statements/while_statement.py index 8374fab8c..79b46e259 100644 --- a/src/codegen/sdk/typescript/statements/while_statement.py +++ b/src/codegen/sdk/typescript/statements/while_statement.py @@ -7,7 +7,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.statements.while_statement import WhileStatement from codegen.sdk.typescript.interfaces.has_block import TSHasBlock -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/typescript/symbol.py b/src/codegen/sdk/typescript/symbol.py index 2a8dace5d..c3a978f5e 100644 --- a/src/codegen/sdk/typescript/symbol.py +++ b/src/codegen/sdk/typescript/symbol.py @@ -22,7 +22,7 @@ from codegen.sdk.typescript.import_resolution import TSImport from codegen.sdk.typescript.statements.comment import TSComment, TSCommentType from codegen.sdk.typescript.symbol_groups.comment_group import TSCommentGroup -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.core.file import SourceFile diff --git a/src/codegen/sdk/typescript/symbol_groups/comment_group.py b/src/codegen/sdk/typescript/symbol_groups/comment_group.py index 09a8d0d0d..051a795e6 100644 --- a/src/codegen/sdk/typescript/symbol_groups/comment_group.py +++ b/src/codegen/sdk/typescript/symbol_groups/comment_group.py @@ -4,7 +4,7 @@ from codegen.sdk.core.symbol_groups.comment_group import CommentGroup from codegen.sdk.typescript.statements.comment import TSComment, TSCommentType -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.typescript.symbol import TSSymbol diff --git a/src/codegen/sdk/typescript/symbol_groups/dict.py b/src/codegen/sdk/typescript/symbol_groups/dict.py index 9cbc57368..1ceb4dcc6 100644 --- a/src/codegen/sdk/typescript/symbol_groups/dict.py +++ b/src/codegen/sdk/typescript/symbol_groups/dict.py @@ -11,7 +11,7 @@ from codegen.sdk.core.node_id_factory import NodeId from codegen.sdk.core.symbol_groups.dict import Dict, Pair from codegen.sdk.extensions.autocommit import reader -from codegen.sdk.writer_decorators import apidoc, noapidoc, ts_apidoc +from codegen.utils.decorators.docs import apidoc, noapidoc, ts_apidoc if TYPE_CHECKING: from codegen.sdk.codebase.codebase_graph import CodebaseGraph diff --git a/src/codegen/sdk/typescript/ts_config.py b/src/codegen/sdk/typescript/ts_config.py index 7cb841ef6..012e23f94 100644 --- a/src/codegen/sdk/typescript/ts_config.py +++ b/src/codegen/sdk/typescript/ts_config.py @@ -8,7 +8,7 @@ from codegen.sdk.core.directory import Directory from codegen.sdk.core.file import File -from codegen.sdk.writer_decorators import ts_apidoc +from codegen.utils.decorators.docs import ts_apidoc if TYPE_CHECKING: from codegen.sdk.typescript.config_parser import TSConfigParser diff --git a/src/codegen/sdk/typescript/type_alias.py b/src/codegen/sdk/typescript/type_alias.py index 2c85b8c51..b34bf6212 100644 --- a/src/codegen/sdk/typescript/type_alias.py +++ b/src/codegen/sdk/typescript/type_alias.py @@ -7,7 +7,7 @@ from codegen.sdk.typescript.interfaces.has_block import TSHasBlock from codegen.sdk.typescript.statements.attribute import TSAttribute from codegen.sdk.typescript.symbol import TSSymbol -from codegen.sdk.writer_decorators import noapidoc, ts_apidoc +from codegen.utils.decorators.docs import noapidoc, ts_apidoc @ts_apidoc diff --git a/src/codegen/sdk/writer_decorators.py b/src/codegen/sdk/writer_decorators.py index 0d39945cc..91f677a51 100644 --- a/src/codegen/sdk/writer_decorators.py +++ b/src/codegen/sdk/writer_decorators.py @@ -1,87 +1,9 @@ -import bisect -import inspect -from collections.abc import Callable -from dataclasses import dataclass -from typing import TypeVar - - -@dataclass -class DocumentedObject: - name: str - module: str - object: any - - def __lt__(self, other): - return self.module < other.module - - def signature(self) -> str: - return f"{self.name}" +from codegen.sdk.enums import ProgrammingLanguage def canonical(codemod): """Decorator for canonical Codemods that will be used for AI-agent prompts.""" - from codegen.sdk.enums import ProgrammingLanguage - codemod._canonical = True if not hasattr(codemod, "language") or codemod.language not in (ProgrammingLanguage.PYTHON, ProgrammingLanguage.TYPESCRIPT): raise AttributeError("Canonical codemods must have a `language` attribute (PYTHON or TYPESCRIPT).") return codemod - - -apidoc_objects: list[DocumentedObject] = [] - - -def apidoc(obj): - """Decorator for objects that will be used as API documentation for AI-agent prompts.""" - obj._apidoc = True - obj._api_doc_lang = "core" - if doc_obj := get_documented_object(obj): - bisect.insort(apidoc_objects, doc_obj) - return obj - - -py_apidoc_objects: list[DocumentedObject] = [] - - -def py_apidoc(obj): - """Decorator for objects that will be used as Python API documentation for AI-agent prompts.""" - obj._py_apidoc = True - obj._api_doc_lang = "python" - if doc_obj := get_documented_object(obj): - bisect.insort(py_apidoc_objects, doc_obj) - return obj - - -ts_apidoc_objects: list[DocumentedObject] = [] - - -def ts_apidoc(obj): - """Decorator for objects that will be used as Typescript API documentation for AI-agent prompts.""" - obj._ts_apidoc = True - obj._api_doc_lang = "typescript" - if doc_obj := get_documented_object(obj): - bisect.insort(ts_apidoc_objects, doc_obj) - return obj - - -no_apidoc_objects: list[DocumentedObject] = [] -no_apidoc_signatures: set[str] = set() - -T = TypeVar("T", bound=Callable) - - -def noapidoc(obj: T) -> T: - """Decorator for things that are hidden from the API documentation for AI-agent prompts.""" - obj._apidoc = False - obj._api_doc_lang = None - if doc_obj := get_documented_object(obj): - bisect.insort(no_apidoc_objects, doc_obj) - no_apidoc_signatures.add(doc_obj.signature()) - return obj - - -def get_documented_object(obj) -> DocumentedObject | None: - module = inspect.getmodule(obj) - module_name = module.__name__ if module else "" - if module_name: - return DocumentedObject(name=obj.__name__, module=module_name, object=obj) diff --git a/src/codegen/utils/csv_utils.py b/src/codegen/utils/csv_utils.py new file mode 100644 index 000000000..3cd4fd366 --- /dev/null +++ b/src/codegen/utils/csv_utils.py @@ -0,0 +1,21 @@ +def list_to_comma_separated(items: list[str]) -> str: + """Given a list of items, returns a comma separated string of the items""" + return ",".join(items) + + +def comma_separated_to_list(comma_separated: str) -> list[str]: + """Given a comma separated string, returns a list of the comma separated items. + Strips whitespace from each item, drops any items that are whitespace only + """ + items = comma_separated.split(",") if comma_separated else [] + non_empty_items = [item.strip() for item in items if item.strip()] + return non_empty_items + + +def comma_separated_to_set(comma_separated: str) -> set[str]: + """Given a comma separated string, returns a set of the comma separated items. + Strips whitespace from each item, drops any items that are whitespace only + """ + items = comma_separated.split(",") if comma_separated else [] + non_empty_items = {item.strip() for item in items if item.strip()} + return non_empty_items diff --git a/src/codegen/utils/decorators/docs.py b/src/codegen/utils/decorators/docs.py new file mode 100644 index 000000000..e7c2b1d24 --- /dev/null +++ b/src/codegen/utils/decorators/docs.py @@ -0,0 +1,77 @@ +import bisect +import inspect +from collections.abc import Callable +from dataclasses import dataclass +from typing import TypeVar + + +@dataclass +class DocumentedObject: + name: str + module: str + object: any + + def __lt__(self, other): + return self.module < other.module + + def signature(self) -> str: + return f"{self.name}" + + +apidoc_objects: list[DocumentedObject] = [] + + +def apidoc(obj): + """Decorator for objects that will be used as API documentation for AI-agent prompts.""" + obj._apidoc = True + obj._api_doc_lang = "core" + if doc_obj := get_documented_object(obj): + bisect.insort(apidoc_objects, doc_obj) + return obj + + +py_apidoc_objects: list[DocumentedObject] = [] + + +def py_apidoc(obj): + """Decorator for objects that will be used as Python API documentation for AI-agent prompts.""" + obj._py_apidoc = True + obj._api_doc_lang = "python" + if doc_obj := get_documented_object(obj): + bisect.insort(py_apidoc_objects, doc_obj) + return obj + + +ts_apidoc_objects: list[DocumentedObject] = [] + + +def ts_apidoc(obj): + """Decorator for objects that will be used as Typescript API documentation for AI-agent prompts.""" + obj._ts_apidoc = True + obj._api_doc_lang = "typescript" + if doc_obj := get_documented_object(obj): + bisect.insort(ts_apidoc_objects, doc_obj) + return obj + + +no_apidoc_objects: list[DocumentedObject] = [] +no_apidoc_signatures: set[str] = set() + +T = TypeVar("T", bound=Callable) + + +def noapidoc(obj: T) -> T: + """Decorator for things that are hidden from the API documentation for AI-agent prompts.""" + obj._apidoc = False + obj._api_doc_lang = None + if doc_obj := get_documented_object(obj): + bisect.insort(no_apidoc_objects, doc_obj) + no_apidoc_signatures.add(doc_obj.signature()) + return obj + + +def get_documented_object(obj) -> DocumentedObject | None: + module = inspect.getmodule(obj) + module_name = module.__name__ if module else "" + if module_name: + return DocumentedObject(name=obj.__name__, module=module_name, object=obj) diff --git a/src/codegen/sdk/codebase/control_flow.py b/src/codegen/utils/exceptions/control_flow.py similarity index 100% rename from src/codegen/sdk/codebase/control_flow.py rename to src/codegen/utils/exceptions/control_flow.py diff --git a/src/codegen/utils/performance/memory_utils.py b/src/codegen/utils/performance/memory_utils.py new file mode 100644 index 000000000..bd04a7e3d --- /dev/null +++ b/src/codegen/utils/performance/memory_utils.py @@ -0,0 +1,20 @@ +import os +from dataclasses import dataclass + +import psutil + + +@dataclass +class MemoryStats: + memory_rss_gb: float + memory_vms_gb: float + + +def get_memory_stats() -> MemoryStats: + process = psutil.Process(os.getpid()) + memory_info = process.memory_info() + + return MemoryStats( + memory_rss_gb=memory_info.rss / 1024 / 1024 / 1024, + memory_vms_gb=memory_info.vms / 1024 / 1024 / 1024, + ) diff --git a/src/codegen/git/utils/stopwatch_utils.py b/src/codegen/utils/performance/stopwatch_utils.py similarity index 96% rename from src/codegen/git/utils/stopwatch_utils.py rename to src/codegen/utils/performance/stopwatch_utils.py index 4e2a00bd7..ac4b76e9f 100644 --- a/src/codegen/git/utils/stopwatch_utils.py +++ b/src/codegen/utils/performance/stopwatch_utils.py @@ -5,7 +5,7 @@ import sentry_sdk -from codegen.git.utils.time_utils import humanize_duration +from codegen.utils.time_utils import humanize_duration logger = logging.getLogger(__name__) diff --git a/src/codegen/git/utils/time_utils.py b/src/codegen/utils/time_utils.py similarity index 100% rename from src/codegen/git/utils/time_utils.py rename to src/codegen/utils/time_utils.py diff --git a/tests/unit/codebase/transaction_manager/test_transaction_mangager_checks.py b/tests/unit/codebase/transaction_manager/test_transaction_mangager_checks.py index e9d5b8aae..2c6e41087 100644 --- a/tests/unit/codebase/transaction_manager/test_transaction_mangager_checks.py +++ b/tests/unit/codebase/transaction_manager/test_transaction_mangager_checks.py @@ -1,9 +1,9 @@ import pytest from codegen.sdk.codebase.config import SessionOptions -from codegen.sdk.codebase.control_flow import MaxPreviewTimeExceeded, MaxTransactionsExceeded from codegen.sdk.codebase.factory.get_session import get_codebase_session from codegen.sdk.enums import ProgrammingLanguage +from codegen.utils.exceptions.control_flow import MaxPreviewTimeExceeded, MaxTransactionsExceeded def test_check_max_preview_time_exceeded(tmpdir): diff --git a/tests/unit/python/codebase/test_codebase_log.py b/tests/unit/python/codebase/test_codebase_log.py index eb09b87a3..413ec7d72 100644 --- a/tests/unit/python/codebase/test_codebase_log.py +++ b/tests/unit/python/codebase/test_codebase_log.py @@ -3,9 +3,9 @@ import pytest from codegen.sdk.codebase.config import SessionOptions -from codegen.sdk.codebase.control_flow import MaxPreviewTimeExceeded from codegen.sdk.codebase.factory.get_session import get_codebase_session from codegen.sdk.enums import ProgrammingLanguage +from codegen.utils.exceptions.control_flow import MaxPreviewTimeExceeded def test_log_writes_to_console(tmpdir) -> None: diff --git a/uv.lock b/uv.lock index 1cbfffb34..d7fc4977e 100644 --- a/uv.lock +++ b/uv.lock @@ -396,6 +396,7 @@ dependencies = [ { name = "openai" }, { name = "pip" }, { name = "plotly" }, + { name = "psutil" }, { name = "pydantic" }, { name = "pydantic-core" }, { name = "pygit2" }, @@ -489,6 +490,7 @@ requires-dist = [ { name = "openai", specifier = "==1.59.9" }, { name = "pip", specifier = ">=24.3.1" }, { name = "plotly", specifier = ">=5.24.0,<6.0.0" }, + { name = "psutil", specifier = ">=5.8.0" }, { name = "pydantic", specifier = ">=2.9.2,<3.0.0" }, { name = "pydantic-core", specifier = ">=2.23.4" }, { name = "pygit2", specifier = ">=1.16.0" },