add interested functionality

Author: transcental

app.py

@@ -1,5 +1,10 @@
 from utils.slack import app
 from utils.env import env
+from threading import Thread
+from utils.rsvp_checker import rsvp_checker
 
 if __name__ == "__main__":
+    rsvp_thread = Thread(target=rsvp_checker, daemon=True)
+    rsvp_thread.start()
+
     app.start(env.port)

events/buttons/approve_event.py

@@ -46,4 +46,4 @@ def handle_approve_event_btn(ack: Callable, body: dict[str, Any], client: WebCli
         text=f"<@{user_id}> approved {event['fields']['Title']} for <@{event['fields']['Leader Slack ID']}>.",
     )
 
-    client.views_publish(user_id=user_id, view=get_home(user_id))
+    client.views_publish(user_id=user_id, view=get_home(user_id, client))

events/buttons/rsvp.py

@@ -0,0 +1,24 @@
+from typing import Any, Callable
+from slack_sdk import WebClient
+from utils.env import env
+from views.app_home import get_home
+
+
+def handle_rsvp_btn(ack: Callable, body: dict[str, Any], client: WebClient):
+    ack()
+    user_id = body["user"]["id"]
+    event_id = body["actions"][0]["value"]
+    user = env.airtable.rsvp_to_event(event_id, user_id)
+    event = env.airtable.get_event(event_id)
+    if event["id"] not in user["fields"].get("Interesting Events", []):
+        client.chat_postMessage(
+            channel=user_id,
+            text=f"You're no longer interested in {event['fields']['Title']}! :(",
+        )
+    else:
+        client.chat_postMessage(
+            channel=user_id,
+            text=f"You're interested in {event['fields']['Title']}! We'll let you know when it starts.",
+        )
+
+    client.views_publish(user_id=user_id, view=get_home(user_id, client))

events/views/create_event.py

@@ -4,6 +4,7 @@
 
 from utils.env import env
 from utils.utils import rich_text_to_md, md_to_mrkdwn
+from views.app_home import get_home
 
 
 def handle_create_event_view(ack: Callable, body: dict[str, Any], client: WebClient):
@@ -16,8 +17,11 @@ def handle_create_event_view(ack: Callable, body: dict[str, Any], client: WebCli
     start_time = (values["start_time"]["start_time"]["selected_date_time"],)
     end_time = (values["end_time"]["end_time"]["selected_date_time"],)
     host_id = values["host"]["host"]["selected_user"]
-    location = values.get("location", {}).get("location", {}).get("value") or "https://hackclub.slack.com/archives/C07TNAZGMHS"
-    
+    location = (
+        values.get("location", {}).get("location", {}).get("value")
+        or "https://hackclub.slack.com/archives/C07TNAZGMHS"
+    )
+
     user = client.users_info(user=host_id)
     host_name = user["user"]["real_name"]
     host_pfp = user["user"]["profile"]["image_192"]
@@ -54,3 +58,5 @@ def handle_create_event_view(ack: Callable, body: dict[str, Any], client: WebCli
             }
         ],
     )
+
+    client.views_publish(user_id=user_id, view=get_home(user_id, client))

requirements.txt

@@ -1,3 +1,21 @@
+annotated-types==0.7.0
+black==24.10.0
+certifi==2024.8.30
+charset-normalizer==3.4.0
+click==8.1.7
+idna==3.10
+inflection==0.5.1
+mypy-extensions==1.0.0
+packaging==24.2
+pathspec==0.12.1
+platformdirs==4.3.6
 pyairtable==2.3.3
+pydantic==2.9.2
+pydantic_core==2.23.4
 python-dotenv==1.0.1
-slack_bolt==1.20.1
+requests==2.32.3
+schedule==1.2.2
+slack_bolt==1.20.1
+slack_sdk==3.33.3
+typing_extensions==4.12.2
+urllib3==2.2.3

utils/airtable.py

@@ -66,3 +66,40 @@ def update_event(
     ):
         event = self.events_table.update(id, updates)
         return event
+
+    def create_user(self, slack_id: str):
+        user = self.users_table.create(
+            {
+                "Slack ID": slack_id,
+            }
+        )
+        return user
+
+    def get_user(self, slack_id: str):
+        user = self.users_table.first(formula=f"{{Slack ID}} = '{slack_id}'")
+        if not user:
+            user = self.create_user(slack_id)
+        return user
+
+    def update_user(self, slack_id: str, **updates: dict):
+        user = self.users_table.update(slack_id, updates)
+        return user
+
+    def get_rsvps_from_event(self, event_id: str):
+        rsvps = self.users_table.all()
+        rsvps = [
+            rsvp
+            for rsvp in rsvps
+            if event_id in rsvp["fields"].get("Interesting Events", [])
+        ]
+        return rsvps
+
+    def rsvp_to_event(self, event_id: str, slack_id: str):
+        user = self.get_user(slack_id)
+        events = user["fields"].get("Interesting Events", [])
+        if event_id in events:
+            events.remove(event_id)
+        else:
+            events.append(event_id)
+        user = self.update_user(user["id"], **{"Interesting Events": events})
+        return user

utils/email.py

@@ -0,0 +1,55 @@
+import smtplib
+from email.message import EmailMessage
+
+
+class Email:
+    """Connect to SMTP server and send emails
+
+    Keyword arguments:
+    recipient -- str: email address of recipient
+    sender -- str: email address of sender
+    password -- str: password of sender
+    server -- str: SMTP server (default 'smtp.yandex.com')
+    port -- int: SMTP port (default 465)
+
+    """
+
+    def __init__(
+        self,
+        sender: str,
+        password: str,
+        server: str = "smtp.gmail.com",
+        port: int = 465,
+    ):
+        """Connect to SMTP server and set self variables"""
+        self.sender = sender
+        self.server = smtplib.SMTP_SSL(server, port)
+        self.password = password
+        try:
+            self.server.connect(server, port)
+            self.server.login(self.sender, password)
+        except Exception as e:
+            print(e)
+
+    def send_email(self, recipient: str, subject: str, message: str):
+        """Send an email
+
+        Keyword arguments:
+        subject -- str: Email subject
+        message -- str: Email content/message
+        """
+        #         email = f'Name: {self.name}\nEmail: {self.email}\nSubject: {self.subject}\nMessage: {message}'
+        msg = EmailMessage()
+        msg.set_content(message)
+        msg["Subject"] = subject
+        msg["From"] = f"Isabelle <{self.sender}>"
+        msg["To"] = recipient
+        try:
+            with self.server as server:
+                server.login(self.sender, self.password)
+                server.send_message(msg)
+            print("Email successfully sent")
+            return True
+        except Exception as e:
+            print("Error", e)
+            return False

utils/env.py

@@ -1,6 +1,7 @@
 from .airtable import AirtableManager
 from dotenv import load_dotenv
 import os
+from .email import Email
 
 load_dotenv()
 
@@ -13,6 +14,8 @@ def __init__(self):
         self.slack_sad_channel = os.environ.get("SLACK_SAD_CHANNEL")
         self.airtable_api_key = os.environ.get("AIRTABLE_API_KEY")
         self.airtable_base_id = os.environ.get("AIRTABLE_BASE_ID")
+        google_username = os.environ.get("GOOGLE_USERNAME")
+        google_password = os.environ.get("GOOGLE_PASSWORD")
 
         self.port = int(os.environ.get("PORT", 3000))
 
@@ -28,10 +31,17 @@ def __init__(self):
             raise Exception("AIRTABLE_API_KEY is not set")
         if not self.airtable_base_id:
             raise Exception("AIRTABLE_BASE_ID is not set")
+        if not google_username:
+            raise Exception("GOOGLE_USERNAME is not set")
+        if not google_password:
+            raise Exception("GOOGLE_PASSWORD is not set")
 
         self.airtable = AirtableManager(
             api_key=self.airtable_api_key, base_id=self.airtable_base_id
         )
+
+        self.mailer = Email(sender=google_username, password=google_password)
+
         self.authorised_users = [
             "U054VC2KM9P",  # Amber
             "U0409FSKU82",  # Arpan

utils/rsvp_checker.py

@@ -0,0 +1,73 @@
+from slack_sdk import WebClient
+from typing import Any
+import schedule
+import time
+from datetime import datetime
+
+from .env import env
+
+client = WebClient(token=env.slack_bot_token)
+
+
+def send_reminder(
+    user_id: str, message: str, event: dict[str, Any], email: bool = False
+):
+    client.chat_postMessage(channel=user_id, text=message)
+    if email:
+        email_addr = client.users_info(user=user_id)["user"]["profile"]["email"]
+        env.mailer.send_email(
+            email_addr, f"{event["fields"]["Title"]} Reminder!", message
+        )
+
+
+def check_rsvps():
+    events = env.airtable.get_all_events()
+
+    for event in events:
+        if not event["fields"].get("Approved", False):
+            continue
+        start_time = datetime.fromisoformat(event["fields"]["Start Time"]).timestamp()
+        rsvps = env.airtable.get_rsvps_from_event(event["id"])
+
+        # Handle 1 day reminders
+        if start_time - time.time() <= 86400 and not event["fields"].get(
+            "Sent 1 Day Reminder", False
+        ):
+            for user in rsvps:
+                send_reminder(
+                    user["fields"]["Slack ID"],
+                    f"Hey! Just a reminder that {event['fields']['Title']} run by {event['fields']['Leader']} is tomorrow! Hope to see you there!",
+                    event,
+                )
+            env.airtable.update_event(event["id"], **{"Sent 1 Day Reminder": True})
+
+        # Handle 1 hour reminders
+        elif start_time - time.time() <= 3600 and not event["fields"].get(
+            "Sent 1 Hour Reminder", False
+        ):
+            for user in rsvps:
+                send_reminder(
+                    user["fields"]["Slack ID"],
+                    f"Hey! Just a reminder that {event['fields']['Title']} run by {event['fields']['Leader']} starts in 1 hour! Hope to see you there!\nYou can join the event at {event['fields'].get('Event Link', 'the Slack!')}",
+                    event,
+                )
+            env.airtable.update_event(event["id"], **{"Sent 1 Hour Reminder": True})
+
+        elif start_time - time.time() <= 0 and not event["fields"].get(
+            "Sent Started Reminder", False
+        ):
+            for user in rsvps:
+                send_reminder(
+                    user["fields"]["Slack ID"],
+                    f"Hey! Just a reminder that {event['fields']['Title']} run by {event['fields']['Leader']} has started!\nYou can join the event at {event['fields'].get('Event Link', 'the Slack!')}\nHope you enjoy it!",
+                    event,
+                    email=True,
+                )
+            env.airtable.update_event(event["id"], **{"Sent Starting Reminder": True})
+
+
+def rsvp_checker():
+    schedule.every(1).minutes.do(check_rsvps)
+    while True:
+        schedule.run_pending()
+        time.sleep(1)

utils/slack.py

@@ -7,6 +7,7 @@
 from events.views.create_event import handle_create_event_view
 from events.buttons.propose_event import handle_propose_event_btn
 from events.buttons.approve_event import handle_approve_event_btn
+from events.buttons.rsvp import handle_rsvp_btn
 from views.app_home import get_home
 
 from typing import Any, Callable
@@ -51,3 +52,8 @@ def create_event(ack: Callable, body: dict[str, Any], client: WebClient):
 @app.action("add-to-gcal")
 def add_to_gcal(ack: Callable):
     ack()
+
+
+@app.action("rsvp")
+def rsvp(ack: Callable, body: dict[str, Any], client: WebClient):
+    handle_rsvp_btn(ack, body, client)