AI 工具开发实战3开发一个 AI 代码审查 CLI——自动检查 Git 变更、发现潜在问题前两篇做了翻译工具和 RAG 知识库这篇做一个程序员最需要的工具AI 代码审查 CLI。在 Git 仓库里跑一条命令自动检查这次提交改了哪些代码调用 AI 分析潜在问题——安全漏洞、性能隐患、逻辑错误。项目结构aicr/ ├── aicr.py # CLI 主程序 ├── reviewer.py # AI 审查引擎 ├── requirements.txt └── .env审查引擎# reviewer.pyfromopenaiimportOpenAIimportosfromdotenvimportload_dotenv load_dotenv()clientOpenAI(api_keyos.getenv(DEEPSEEK_API_KEY),base_urlhttps://api.deepseek.com/v1,)REVIEW_PROMPT你是一个资深代码审查员。审查以下 Git diff从三个维度分析 1. 安全风险SQL 注入、XSS、密钥泄露、权限问题 2. 性能问题N1 查询、不必要的循环、内存泄漏 3. 代码质量命名不规范、过度复杂、缺少异常处理 格式要求 - 只报告确实存在的问题不要猜测 - 对于小改动或纯格式变更直接说✅ 未发现问题 - 每个问题标注严重程度高/中/低 以下是 Git diff {diff}defreview_diff(diff:str,modeldeepseek-chat)-dict:审查 Git diff 并返回问题列表。iflen(diff)15000:diffdiff[:15000]\n... (diff truncated)responseclient.chat.completions.create(modelmodel,messages[{role:system,content:REVIEW_PROMPT.format(diffdiff)},],temperature0.1,)returnresponse.choices[0].message.contentCLI 主程序#!/usr/bin/env python3# aicr.pyimportsubprocessimportclickfrompathlibimportPathfromreviewerimportreview_diffdefget_git_diff(stagedFalse,branchNone):获取 Git diff。cmd[git,diff]ifstaged:cmd.append(--staged)ifbranch:cmd.extend([forigin/{branch}..HEAD])resultsubprocess.run(cmd,capture_outputTrue,textTrue)ifresult.returncode!0:raiseclick.ClickException(fGit error:{result.stderr})returnresult.stdoutdefget_changed_files(stagedFalse,branchNone):获取变更文件列表。cmd[git,diff,--name-only]ifstaged:cmd.append(--staged)ifbranch:cmd.extend([forigin/{branch}..HEAD])resultsubprocess.run(cmd,capture_outputTrue,textTrue)return[fforfinresult.stdout.strip().split(\n)iff]click.group()defcli():aicr - AI 代码审查工具passcli.command()click.option(--staged,-s,is_flagTrue,help只审查已暂存staged的改动)click.option(--branch,-b,defaultNone,help对比远程分支如 main)click.option(--model,-m,defaultdeepseek-chat,help模型名称)defcheck(staged,branch,model):审查未提交的代码改动。click.echo( 正在审查代码改动...)diffget_git_diff(stagedstaged,branchbranch)ifnotdiff.strip():click.echo(✅ 没有待审查的改动)returnfilesget_changed_files(stagedstaged,branchbranch)click.echo(f 变更文件 ({len(files)}个))forfinfiles:click.echo(f{f})resultreview_diff(diff,modelmodel)click.echo(f\n{result})cli.command()click.argument(commit_range,defaultHEAD~1..HEAD)click.option(--model,-m,defaultdeepseek-chat)defhistory(commit_range,model):审查历史提交。用法: aicr history HEAD~3..HEADresultsubprocess.run([git,diff,commit_range],capture_outputTrue,textTrue)ifresult.returncode!0:raiseclick.ClickException(fGit error:{result.stderr})diffresult.stdoutifnotdiff.strip():click.echo(✅ 没有改动)returnclick.echo(f 审查{commit_range}的改动...)resultreview_diff(diff,modelmodel)click.echo(f\n{result})cli.command()click.argument(filepath,typeclick.Path(existsTrue))click.option(--model,-m,defaultdeepseek-chat)deffile(filepath,model):审查单个文件。withopen(filepath,r,encodingutf-8)asf:contentf.read()# 构造一个虚拟 diffdifff b/{Path(filepath).name}\n\n.join(f{line}forlineincontent.split(\n))click.echo(f 审查文件{filepath})resultreview_diff(diff,modelmodel)click.echo(f\n{result})if__name____main__:cli()使用方式# 审查未提交的改动aicr check# 正在审查代码改动...# 变更文件 (2 个)# src/api/users.py# src/models/user.py## 安全风险# [高] 第 23 行SQL 查询直接拼接用户输入存在注入风险## 性能问题# [中] 第 45 行循环内查询数据库建议改用批量查询## 代码质量# [低] 第 12 行变量名 user_data 建议改为 user_profile 更清晰# 只审查已暂存的文件aicr check--staged# 对比远程主分支aicr check--branchmain# 审查最近 3 次提交aicrhistoryHEAD~3..HEAD# 审查单个文件aicrfilesrc/api/auth.py设为 Git Hook在.git/hooks/pre-commit中加#!/bin/bashecho AI 正在审查代码...aicr check--staged每次git commit前自动运行发现问题会阻止提交需要加--force跳过。实际效果在项目中测试审查 200 行代码改动5 个文件用时约 3 秒发现1 个安全风险SQL 拼接1 个性能问题循环内查询2 个代码质量建议成本用 DeepSeek-V4 审查一次约消耗 2000-5000 token费用约 ¥0.01-0.02几乎可以忽略。总结一个实用的 AI 代码审查工具核心就两步用 Git 命令获取代码改动把 diff 丢给 AI 分析配合 Git Hook 实现提交前自动审查可以有效减少低级问题进入代码库。本文是《AI 开发者工具链实战》系列的第 3 篇。上一篇本地 RAG 知识库下一篇VS Code AI 插件如果觉得有用欢迎点赞 收藏 关注后续会持续更新 AI 工具开发实战系列全部附完整源码。