ai-cr 3.2.2__py3-none-any.whl → 3.3.0__py3-none-any.whl

gito/cli_base.py

@@ -1,94 +1,104 @@
-import contextlib
-import logging
-import tempfile
-
-import microcore as mc
-import typer
-from git import Repo
-from gito.utils import parse_refs_pair
-
-
-def args_to_target(refs, what, against) -> tuple[str | None, str | None]:
-    _what, _against = parse_refs_pair(refs)
-    if _what:
-        if what:
-            raise typer.BadParameter(
-                "You cannot specify both 'refs' <WHAT>..<AGAINST> and '--what'. Use one of them."
-            )
-    else:
-        _what = what
-    if _against:
-        if against:
-            raise typer.BadParameter(
-                "You cannot specify both 'refs' <WHAT>..<AGAINST> and '--against'. Use one of them."
-            )
-    else:
-        _against = against
-    return _what, _against
-
-
-def arg_refs() -> typer.Argument:
-    return typer.Argument(
-        default=None,
-        help=(
-            "Git refs to review, [what]..[against] (e.g., 'HEAD..HEAD~1'). "
-            "If omitted, the current index (including added but not committed files) "
-            "will be compared to the repository’s main branch."
-        ),
-    )
-
-
-def arg_what() -> typer.Option:
-    return typer.Option(None, "--what", "-w", help="Git ref to review")
-
-
-def arg_filters() -> typer.Option:
-    return typer.Option(
-        "", "--filter", "-f", "--filters",
-        help="""
-            filter reviewed files by glob / fnmatch pattern(s),
-            e.g. 'src/**/*.py', may be comma-separated
-            """,
-    )
-
-
-def arg_out() -> typer.Option:
-    return typer.Option(
-        None,
-        "--out", "-o", "--output",
-        help="Output folder for the code review report"
-    )
-
-
-def arg_against() -> typer.Option:
-    return typer.Option(
-        None,
-        "--against", "-vs", "--vs",
-        help="Git ref to compare against"
-    )
-
-
-app = typer.Typer(pretty_exceptions_show_locals=False)
-
-
-@contextlib.contextmanager
-def get_repo_context(url: str, branch: str):
-    """Context manager for handling both local and remote repositories."""
-    if url:
-        with tempfile.TemporaryDirectory() as temp_dir:
-            logging.info(
-                f"get_repo_context: "
-                f"Cloning [{mc.ui.green(url)}] to {mc.utils.file_link(temp_dir)} ..."
-            )
-            repo = Repo.clone_from(url, branch=branch, to_path=temp_dir)
-            try:
-                yield repo, temp_dir
-            finally:
-                repo.close()
-    else:
-        logging.info("get_repo_context: Using local repo...")
-        repo = Repo(".")
-        try:
-            yield repo, "."
-        finally:
-            repo.close()
+import contextlib
+import logging
+import tempfile
+
+import microcore as mc
+import typer
+from git import Repo
+from gito.constants import REFS_VALUE_ALL
+
+from .utils import parse_refs_pair
+from .env import Env
+
+
+def args_to_target(refs, what, against) -> tuple[str | None, str | None]:
+    if refs == REFS_VALUE_ALL:
+        return REFS_VALUE_ALL, None
+    _what, _against = parse_refs_pair(refs)
+    if _what:
+        if what:
+            raise typer.BadParameter(
+                "You cannot specify both 'refs' <WHAT>..<AGAINST> and '--what'. Use one of them."
+            )
+    else:
+        _what = what
+    if _against:
+        if against:
+            raise typer.BadParameter(
+                "You cannot specify both 'refs' <WHAT>..<AGAINST> and '--against'. Use one of them."
+            )
+    else:
+        _against = against
+    return _what, _against
+
+
+def arg_refs() -> typer.Argument:
+    return typer.Argument(
+        default=None,
+        help=(
+            "Git refs to review, [what]..[against] (e.g., 'HEAD..HEAD~1'). "
+            "If omitted, the current index (including added but not committed files) "
+            "will be compared to the repository’s main branch."
+        ),
+    )
+
+
+def arg_what() -> typer.Option:
+    return typer.Option(None, "--what", "-w", help="Git ref to review")
+
+
+def arg_filters() -> typer.Option:
+    return typer.Option(
+        "", "--filter", "-f", "--filters",
+        help="""
+            filter reviewed files by glob / fnmatch pattern(s),
+            e.g. 'src/**/*.py', may be comma-separated
+            """,
+    )
+
+
+def arg_out() -> typer.Option:
+    return typer.Option(
+        None,
+        "--out", "-o", "--output",
+        help="Output folder for the code review report"
+    )
+
+
+def arg_against() -> typer.Option:
+    return typer.Option(
+        None,
+        "--against", "-vs", "--vs",
+        help="Git ref to compare against"
+    )
+
+
+app = typer.Typer(pretty_exceptions_show_locals=False)
+
+
+@contextlib.contextmanager
+def get_repo_context(url: str, branch: str):
+    if branch == REFS_VALUE_ALL:
+        branch = None
+    """Context manager for handling both local and remote repositories."""
+    if url:
+        with tempfile.TemporaryDirectory() as temp_dir:
+            logging.info(
+                f"get_repo_context: "
+                f"Cloning [{mc.ui.green(url)}] to {mc.utils.file_link(temp_dir)} ..."
+            )
+            repo = Repo.clone_from(url, branch=branch, to_path=temp_dir)
+            prev_folder = Env.working_folder
+            Env.working_folder = temp_dir
+            try:
+                yield repo, temp_dir
+            finally:
+                repo.close()
+                Env.working_folder = prev_folder
+    else:
+        logging.info("get_repo_context: Using local repo...")
+        repo = Repo(".")
+        try:
+            yield repo, "."
+        finally:
+            repo.close()

gito/commands/__init__.py

@@ -1 +1 @@
-# Command modules register themselves with the CLI app
+# Command modules register themselves with the CLI app

gito/commands/deploy.py

@@ -1,138 +1,138 @@
-import logging
-from pathlib import Path
-
-import microcore as mc
-from microcore import ApiType, ui, utils
-from git import Repo, GitCommandError
-import typer
-
-from ..core import get_base_branch
-from ..utils import version, extract_gh_owner_repo
-from ..cli_base import app
-from ..gh_api import gh_api
-
-
-@app.command(
-    name="deploy",
-    help="\bCreate and configure Gito GitHub Actions for current repository.\naliases: init"
-)
-@app.command(name="init", hidden=True)
-def deploy(
-    api_type: ApiType = None,
-    commit: bool = None,
-    rewrite: bool = False,
-    to_branch: str = typer.Option(
-        default="gito_deploy",
-        help="Branch name for new PR containing with Gito workflows commit"
-    ),
-    token: str = typer.Option(
-        "", help="GitHub token (or set GITHUB_TOKEN env var)"
-    ),
-):
-    repo = Repo(".")
-    workflow_files = dict(
-        code_review=Path(".github/workflows/gito-code-review.yml"),
-        react_to_comments=Path(".github/workflows/gito-react-to-comments.yml")
-    )
-    for file in workflow_files.values():
-        if file.exists():
-            message = f"Gito workflow already exists at {utils.file_link(file)}."
-            if rewrite:
-                ui.warning(message)
-            else:
-                message += "\nUse --rewrite to overwrite it."
-                ui.error(message)
-                return False
-
-    api_types = [ApiType.ANTHROPIC, ApiType.OPEN_AI, ApiType.GOOGLE_AI_STUDIO]
-    default_models = {
-        ApiType.ANTHROPIC: "claude-sonnet-4-20250514",
-        ApiType.OPEN_AI: "gpt-4.1",
-        ApiType.GOOGLE_AI_STUDIO: "gemini-2.5-pro",
-    }
-    secret_names = {
-        ApiType.ANTHROPIC: "ANTHROPIC_API_KEY",
-        ApiType.OPEN_AI: "OPENAI_API_KEY",
-        ApiType.GOOGLE_AI_STUDIO: "GOOGLE_AI_API_KEY",
-    }
-    if not api_type:
-        api_type = mc.ui.ask_choose(
-            "Choose your LLM API type",
-            api_types,
-        )
-    elif api_type not in api_types:
-        mc.ui.error(f"Unsupported API type: {api_type}")
-        return False
-    major, minor, *_ = version().split(".")
-    template_vars = dict(
-        model=default_models[api_type],
-        api_type=api_type,
-        secret_name=secret_names[api_type],
-        major=major,
-        minor=minor,
-        ApiType=ApiType,
-        remove_indent=True,
-    )
-    gito_code_review_yml = mc.tpl(
-        "github_workflows/gito-code-review.yml.j2",
-        **template_vars
-    )
-    gito_react_to_comments_yml = mc.tpl(
-        "github_workflows/gito-react-to-comments.yml.j2",
-        **template_vars
-    )
-
-    workflow_files["code_review"].parent.mkdir(parents=True, exist_ok=True)
-    workflow_files["code_review"].write_text(gito_code_review_yml)
-    workflow_files["react_to_comments"].write_text(gito_react_to_comments_yml)
-    print(
-        mc.ui.green("Gito workflows have been created.\n")
-        + f"  - {mc.utils.file_link(workflow_files['code_review'])}\n"
-        + f"  - {mc.utils.file_link(workflow_files['react_to_comments'])}\n"
-    )
-    owner, repo_name = extract_gh_owner_repo(repo)
-    if commit is True or commit is None and mc.ui.ask_yn(
-        "Do you want to commit and push created GitHub workflows to a new branch?"
-    ):
-        repo.git.add([str(file) for file in workflow_files.values()])
-        if not repo.active_branch.name.startswith(to_branch):
-            repo.git.checkout("-b", to_branch)
-        try:
-            repo.git.commit("-m", "Deploy Gito workflows")
-        except GitCommandError as e:
-            if "nothing added" in str(e):
-                ui.warning("Failed to commit changes: nothing was added")
-            else:
-                ui.error(f"Failed to commit changes: {e}")
-                return False
-
-        repo.git.push("origin", to_branch)
-        print(f"Changes pushed to {to_branch} branch.")
-        try:
-            api = gh_api(repo=repo)
-            base = get_base_branch(repo).split('/')[-1]
-            logging.info(f"Creating PR {ui.green(to_branch)} -> {ui.yellow(base)}...")
-            res = api.pulls.create(
-                head=to_branch,
-                base=base,
-                title="Deploy Gito workflows",
-            )
-            print(f"Pull request #{res.number} created successfully:\n{res.html_url}")
-        except Exception as e:
-            mc.ui.error(f"Failed to create pull request automatically: {e}")
-            print(
-                f"Please create a PR from '{to_branch}' to your main branch and merge it:\n"
-                f"https://github.com/{owner}/{repo_name}/compare/{to_branch}?expand=1"
-            )
-    else:
-        print(
-            "Now you can commit and push created GitHub workflows to your main repository branch.\n"
-        )
-
-    print(
-        "(!IMPORTANT):\n"
-        f"Add {mc.ui.cyan(secret_names[api_type])} with actual API_KEY "
-        "to your repository secrets here:\n"
-        f"https://github.com/{owner}/{repo_name}/settings/secrets/actions"
-    )
-    return True
+import logging
+from pathlib import Path
+
+import microcore as mc
+from microcore import ApiType, ui, utils
+from git import Repo, GitCommandError
+import typer
+
+from ..core import get_base_branch
+from ..utils import version, extract_gh_owner_repo
+from ..cli_base import app
+from ..gh_api import gh_api
+
+
+@app.command(
+    name="deploy",
+    help="\bCreate and configure Gito GitHub Actions for current repository.\naliases: init"
+)
+@app.command(name="init", hidden=True)
+def deploy(
+    api_type: ApiType = None,
+    commit: bool = None,
+    rewrite: bool = False,
+    to_branch: str = typer.Option(
+        default="gito_deploy",
+        help="Branch name for new PR containing with Gito workflows commit"
+    ),
+    token: str = typer.Option(
+        "", help="GitHub token (or set GITHUB_TOKEN env var)"
+    ),
+):
+    repo = Repo(".")
+    workflow_files = dict(
+        code_review=Path(".github/workflows/gito-code-review.yml"),
+        react_to_comments=Path(".github/workflows/gito-react-to-comments.yml")
+    )
+    for file in workflow_files.values():
+        if file.exists():
+            message = f"Gito workflow already exists at {utils.file_link(file)}."
+            if rewrite:
+                ui.warning(message)
+            else:
+                message += "\nUse --rewrite to overwrite it."
+                ui.error(message)
+                return False
+
+    api_types = [ApiType.ANTHROPIC, ApiType.OPEN_AI, ApiType.GOOGLE_AI_STUDIO]
+    default_models = {
+        ApiType.ANTHROPIC: "claude-sonnet-4-20250514",
+        ApiType.OPEN_AI: "gpt-4.1",
+        ApiType.GOOGLE_AI_STUDIO: "gemini-2.5-pro",
+    }
+    secret_names = {
+        ApiType.ANTHROPIC: "ANTHROPIC_API_KEY",
+        ApiType.OPEN_AI: "OPENAI_API_KEY",
+        ApiType.GOOGLE_AI_STUDIO: "GOOGLE_AI_API_KEY",
+    }
+    if not api_type:
+        api_type = mc.ui.ask_choose(
+            "Choose your LLM API type",
+            api_types,
+        )
+    elif api_type not in api_types:
+        mc.ui.error(f"Unsupported API type: {api_type}")
+        return False
+    major, minor, *_ = version().split(".")
+    template_vars = dict(
+        model=default_models[api_type],
+        api_type=api_type,
+        secret_name=secret_names[api_type],
+        major=major,
+        minor=minor,
+        ApiType=ApiType,
+        remove_indent=True,
+    )
+    gito_code_review_yml = mc.tpl(
+        "github_workflows/gito-code-review.yml.j2",
+        **template_vars
+    )
+    gito_react_to_comments_yml = mc.tpl(
+        "github_workflows/gito-react-to-comments.yml.j2",
+        **template_vars
+    )
+
+    workflow_files["code_review"].parent.mkdir(parents=True, exist_ok=True)
+    workflow_files["code_review"].write_text(gito_code_review_yml)
+    workflow_files["react_to_comments"].write_text(gito_react_to_comments_yml)
+    print(
+        mc.ui.green("Gito workflows have been created.\n")
+        + f"  - {mc.utils.file_link(workflow_files['code_review'])}\n"
+        + f"  - {mc.utils.file_link(workflow_files['react_to_comments'])}\n"
+    )
+    owner, repo_name = extract_gh_owner_repo(repo)
+    if commit is True or commit is None and mc.ui.ask_yn(
+        "Do you want to commit and push created GitHub workflows to a new branch?"
+    ):
+        repo.git.add([str(file) for file in workflow_files.values()])
+        if not repo.active_branch.name.startswith(to_branch):
+            repo.git.checkout("-b", to_branch)
+        try:
+            repo.git.commit("-m", "Deploy Gito workflows")
+        except GitCommandError as e:
+            if "nothing added" in str(e):
+                ui.warning("Failed to commit changes: nothing was added")
+            else:
+                ui.error(f"Failed to commit changes: {e}")
+                return False
+
+        repo.git.push("origin", to_branch)
+        print(f"Changes pushed to {to_branch} branch.")
+        try:
+            api = gh_api(repo=repo)
+            base = get_base_branch(repo).split('/')[-1]
+            logging.info(f"Creating PR {ui.green(to_branch)} -> {ui.yellow(base)}...")
+            res = api.pulls.create(
+                head=to_branch,
+                base=base,
+                title="Deploy Gito workflows",
+            )
+            print(f"Pull request #{res.number} created successfully:\n{res.html_url}")
+        except Exception as e:
+            mc.ui.error(f"Failed to create pull request automatically: {e}")
+            print(
+                f"Please create a PR from '{to_branch}' to your main branch and merge it:\n"
+                f"https://github.com/{owner}/{repo_name}/compare/{to_branch}?expand=1"
+            )
+    else:
+        print(
+            "Now you can commit and push created GitHub workflows to your main repository branch.\n"
+        )
+
+    print(
+        "(!IMPORTANT):\n"
+        f"Add {mc.ui.cyan(secret_names[api_type])} with actual API_KEY "
+        "to your repository secrets here:\n"
+        f"https://github.com/{owner}/{repo_name}/settings/secrets/actions"
+    )
+    return True