zajebis

src/leaderboard.py

@@ -0,0 +1,48 @@
+from flask_sqlalchemy import SQLAlchemy
+from flask import g
+import os
+import sqlite3
+from pathlib import Path
+
+def get_db():
+    if 'db' not in g:
+        db_path = Path(__file__).parent / 'database' / 'db.sqlite3'
+        db_path.parent.mkdir(exist_ok=True)  # Ensure /database folder exists
+        g.db = sqlite3.connect(db_path)
+    return g.db
+
+def create_leaderboard_table():
+    db = get_db()
+    cursor = db.cursor()
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS leaderboard (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            username TEXT,
+            problem_id INTEGER,
+            runtime REAL,
+            memory_used INTEGER,
+            line_number INTEGER,
+            timestamp DATETIME DEFAULT CURRENT_TIMESTAMP
+        )
+    ''')
+    db.commit()
+
+def log_leaderboard(username, problem_id, runtime, memory_used, line_number):
+    db = get_db()
+    cursor = db.cursor()
+    cursor.execute('''
+        INSERT INTO leaderboard (username, problem_id, runtime, memory_used, line_number)
+        VALUES (?, ?, ?, ?, ?)
+    ''', (username, problem_id, runtime, memory_used, line_number))
+    db.commit()
+
+def get_leaderboard():
+    db = get_db()
+    cursor = db.cursor()
+    cursor.execute('''
+        SELECT username, problem_id, runtime, memory_used, line_number, timestamp
+        FROM leaderboard
+        ORDER BY runtime ASC, memory_used ASC
+        LIMIT 20
+    ''')
+    return cursor.fetchall()