Amazon Q Developer：AI编程助手

from typing import List, Dict, Optional class CodeCompleter:    def __init__(self):        self.languages = ['python', 'java', 'javascript', 'typescript']        self.aws_patterns = {            's3_upload': 'boto3.client("s3").upload_file(...)',            'lambda_invoke': 'boto3.client("lambda").invoke(...)',            'dynamodb_put': 'boto3.resource("dynamodb").Table(...).put_item(...)'        }    def complete(self, context: str, language: str) -> List[str]:        suggestions = []        for key, pattern in self.aws_patterns.items():            if key.split('_')[0] in context.lower():                suggestions.append(pattern)        if not suggestions:            suggestions.append(f'# {language} code suggestion')        return suggestions    def generate_function(self, comment: str, language: str) -> str:        if 's3' in comment.lower():            return 'import boto3\ndef upload_to_s3(bucket, key, path):\n    client = boto3.client("s3")\n    client.upload_file(path, bucket, key)\n    return f"Uploaded to s3://{bucket}/{key}"'        return f'def generated_function():\n    # {comment}\n    pass' class SecurityScanner:    def __init__(self):        self.vulnerabilities = [            {'pattern': 'Action": "*"', 'severity': 'HIGH', 'fix': '收敛到具体Action'},            {'pattern': 'PublicAccessBlockConfiguration: false', 'severity': 'CRITICAL', 'fix': '设置BlockPublicAccess为true'},            {'pattern': '0.0.0.0/0', 'severity': 'MEDIUM', 'fix': '限制CIDR范围'}        ]    def scan(self, code: str) -> List[Dict]:        findings = []        for vuln in self.vulnerabilities:            if vuln['pattern'] in code:                findings.append({'type': vuln['pattern'], 'severity': vuln['severity'], 'fix': vuln['fix']})        return findings class CloudFormationGenerator:    def generate(self, description: str) -> Dict:        template = {'AWSTemplateFormatVersion': '2010-09-09', 'Resources': {}}        if 'vpc' in description.lower():            template['Resources']['VPC'] = {'Type': 'AWS::EC2::VPC', 'Properties': {'CidrBlock': '10.0.0.0/16'}}        if 'lambda' in description.lower():            template['Resources']['LambdaFn'] = {'Type': 'AWS::Lambda::Function', 'Properties': {'Runtime': 'python3.9', 'Handler': 'index.handler'}}        if 'rds' in description.lower():            template['Resources']['Database'] = {'Type': 'AWS::RDS::DBInstance', 'Properties': {'Engine': 'mysql', 'DBInstanceClass': 'db.t3.micro'}}        return template class AmazonQAssistant:    def __init__(self):        self.completer = CodeCompleter()        self.scanner = SecurityScanner()        self.cfn_gen = CloudFormationGenerator()    def assist(self, task_type: str, input_text: str, language: str = 'python') -> Dict:        if task_type == 'complete':            return {'suggestions': self.completer.complete(input_text, language)}        elif task_type == 'generate_function':            return {'code': self.completer.generate_function(input_text, language)}        elif task_type == 'security_scan':            return {'findings': self.scanner.scan(input_text)}        elif task_type == 'generate_template':            return {'template': self.cfn_gen.generate(input_text)}        return {'error': 'unknown task type'} def demo():    q = AmazonQAssistant()    r1 = q.assist('complete', 'upload file to s3', 'python')    print(f'补全建议: {r1["suggestions"]}')    r2 = q.assist('security_scan', '{"Action": "*", "Effect": "Allow"}')    print(f'安全扫描: {r2["findings"]}')    r3 = q.assist('generate_template', 'create VPC with Lambda and RDS')    print(f'模板资源: {list(r3["template"]["Resources"].keys())}') if __name__ == '__main__':    demo()