updates

Author: ford-at-aws

aws_doc_sdk_examples_tools/stats/app.ts

@@ -16,10 +16,10 @@ class CodeCommitCloneStack extends cdk.Stack {
   constructor(scope: Construct, id: string, props?: cdk.StackProps) {
     super(scope, id, props);
 
-    // Initialize Lambda function
+    // Create Lambda function
     const cloneLambda = this.initCloneLambda();
 
-    // Set up EventBridge rule to trigger Lambda on CodeCommit repository changes
+    // Create EventBridge rule to trigger Lambda on CodeCommit repository changes
     this.initCodeCommitTrigger(cloneLambda);
   }
 
@@ -47,19 +47,19 @@ class CodeCommitCloneStack extends cdk.Stack {
       })
     );
 
-    // Grant necessary permissions to S3 bucket "codeexamplestats" for Get and Put
+    // Grant necessary permissions to S3 bucket "codeexamplestats" for Get/Put
     lambdaExecutionRole.addToPolicy(
       new iam.PolicyStatement({
         actions: ["s3:GetObject", "s3:PutObject"],
-        resources: [`arn:aws:s3:::codeexamplestats/*`],  // Allow access to all objects in the bucket
+        resources: [`arn:aws:s3:::codeexamplestats/*`],  // Allow all objects in the bucket
       })
     );
 
-    // Define the Lambda function, pointing directly to the source code directory
+    // Define the Lambda function, pointing directly to the source code dir
     const cloneLambda = new lambda.Function(this, "CodeCommitCloneLambda", {
       runtime: lambda.Runtime.PYTHON_3_9,
       handler: "index.lambda_handler",
-      code: lambda.Code.fromAsset(path.join(__dirname, "lambda")), // Pointing to the directory of the lambda function code
+      code: lambda.Code.fromAsset(path.join(__dirname, "lambda")),
       environment: {
         REPO_NAME: repoName,
       },
@@ -71,7 +71,7 @@ class CodeCommitCloneStack extends cdk.Stack {
   }
 
   private initCodeCommitTrigger(cloneLambda: lambda.Function): void {
-    // Create EventBridge rule for CodeCommit repository updates
+    // EventBridge rule for CodeCommit repo updates
     const codeCommitRule = new events.Rule(this, "CodeCommitUpdateRule", {
       eventPattern: {
         source: ["aws.codecommit"],
@@ -87,7 +87,7 @@ class CodeCommitCloneStack extends cdk.Stack {
       }
     });
 
-    // Add Lambda function as the target of the EventBridge rule
+    // Add Lambda function as target of the EventBridge rule
     codeCommitRule.addTarget(new targets.LambdaFunction(cloneLambda));
   }
 }
@@ -96,7 +96,7 @@ const app = new cdk.App();
 new CodeCommitCloneStack(app, "CodeCommitCloneStack", {
   env: {
     account: process.env.CDK_DEFAULT_ACCOUNT,
-    region: "us-west-2",  // Where codecommit is stored
+    region: "us-west-2",  // Where codecommit is stored (internal requirement)
   },
 });
 app.synth();

aws_doc_sdk_examples_tools/stats/lambda/index.py

@@ -6,11 +6,12 @@
 import csv
 from datetime import datetime  # Import datetime module
 
+
 def lambda_handler(event, context):
     # Initialize boto3 clients
     codecommit = boto3.client("codecommit", region_name=os.environ["AWS_REGION"])
     s3 = boto3.client("s3")
-    
+
     # Environment variables
     repo_name = os.environ.get("REPO_NAME")
     branch_name = os.environ.get("BRANCH_NAME", "mainline")  # Default to "mainline"
@@ -21,39 +22,41 @@ def lambda_handler(event, context):
         # Step 1: Retrieve the .stats file content directly from CodeCommit
         file_response = codecommit.get_file(
             repositoryName=repo_name,
-            filePath='.stats',  # Specify the path to the .stats file
-            commitSpecifier=branch_name
+            filePath=".stats",  # Specify the path to the .stats file
+            commitSpecifier=branch_name,
         )
-        
+
         # Convert .stats content to valid JSON if necessary
-        file_content = file_response['fileContent'].decode("utf-8")
+        file_content = file_response["fileContent"].decode("utf-8")
         try:
             stats_data = json.loads(file_content)  # Valid JSON parsing
         except json.JSONDecodeError:
-            file_content = file_content.replace("'", "\"")  # Replace single quotes
+            file_content = file_content.replace("'", '"')  # Replace single quotes
             stats_data = json.loads(file_content)
-        
+
         # Step 2: Fetch the current stats.csv file from S3
         existing_rows = []
         try:
             csv_obj = s3.get_object(Bucket=bucket_name, Key=csv_file_key)
-            csv_data = csv_obj['Body'].read().decode('utf-8')
+            csv_data = csv_obj["Body"].read().decode("utf-8")
             csv_reader = csv.DictReader(StringIO(csv_data))
             existing_rows = list(csv_reader)
         except s3.exceptions.NoSuchKey:
             existing_rows = []
 
         # Step 3: Append the new data from .stats file with a formatted timestamp
         new_row = {
-            'sdks': stats_data['sdks'],
-            'services': stats_data['services'],
-            'examples': stats_data['examples'],
-            'versions': stats_data['versions'],
-            'snippets': stats_data['snippets'],
-            'genai_none': stats_data['genai']['none'],
-            'genai_some': stats_data['genai']['some'],
-            'genai_most': stats_data['genai']['most'],
-            'timestamp': datetime.now().strftime("%d/%m/%Y %H:%M:%S")  # Formatted timestamp
+            "sdks": stats_data["sdks"],
+            "services": stats_data["services"],
+            "examples": stats_data["examples"],
+            "versions": stats_data["versions"],
+            "snippets": stats_data["snippets"],
+            "genai_none": stats_data["genai"]["none"],
+            "genai_some": stats_data["genai"]["some"],
+            "genai_most": stats_data["genai"]["most"],
+            "timestamp": datetime.now().strftime(
+                "%d/%m/%Y %H:%M:%S"
+            ),  # Formatted timestamp
         }
         existing_rows.append(new_row)
 
@@ -63,10 +66,14 @@ def lambda_handler(event, context):
             writer = csv.DictWriter(output, fieldnames=new_row.keys())
             writer.writeheader()  # Ensure we have headers
             writer.writerows(existing_rows)
-            s3.put_object(Bucket=bucket_name, Key=csv_file_key, Body=output.getvalue().encode('utf-8'))
-        
+            s3.put_object(
+                Bucket=bucket_name,
+                Key=csv_file_key,
+                Body=output.getvalue().encode("utf-8"),
+            )
+
         print("Successfully updated stats.csv in S3 with new .stats data.")
-    
+
     except Exception as e:
         print(f"Error occurred: {e}")
         return {"statusCode": 500, "body": "Failed to update stats.csv."}