用 Amazon Quick 把企业数据快速推向 AI 决策——五项新能力拆解与实践

import boto3
import json

qs = boto3.client("quicksight", region_name="us-east-1")

AWS_ACCOUNT_ID = "123456789012"   # 替换为你的 AWS 账号
ANALYSIS_ID    = "your-analysis-id"
DATA_SET_ARN   = "arn:aws:quicksight:us-east-1:123456789012:dataset/your-dataset-id"

# 向 Q 提问
response = qs.generate_q_query(
    AwsAccountId=AWS_ACCOUNT_ID,
    QueryText="上季度华东区域各品类销售额同比变化",
    AnalysisId=ANALYSIS_ID,
)

query_id = response["QueryId"]
print(f"查询已提交, QueryId: {query_id}")

# 查询结果（实际场景中可能需要轮询等待完成）
result = qs.get_q_query_result(
    AwsAccountId=AWS_ACCOUNT_ID,
    QueryId=query_id,
    AnalysisId=ANALYSIS_ID,
)

# 打印 Q 返回的 SQL 和可视化建议
print("生成的 SQL:")
print(result.get("GeneratedSql", "N/A"))
print("\n可视化类型建议:")
print(result.get("VisualType", "N/A"))

# 1. 创建权限规则 JSON 文件
cat > row-level-permissions.json << 'EOF'
{
  "RowLevelPermissionDataSet": {
    "Arn": "arn:aws:quicksight:us-east-1:123456789012:dataset/rlp-rules-dataset",
    "PermissionPolicy": "GRANT_ACCESS",
    "FormatVersion": "VERSION_2"
  },
  "RowLevelPermissionTagConfiguration": {
    "Status": "ENABLED"
  }
}
EOF

# 2. 应用到目标数据集
aws quicksight update-data-set-permissions \
  --aws-account-id 123456789012 \
  --data-set-id your-target-dataset-id \
  --row-level-permissions file://row-level-permissions.json

# quicksight-integration.yaml — 用 AWS CloudFormation 部署最小集成
AWSTemplateFormatVersion: "2010-09-09"
Description: "Minimal QuickSight + Q integration stack"

Resources:
  # 数据集：指向 Glue 表
  SalesDataset:
    Type: AWS::QuickSight::DataSet
    Properties:
      AwsAccountId: !Ref AWS::AccountId
      DataSetId: sales-ai-dataset
      Name: "Sales Data for AI Insights"
      PhysicalTableMap:
        PhysicalTableMapKey: sales-glue-table
        PhysicalTable:
          RelationalTable:
            DataSourceArn: !GetAtt SalesDataSource.Arn
            Schema: "sales_db"
            Name: "transactions"
            InputColumns:
              - Name: "region"
                Type: "STRING"
              - Name: "category"
                Type: "STRING"
              - Name: "amount"
                Type: "DECIMAL"
              - Name: "quarter"
                Type: "STRING"

  # 数据源
  SalesDataSource:
    Type: AWS::QuickSight::DataSource
    Properties:
      AwsAccountId: !Ref AWS::AccountId
      DataSourceId: sales-glue-source
      Name: "Glue Sales Source"
      Type: "ATHENA"

  # 启用 Q（需在 QuickSight 控制台或通过 API 激活 Q capacity）
  # 注意：Q 的启用目前主要通过控制台完成，以下为 API 方式的示意
  QCapacity:
    Type: Custom::QCapacity
    Properties:
      ServiceToken: !GetAtt QCapacityFunction.Arn
      AwsAccountId: !Ref AWS::AccountId