[ez] make all connections to the database allow DISABLE_SSL (#297)

* [ez] make all connections to the database allow DISABLE_SSL

* lint

Author: PaliC

scripts/flush_db.py

@@ -1,26 +1,27 @@
 #!/usr/bin/env python3
 
-import os
 
 import psycopg2
 from dotenv import load_dotenv
+from env import DATABASE_URL, DISABLE_SSL
 from psycopg2 import Error
 
 
 def flush_database():
     # Load environment variables
     load_dotenv()
 
-    DATABASE_URL = os.getenv("DATABASE_URL")
-
     if DATABASE_URL is None:
         print("❌ Missing DATABASE_URL environment variable")
         return
 
     try:
         # Connect to database
         print("📡 Connecting to database...")
-        connection = psycopg2.connect(DATABASE_URL, sslmode="require")
+        connection = psycopg2.connect(
+            DATABASE_URL,
+            sslmode="disable" if DISABLE_SSL else "require"
+        )
         cursor = connection.cursor()
 
         # Drop existing tables

src/discord-cluster-manager/cogs/misc_cog.py

@@ -5,7 +5,7 @@
 import psycopg2
 from discord import app_commands
 from discord.ext import commands
-from env import DATABASE_URL
+from env import DATABASE_URL, DISABLE_SSL
 from utils import send_discord_message, setup_logging
 
 if TYPE_CHECKING:
@@ -33,7 +33,8 @@ async def verify_db(self, interaction: discord.Interaction):
             return
 
         try:
-            with psycopg2.connect(DATABASE_URL, sslmode="require") as conn:
+            sslmode = "disable" if DISABLE_SSL else "require"
+            with psycopg2.connect(DATABASE_URL, sslmode=sslmode) as conn:
                 with conn.cursor() as cursor:
                     cursor.execute("SELECT RANDOM()")
                     result = cursor.fetchone()