gg

ai-engine/test_db.py

@@ -0,0 +1,7 @@
+import os, psycopg2
+from dotenv import load_dotenv
+load_dotenv('/Users/piton/Documents/Suggest-Bet-BE/.env')
+conn = psycopg2.connect(os.getenv('DATABASE_URL').split('?')[0])
+cur = conn.cursor()
+cur.execute('SELECT mpe.match_id, SUM(CASE WHEN event_type::text LIKE \'%yellow_card%\' THEN 1 WHEN event_type::text LIKE \'%red_card%\' THEN 2 ELSE 1 END) as cards FROM match_player_events mpe WHERE event_type::text LIKE \'%card%\' GROUP BY mpe.match_id LIMIT 5')
+print(cur.fetchall())