def process_users(users):
    result = []
    for user in users:
        if user['age'] > 18:
            name = user['name'].upper()
            email = user['email'].lower()
            result.append({
                'name': name,
                'email': email,
                'adult': True
            })
    return result

def code_review_with_ai(code: str, focus: str = "all"):
    """使用 AI 进行代码审查"""
    
    prompt = f"""
    请审查以下代码，重点关注：{focus}
    
    ```
    {code}
    ```
    
    请按以下格式输出：
    
    ## 🎯 整体评价
    [总体评价]
    
    ## ✅ 优点
    - [优点1]
    - [优点2]
    
    ## ⚠️ 问题
    ### 严重问题
    - [问题1]
      - 位置：[行号]
      - 说明：[...]
      - 建议：[...]
    
    ### 一般问题
    - [问题2]
      - 位置：[行号]
      - 说明：[...]
      - 建议：[...]
    
    ## 💡 优化建议
    - [建议1]
    - [建议2]
    
    ## 📝 改进后的代码
    ```[language]
    [改进后的代码]
    ```
    """
    
    return call_ai(prompt)

# 使用示例
review = code_review_with_ai(
    code=my_code,
    focus="安全性和性能"
)

print(review)

# Step 1：让 AI 生成测试
你的输入：
"""
功能：计算购物车总价
需求：
- 支持折扣
- 支持优惠券
- 满减活动
请生成测试用例
"""

# AI 生成测试
```python
def test_calculate_total_basic():
    """基本价格计算"""
    cart = Cart()
    cart.add_item(Item("Apple", 10.0, 2))
    assert cart.calculate_total() == 20.0

def test_calculate_total_with_discount():
    """应用折扣"""
    cart = Cart()
    cart.add_item(Item("Apple", 10.0, 2))
    cart.apply_discount(0.1)  # 10% 折扣
    assert cart.calculate_total() == 18.0

def test_calculate_total_with_coupon():
    """使用优惠券"""
    cart = Cart()
    cart.add_item(Item("Apple", 10.0, 2))
    cart.apply_coupon(Coupon("SAVE5", 5.0))
    assert cart.calculate_total() == 15.0

def test_calculate_total_full_price():
    """满减活动"""
    cart = Cart()
    cart.add_item(Item("Apple", 10.0, 10))  # 总计100
    # 满100减20
    assert cart.calculate_total() == 80.0

def ai_quality_check(code: str):
    """AI 辅助质量检查"""
    
    checks = {
        "代码风格": check_code_style(code),
        "类型安全": check_type_safety(code),
        "安全漏洞": check_security(code),
        "性能问题": check_performance(code),
        "代码复杂度": check_complexity(code)
    }
    
    # 生成报告
    report = generate_report(checks)
    
    # AI 提供改进建议
    suggestions = ai_suggest_improvements(code, checks)
    
    return {
        "report": report,
        "suggestions": suggestions
    }

def security_checklist(code: str):
    """安全检查清单"""
    
    prompt = f"""
    请检查以下代码的安全性：
    
    ```
    {code}
    ```
    
    重点检查：
    
    1. 输入验证
       - [ ] 所有用户输入都经过验证
       - [ ] 使用白名单而非黑名单
       - [ ] 长度限制
    
    2. SQL 注入
       - [ ] 使用参数化查询
       - [ ] 避免动态 SQL 拼接
    
    3. XSS 防护
       - [ ] 输出编码
       - [ ] Content Security Policy
    
    4. 认证授权
       - [ ] 密码加密存储
       - [ ] 会话管理安全
       - [ ] 权限检查
    
    5. 敏感数据
       - [ ] 不在日志中记录敏感信息
       - [ ] 使用环境变量存储密钥
       - [ ] 加密传输
    
    6. 错误处理
       - [ ] 不泄露系统信息
       - [ ] 统一错误响应格式
    
    对于每个检查项：
    - ✅ 通过
    - ⚠️ 需要改进
    - ❌ 存在问题
    
    并给出具体的改进建议。
    """
    
    return call_ai(prompt)

# ❌ 不安全
def get_user(username):
    query = f"SELECT * FROM users WHERE username = '{username}'"
    return db.execute(query)

# ✅ 安全
def get_user(username):
    query = "SELECT * FROM users WHERE username = ?"
    return db.execute(query, (username,))

# ❌ 不安全
def save_password(password):
    user.password = password  # 明文存储

# ✅ 安全
import bcrypt

def save_password(password):
    salt = bcrypt.gensalt()
    hashed = bcrypt.hashpw(password.encode(), salt)
    user.password = hashed

# ❌ 不安全
API_KEY = "sk-1234567890abcdef"  # 硬编码在代码中

# ✅ 安全
import os
API_KEY = os.environ.get("API_KEY")  # 从环境变量读取

# .ai-config.yaml

# AI 生成代码的风格配置
code_style:
  language: python
  version: "3.10+"
  
  formatting:
    line_length: 88
    quotes: double
    indent: 4
    
  naming:
    function: snake_case
    class: PascalCase
    constant: UPPER_CASE
    
  documentation:
    style: google  # Google, NumPy, or reStructuredText
    include_examples: true
    include_types: true
    
  imports:
    order: [stdlib, third_party, local]
    format: black
    
  type_hints:
    enabled: true
    strict: false
    
  error_handling:
    prefer_specific_exceptions: true
    always_cleanup: true
    
  testing:
    framework: pytest
    coverage_target: 80
    include_edge_cases: true

# 使用方式
# 在提示词中引用：
# "请按照我们的代码风格配置生成代码（参考 .ai-config.yaml）"

async def update_knowledge_base(code_changes, issue_resolution):
    """自动更新知识库"""
    
    # 1. 分析代码变更
    patterns = await ai_analyze_patterns(code_changes)
    
    # 2. 提取最佳实践
    if is_good_practice(patterns):
        await add_to_knowledge_base(
            category="最佳实践",
            content=patterns
        )
    
    # 3. 记录问题解决方案
    if issue_resolution:
        await add_to_knowledge_base(
            category="问题解决",
            content={
                "problem": issue_resolution.problem,
                "solution": issue_resolution.solution,
                "code": issue_resolution.code
            }
        )
    
    # 4. 生成文档
    doc = await ai_generate_documentation(patterns)
    await save_documentation(doc)