Merge pull request #6 from UMLCloudComputing/specify-semesters

Specify semesters

Author: cjcocokrisp

.gitignore

@@ -5,6 +5,7 @@ __pycache__
 .venv
 *.egg-info
 *.env
+cdk.context.json
 # CDK asset staging directory
 .cdk.staging
 cdk.out

attendance/attendance_stack.py

@@ -4,7 +4,8 @@
     Duration,
     aws_dynamodb as dynamodb,
     aws_lambda as _lambda,
-    aws_apigateway as apigateway
+    aws_apigateway as apigateway,
+    aws_ssm as ssm,
 )
 import os
 from constructs import Construct
@@ -28,7 +29,8 @@ def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
                 "DISCORD_ID" : os.getenv('ID'),
                 "DISCORD_TOKEN" : os.getenv('TOKEN'),
                 "DYNAMO_USERTABLE" : user_table.table_name,
-                "DYNAMO_CODETABLE" : code_table.table_name
+                "DYNAMO_CODETABLE" : code_table.table_name,
+                "SSM_PARAMETER_NAME" : os.getenv('SSM_PARAMETER_NAME')
             },            
             code=_lambda.DockerImageCode.from_image_asset(
                 directory="src"
@@ -40,6 +42,15 @@ def __init__(self, scope: Construct, construct_id: str, **kwargs) -> None:
             handler=dockerFunc,
             proxy=True,
         )
+        
+        parameter = ssm.StringParameter.from_string_parameter_name(
+            self, 
+            f'SSMParameter{construct_id}',
+            string_parameter_name=os.getenv('SSM_PARAMETER_NAME')
+        )
+
+        parameter.grant_read(dockerFunc.role)
+        parameter.grant_write(dockerFunc.role)
 
         user_table.grant_read_write_data(dockerFunc.role)
         code_table.grant_read_write_data(dockerFunc.role)

commands/discord_commands.yaml

@@ -45,6 +45,22 @@
       type: 3 # string
       required: true
 
+- name: set_semester
+  description: set the semester that it currently is. (admins only)
+  options:
+    - name: semester
+      description: name of the semester
+      type: 3 # string
+      required: true
+
+- name: generate_report
+  description: generate a report for a semester (admins only)
+  options:
+    - name: semester
+      description: semester to query
+      type: 3 # string
+      required: true
+
 - name: reset
   description: reset a user's statistics only can be used by admins.
   options:

src/Dockerfile

@@ -9,7 +9,7 @@ RUN yum -y install git
 RUN pip install --upgrade -r requirements.txt
 
 # Copy all files in ./app
-COPY app/* ${LAMBDA_TASK_ROOT}
+COPY app/ ${LAMBDA_TASK_ROOT}
 
 # Set the CMD to your handler (could also be done as a parameter override outside of the Dockerfile)
 CMD [ "main.handler" ]

src/app/main.py

@@ -1,4 +1,5 @@
 import os
+import io
 import requests
 import json
 import db
@@ -7,6 +8,8 @@
 from nacl.exceptions import BadSignatureError
 import secrets
 from enum import Enum
+from parameter import get_ssm_parameter, set_ssm_parameter
+from report import generate_report
 
 DISCORD_PUBLIC_KEY = os.environ.get("DISCORD_PUBLIC_KEY")
 DISCORD_TOKEN = os.environ.get("DISCORD_TOKEN")
@@ -62,6 +65,9 @@ def interact(raw_request):
     # The ID of the Guild in which the bot resides in
     guildID = raw_request["guild_id"]
 
+    # The ID of the channel in which the interaction happened
+    channelID = raw_request["channel_id"]
+
     # A boolean variable that determines if the user who executed the command is an administrator or not
     admin = (int(raw_request["member"]["permissions"]) & 0x8) == 0x8
 
@@ -120,6 +126,18 @@ def interact(raw_request):
                 user = raw_request["member"]["user"]
             embeds = build_stats_embed(user)
             send_embed(embeds, id, token)
+        case "set_semester":
+            if admin: 
+                semester = str(data["options"][0]["value"])
+                set_ssm_parameter(semester)
+                send(f"The semester has been set to {semester} successfully.", id, token)
+            else: send("Only administrators can set the semester!", id, token)
+        case "generate_report":
+            if admin:
+                semester = str(data["options"][0]["value"])
+                filename = generate_report(semester)
+                send_file(filename, id, token)
+            else: send("Only administrators can generate reports!", id, token)
         case "reset":
             if admin:
                 # send(f"Deleting specified user...", id, token)
@@ -214,7 +232,7 @@ def validate_attendance(userid: str, code: str, type: str) -> AttendanceStatus:
         if active:
             user = db.get_user(userid)
             serialized_code = code + '|' + code_response['expiration']
-            serialized_event = code_response['event_name'] + '|' + type
+            serialized_event = code_response['event_name'] + '|' + type + '|' + get_ssm_parameter()
             if user != None:
                 if serialized_code not in user['codes_used']: 
                     status = AttendanceStatus.VALID
@@ -248,9 +266,14 @@ def build_stats_embed(user):
         i = len(user_stats["events_attended"]) - 1
         while len(embeds['fields']) < 25 and i >= 0:
             event_deserialized = user_stats['events_attended'][i].split('|')
+            try:
+                semester = event_deserialized[2]
+            except:
+                semester = "Fall 2024"
+
             embeds['fields'].append({
                 'name': event_deserialized[0],
-                'value': event_deserialized[1],
+                'value': event_deserialized[1] + f" ({semester})",
                 'inline': True
             })
 
@@ -266,4 +289,26 @@ def get_mentioned_user(userid):
     }
     response = requests.get(f"https://discord.com/api/users/{userid}", headers=headers)
 
-    return response.json()
+    return response.json()
+
+def send_file(filename: str, id, token):
+    url = f"https://discord.com/api/interactions/{id}/{token}/callback"
+    
+    with open(filename, 'rb') as fp:
+        callback_data = {
+            "type": 4,
+            "data": {
+                "content": "Here is the requested report.",
+            }
+        }
+
+        response = requests.post(
+            url,
+            data={"payload_json": json.dumps(callback_data)},
+            files={
+                "file": (filename, fp, 'text/csv')
+            }
+        )
+        
+        print("Response status: code: ")
+        print(response.status_code)

src/app/parameter.py

@@ -0,0 +1,21 @@
+import boto3
+import os
+
+AWS_ID = os.getenv("AWS_ACCESS_KEY_ID")
+AWS_SECRET = os.getenv("AWS_SECRET_ACCESS_KEY")
+PARAMETER_NAME = os.getenv("SSM_PARAMETER_NAME")
+
+def set_ssm_parameter(value: str) -> None:
+    ssm = boto3.client('ssm')
+
+    ssm.put_parameter(
+        Name=PARAMETER_NAME, 
+        Value=value, 
+        Overwrite=True
+    )
+
+def get_ssm_parameter() -> str:
+    ssm = boto3.client('ssm')
+
+    response = ssm.get_parameter(Name='/attendance/semester')
+    return response['Parameter']['Value']

src/app/report.py

@@ -0,0 +1,55 @@
+import boto3
+import csv
+import os
+
+AWS_ID = os.getenv("AWS_ACCESS_KEY_ID")
+AWS_SECRET = os.getenv("AWS_SECRET_ACCESS_KEY")
+USER_TABLENAME = os.getenv("DYNAMO_USERTABLE")
+
+def generate_report(report_semester: str) -> bytes:
+    filename = '/tmp/attendance-report-' + report_semester.lower().replace(' ', '-') + '.csv'
+
+    dynamodb = boto3.client("dynamodb")
+
+    response = dynamodb.scan(
+        TableName=USER_TABLENAME,
+        Select="ALL_ATTRIBUTES"
+    )
+
+    fp = open(filename, 'w')
+    csvwriter = csv.DictWriter(fp, fieldnames=['meeting_name', 'total_attended', 'in_person', 'virtual'])
+    csvwriter.writeheader()
+    data = {}
+    
+    for item in response['Items']:
+        for event in item['events_attended']['L']:
+            try:
+                name, type, semester = event['S'].split('|')
+            except ValueError:
+                name, type = event['S'].split('|')
+                semester = "Fall 2024"
+
+
+            if semester == report_semester:
+                if name not in data.keys():
+                    data[name] = {
+                        'total_attended': 0,
+                        'in_person': 0,
+                        'virtual': 0
+                    }
+                    
+                data[name][type.lower().replace(' ', '_')] += 1
+                data[name]['total_attended'] += 1
+
+    rows = []
+    for event in data.keys():
+        row = {
+            'meeting_name': event,
+            'total_attended': data[event]['total_attended'],
+            'in_person': data[event]['in_person'],
+            'virtual': data[event]['virtual']
+        }
+        rows.append(row)
+    csvwriter.writerows(rows)
+
+    return filename