manage_params.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
驾驶员手部检测系统 - 简化参数管理工具
"""

import boto3
import argparse
import json
from botocore.exceptions import ClientError

class SimpleParameterManager:
    """简化的参数管理器"""
    
    def __init__(self, region='us-east-1'):
        self.ssm = boto3.client('ssm', region_name=region)
        self.base_path = '/driver-hand-detection/config'
        
    def list_parameters(self):
        """列出所有参数"""
        try:
            response = self.ssm.get_parameters_by_path(
                Path=self.base_path,
                Recursive=True
            )
            
            print(f"\n📋 当前配置参数:")
            print("-" * 60)
            
            for param in response['Parameters']:
                name = param['Name'].replace(f"{self.base_path}/", "")
                value = param['Value']
                print(f"   {name}: {value}")
                
            print(f"\n共 {len(response['Parameters'])} 个参数")
            
        except ClientError as e:
            print(f"❌ 获取参数失败: {e}")
    
    def get_parameter(self, name):
        """获取单个参数"""
        try:
            full_name = f"{self.base_path}/{name}"
            response = self.ssm.get_parameter(Name=full_name)
            value = response['Parameter']['Value']
            print(f"✅ {name}: {value}")
            return value
        except ClientError as e:
            if e.response['Error']['Code'] == 'ParameterNotFound':
                print(f"❌ 参数 '{name}' 不存在")
            else:
                print(f"❌ 获取参数失败: {e}")
            return None
    
    def set_parameter(self, name, value, description=""):
        """设置参数"""
        try:
            full_name = f"{self.base_path}/{name}"
            self.ssm.put_parameter(
                Name=full_name,
                Value=str(value),
                Type='String',
                Description=description,
                Overwrite=True
            )
            print(f"✅ 已设置: {name} = {value}")
            return True
        except ClientError as e:
            print(f"❌ 设置参数失败: {e}")
            return False
    
    def delete_parameter(self, name):
        """删除参数"""
        try:
            full_name = f"{self.base_path}/{name}"
            self.ssm.delete_parameter(Name=full_name)
            print(f"✅ 已删除: {name}")
            return True
        except ClientError as e:
            print(f"❌ 删除参数失败: {e}")
            return False
    
    def show_calculation(self):
        """显示当前配置的计算结果"""
        try:
            frame_interval = self.get_parameter_value('frame-interval')
            frames_per_segment = self.get_parameter_value('frames-per-segment')
            
            if frame_interval and frames_per_segment:
                segment_duration = int(frames_per_segment) * int(frame_interval)
                print(f"\n🧮 配置计算:")
                print("-" * 40)
                print(f"   帧间隔: {frame_interval}秒")
                print(f"   每段帧数: {frames_per_segment}帧")
                print(f"   分段时长: {segment_duration}秒")
                print(f"   36分钟视频预期分段数: {36*60//segment_duration}段")
                
        except Exception as e:
            print(f"❌ 计算失败: {e}")
    
    def get_parameter_value(self, name):
        """获取参数值（不打印）"""
        try:
            full_name = f"{self.base_path}/{name}"
            response = self.ssm.get_parameter(Name=full_name)
            return response['Parameter']['Value']
        except:
            return None

def main():
    parser = argparse.ArgumentParser(description="驾驶员手部检测系统参数管理工具")
    parser.add_argument("--region", default="us-east-1", help="AWS区域")
    
    subparsers = parser.add_subparsers(dest='command', help='可用命令')
    
    # 列出参数
    subparsers.add_parser('list', help='列出所有参数')
    
    # 获取参数
    get_parser = subparsers.add_parser('get', help='获取参数值')
    get_parser.add_argument('name', help='参数名称')
    
    # 设置参数
    set_parser = subparsers.add_parser('set', help='设置参数值')
    set_parser.add_argument('name', help='参数名称')
    set_parser.add_argument('value', help='参数值')
    set_parser.add_argument('--desc', help='参数描述')
    
    # 删除参数
    del_parser = subparsers.add_parser('delete', help='删除参数')
    del_parser.add_argument('name', help='参数名称')
    
    # 显示计算
    subparsers.add_parser('calc', help='显示配置计算结果')
    
    # 快速设置
    quick_parser = subparsers.add_parser('quick', help='快速设置常用参数')
    quick_parser.add_argument('--frame-interval', type=int, help='帧间隔（秒）')
    quick_parser.add_argument('--frames-per-segment', type=int, help='每段帧数')
    quick_parser.add_argument('--frame-quality', type=int, help='帧质量(0-100)')
    
    args = parser.parse_args()
    
    if not args.command:
        parser.print_help()
        return
    
    manager = SimpleParameterManager(args.region)
    
    if args.command == 'list':
        manager.list_parameters()
        manager.show_calculation()
        
    elif args.command == 'get':
        manager.get_parameter(args.name)
        
    elif args.command == 'set':
        manager.set_parameter(args.name, args.value, args.desc or "")
        
    elif args.command == 'delete':
        manager.delete_parameter(args.name)
        
    elif args.command == 'calc':
        manager.show_calculation()
        
    elif args.command == 'quick':
        print("🔧 快速设置参数...")
        
        if args.frame_interval:
            manager.set_parameter('frame-interval', args.frame_interval, '抽帧时间间隔（秒）')
            
        if args.frames_per_segment:
            manager.set_parameter('frames-per-segment', args.frames_per_segment, '每个AI Lambda处理的帧数')
            
        if args.frame_quality:
            manager.set_parameter('frame-quality', args.frame_quality, 'JPEG质量(0-100)')
            
        print("\n📊 更新后的配置:")
        manager.list_parameters()
        manager.show_calculation()

if __name__ == "__main__":
    main()