better logging

Author: davidsklar99

services/usage-stats/src/insert.py

@@ -12,11 +12,11 @@
 # Ensure it uses asyncpg
 if raw_url.startswith("postgresql://"):
     raw_url = raw_url.replace("postgresql://", "postgresql+asyncpg://", 1)
-elif not raw_url.startswith("postgresql+asyncpg://"):
+else:
     raise ValueError(
         "Invalid DATABASE_URL: must start with postgresql:// or postgresql+asyncpg://"
     )
-
+    
 DATABASE_URL = raw_url
 
 engine = create_async_engine(DATABASE_URL, echo=True)

services/usage-stats/src/last_id.py

@@ -12,7 +12,7 @@
 # Ensure it uses asyncpg
 if raw_url.startswith("postgresql://"):
     raw_url = raw_url.replace("postgresql://", "postgresql+asyncpg://", 1)
-elif not raw_url.startswith("postgresql+asyncpg://"):
+else:
     raise ValueError(
         "Invalid DATABASE_URL: must start with postgresql:// or postgresql+asyncpg://"
     )

services/usage-stats/src/macrostrat.py

@@ -14,7 +14,7 @@
 # Ensure the URL uses asyncmy driver
 if raw_url.startswith("mysql://"):
     raw_url = raw_url.replace("mysql://", "mysql+asyncmy://", 1)
-elif not raw_url.startswith("mysql+asyncmy://"):
+else:
     raise ValueError(
         "Invalid DATABASE_URL: must start with mysql:// or mysql+asyncmy://"
     )

services/usage-stats/src/rockd.py

@@ -15,7 +15,7 @@
 # Ensure the URL uses asyncmy driver
 if raw_url.startswith("mysql://"):
     raw_url = raw_url.replace("mysql://", "mysql+asyncmy://", 1)
-elif not raw_url.startswith("mysql+asyncmy://"):
+else:
     raise ValueError(
         "Invalid DATABASE_URL: must start with mysql:// or mysql+asyncmy://"
     )