authentication

Author: emanuelecavalleri

.DS_Store

@@ -0,0 +1,15 @@
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+from sqlalchemy.ext.declarative import declarative_base
+from dotenv import load_dotenv
+import os
+
+load_dotenv()
+
+DATABASE_URL = os.getenv("DATABASE_URL")
+
+engine = create_engine(DATABASE_URL) # Creates the database connection engine
+
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) # Creates a database session factory
+
+Base = declarative_base() # Defines the base class for SQLAlchemy models

database.py

@@ -0,0 +1,119 @@
+import asyncpg
+import asyncio
+import os
+from dotenv import load_dotenv
+from gmail_service import send_email
+from datetime import datetime, timedelta, time
+import pytz
+import logging
+
+local_tz = pytz.timezone("Europe/Rome")
+
+load_dotenv()
+
+DATABASE_URL = os.getenv("DATABASE_URL")
+
+async def handle_notify(connection, pid, channel_name, payload):
+    # Handles notifications when they arrive on the user_approval channel
+    user_email, username, status = payload.split(",")
+    
+    if channel_name == "user_status":
+        if status == "active":
+            subject = "Welcome to SchemaLink! Your account has been approved"
+            message = (
+                f"Hi {username},\n\n"
+                "Great news! Your SchemaLink account has been approved.\n\n"
+                "You have been granted a Trial policy that allows up to 10 intelligent requests "
+                "within the next 24 hours. After this period, your access may be limited unless you upgrade "
+                "to a higher tier.\n\n"
+                "Thank you for joining SchemaLink, and we hope you enjoy using the platform!\n\n"
+                f"Best regards,\n"
+                f"The SchemaLink Team"
+            )
+            now = datetime.now(local_tz)
+            end_date = now + timedelta(hours=24)
+
+            if now.time() <= time(12, 0):  # From 00:00:01 to 12:00:00
+                end_date = (now + timedelta(days=1)).replace(hour=12, minute=0, second=0, microsecond=0)
+            else:   # From 12:00:01 to 23:59:59
+                end_date = (now + timedelta(days=2)).replace(hour=0, minute=0, second=0, microsecond=0)
+            
+            try:
+                await connection.execute("""
+                    INSERT INTO UserSubscribesPolicy (
+                        username, startDate, endDate, requestDate, status, policyName
+                    ) VALUES ($1, $2, $3, $4, $5, $6)
+                """, username, now, end_date, now, 'active', 'trial')
+
+                print(f"Assigned 'trial' policy to user {username}")
+            except Exception as e:
+                print(f"Error assigning policy to user {username}: {e}")
+        elif status == "blocked":
+            subject = "SchemaLink account blocked"
+            message = (
+                f"Hi {username},\n\n"
+                "Your SchemaLink account has been blocked. \n\n"
+                f"Best regards,\n"
+                f"The SchemaLink Team"
+            )
+        elif status == "disabled":
+            subject = "SchemaLink account deleted"
+            message = (
+                f"Hi {username},\n\n"
+                "Your SchemaLink account has been successfully deleted. \n\n"
+                f"Best regards,\n"
+                f"The SchemaLink Team"
+            )
+    elif channel_name == "policy_status":
+        if status == "active":
+            subject = "Your SchemaLink policy has been approved"
+            message = (
+                f"Hi {username},\n\n"
+                "Your policy request has been approved. You can now enjoy the benefits associated with your policy.\n\n"
+                f"Best regards,\n"
+                f"The SchemaLink Team"
+            )
+        elif status == "rejected":
+            subject = "Your SchemaLink policy request was rejected"
+            message = (
+                f"Hi {username},\n\n"
+                "We regret to inform you that your policy request has been rejected. \n\n"
+                f"Best regards,\n"
+                f"The SchemaLink Team"
+            )
+        elif status == "expired":
+            subject = "Your SchemaLink policy has expired"
+            message = (
+                f"Hi {username},\n\n"
+                "Your current SchemaLink policy has expired. To continue enjoying uninterrupted service, "
+                "please renew or upgrade your subscription.\n\n"
+                f"Best regards,\n"
+                f"The SchemaLink Team"
+            )
+
+    # Send the email asynchronously in a separate thread
+    if subject and message:
+        await asyncio.to_thread(send_email, user_email, subject, message)
+
+
+async def listen_notifications():
+    # Listens for user approval notifications and sends emails
+    conn = None 
+
+    try:
+        conn = await asyncpg.connect(DATABASE_URL)
+        # Add the listener to the 'user_status' channel
+        await conn.add_listener("user_status", handle_notify)
+        await conn.add_listener("policy_status", handle_notify)
+        print("Listening on 'user_status' and 'policy_status'...")
+
+        # Keep listening for notifications
+        while True:
+            await asyncio.sleep(60)  # Keep the listener active without blocking
+    except Exception as e:
+        print(f"Error connecting to the database: {e}")
+
+    finally:
+        if conn:
+            await conn.close()
+            print("Database connection closed.")

email_listener.py

@@ -0,0 +1,28 @@
+from apscheduler.schedulers.background import BackgroundScheduler
+from sqlalchemy.orm import Session
+from datetime import datetime
+import pytz
+import models
+from database import SessionLocal
+
+def expire_subscriptions_job():
+    print("Running scheduled job...")
+
+    db: Session = SessionLocal()
+
+    try:
+        local_tz = pytz.timezone("Europe/Rome")
+        now = datetime.now(local_tz)
+
+        # Data expiration
+        expired = db.query(models.UserSubscribesPolicy).filter(
+            models.UserSubscribesPolicy.status == 'active',
+            models.UserSubscribesPolicy.endDate <= now
+        ).update({"status": "expired"}, synchronize_session=False)
+
+        db.commit()
+        print(f"Marked {expired} subscriptions as expired based on date.")
+    except Exception as e:
+        print(f"Error during scheduled job: {e}")
+    finally:
+        db.close()

expire_subscriptions_job.py

@@ -0,0 +1,40 @@
+import smtplib
+from email.mime.text import MIMEText
+from email.mime.multipart import MIMEMultipart
+from email.mime.application import MIMEApplication
+from dotenv import load_dotenv
+import os
+
+current_window = None
+
+load_dotenv() 
+
+EMAIL_ADDRESS = 'schemalinkanacleto@gmail.com'
+EMAIL_PASSWORD = os.getenv("EMAIL_PASSWORD")
+SMTP_SERVER = 'smtp.gmail.com'
+SMTP_PORT = 587
+
+def send_email(to_email, subject, message, attachment=None):
+    try:
+        msg = MIMEMultipart()
+        msg['From'] = EMAIL_ADDRESS
+        msg['To'] = to_email
+        msg['Subject'] = subject
+
+        msg.attach(MIMEText(message, 'plain'))
+        
+        if attachment:
+            with open(attachment, 'rb') as f:
+                part = MIMEApplication(f.read(), Name=os.path.basename(attachment))
+                part['Content-Disposition'] = f'attachment; filename="{os.path.basename(attachment)}"'
+                msg.attach(part)
+
+        server = smtplib.SMTP(SMTP_SERVER, SMTP_PORT)
+        server.ehlo()
+        server.starttls()
+        server.login(EMAIL_ADDRESS, EMAIL_PASSWORD)
+        server.sendmail(EMAIL_ADDRESS, to_email, msg.as_string())
+        server.quit()
+        return True
+    except Exception as e:
+        return str(e)