changeset 6:db2ce7097ff3

add string parameters
author Dennis Concepcion Martin <dennisconcepcionmartin@gmail.com>
date Fri, 17 Sep 2021 17:42:42 +0200
parents 54e71cf6e324
children 1b1296559c31
files .aws-sam/build.toml .idea/tweet-analysis.iml src/handlers/sentiment.py
diffstat 3 files changed, 23 insertions(+), 77 deletions(-) [+]
line wrap: on
line diff
--- a/.aws-sam/build.toml	Fri Sep 17 17:42:30 2021 +0200
+++ b/.aws-sam/build.toml	Fri Sep 17 17:42:42 2021 +0200
@@ -1,8 +1,8 @@
 # This file is auto generated by SAM CLI build command
 
 [function_build_definitions]
-[function_build_definitions.d30384fa-aa96-4ffd-9beb-35e450903b28]
-codeuri = "/Users/dennis/Developer/tweet-analysis/src"
+[function_build_definitions.631c5dbb-3ad7-4033-889a-2b55c4ab69a0]
+codeuri = "/Users/dennis/Developer/tweet-analysis/handlers"
 runtime = "python3.9"
 source_md5 = ""
 packagetype = "Zip"
--- a/.idea/tweet-analysis.iml	Fri Sep 17 17:42:30 2021 +0200
+++ b/.idea/tweet-analysis.iml	Fri Sep 17 17:42:42 2021 +0200
@@ -2,6 +2,7 @@
 <module type="PYTHON_MODULE" version="4">
   <component name="NewModuleRootManager">
     <content url="file://$MODULE_DIR$">
+      <sourceFolder url="file://$MODULE_DIR$" isTestSource="false" />
       <excludeFolder url="file://$MODULE_DIR$/.aws-sam" />
     </content>
     <orderEntry type="inheritedJdk" />
--- a/src/handlers/sentiment.py	Fri Sep 17 17:42:30 2021 +0200
+++ b/src/handlers/sentiment.py	Fri Sep 17 17:42:42 2021 +0200
@@ -1,8 +1,11 @@
-import json
 import requests
-import boto3
-import base64
-from botocore.exceptions import ClientError
+
+# noinspection PyUnresolvedReferences
+from secrets_controller import get_secret
+# noinspection PyUnresolvedReferences
+from event_controller import unwrap_sentiment_string_parameters
+# noinspection PyUnresolvedReferences
+from url_controller import create_twitter_url
 
 
 def get_tweet_sentiment(event, context):
@@ -15,80 +18,22 @@
         API Gateway Lambda Proxy Output Format
     """
 
-    twitter_url = create_twitter_url(user='dennisconcep')
-    twitter_key = get_twitter_key()
-    bearer_token = twitter_key['BEARER']
-    twitter_header = {"Authorization": "Bearer {}".format(bearer_token)}  # Auth header
+    # Unwrap query string parameters
+    twitter_user, number_of_tweets = unwrap_sentiment_string_parameters(event)
+
+    # URL creation & authentication
+    twitter_url = create_twitter_url(twitter_user, number_of_tweets)
+    twitter_key = get_secret(secret_name='tweet-analysis-keys')
+    twitter_header = {"Authorization": "Bearer {}".format(twitter_key['BEARER'])}
+
+    # Request tweets to Twitter
     twitter_response = requests.request("GET", twitter_url, headers=twitter_header)
 
+    # Analyse tweets with AWS Comprehend
+
     return {
         "statusCode": 200,
-        "body": json.dumps({
-            "message": "hello world",
+        "body": {
             "tweets": twitter_response.json()
-        }),
+        }
     }
-
-
-def create_twitter_url(user, max_results=100):
-    """
-    Create url to fetch `max_results` of tweets from `user`
-    :param user: string, required
-    :param max_results: int, optional, default 100
-    :return: string url
-    """
-
-    formatted_max_results = 'max_results={}'.format(max_results)
-    formatted_user = 'query=from:{}'.format(user)
-    url = "https://api.twitter.com/2/tweets/search/recent?{}&{}".format(formatted_max_results, formatted_user)
-
-    return url
-
-
-def get_twitter_key():
-    """
-    Get Twitter Api Key from AWS Secrets Manager
-    :return:
-    """
-    secret_name = "tweet-analysis-keys"
-    region_name = "eu-west-2"
-
-    # Create a Secrets Manager client
-    session = boto3.session.Session()
-    client = session.client(
-        service_name='secretsmanager',
-        region_name=region_name
-    )
-
-    try:
-        get_secret_value_response = client.get_secret_value(SecretId=secret_name)
-    except ClientError as e:
-        if e.response['Error']['Code'] == 'DecryptionFailureException':
-            # Secrets Manager can't decrypt the protected secret text using the provided KMS key.
-            # Deal with the exception here, and/or rethrow at your discretion.
-            raise e
-        elif e.response['Error']['Code'] == 'InternalServiceErrorException':
-            # An error occurred on the server side.
-            # Deal with the exception here, and/or rethrow at your discretion.
-            raise e
-        elif e.response['Error']['Code'] == 'InvalidParameterException':
-            # You provided an invalid value for a parameter.
-            # Deal with the exception here, and/or rethrow at your discretion.
-            raise e
-        elif e.response['Error']['Code'] == 'InvalidRequestException':
-            # You provided a parameter value that is not valid for the current state of the resource.
-            # Deal with the exception here, and/or rethrow at your discretion.
-            raise e
-        elif e.response['Error']['Code'] == 'ResourceNotFoundException':
-            # We can't find the resource that you asked for.
-            # Deal with the exception here, and/or rethrow at your discretion.
-            raise e
-    else:
-        # Decrypts secret using the associated KMS CMK.
-        # Depending on whether the secret is a string or binary, one of these fields will be populated.
-        if 'SecretString' in get_secret_value_response:
-            secret = get_secret_value_response['SecretString']
-        else:
-            secret = base64.b64decode(get_secret_value_response['SecretBinary'])
-
-        return json.loads(secret)