ConceptARC db upload

scripts/check_arc_puzzles.py

@@ -0,0 +1,83 @@
+#!/usr/bin/env python3
+"""
+Check arc_puzzles table for ConceptArc entries
+"""
+
+import os
+import pymysql
+from dotenv import load_dotenv
+
+def load_env_config():
+    """Load database configuration from .env file"""
+    load_dotenv()
+    return {
+        'host': os.getenv('DB_HOST'),
+        'user': os.getenv('DB_USER'),
+        'password': os.getenv('DB_PASSWORD'),
+        'database': os.getenv('DB_NAME'),
+        'port': int(os.getenv('DB_PORT', 3306)),
+        'charset': 'utf8mb4'
+    }
+
+def main():
+    print("Checking arc_puzzles table")
+    print("=" * 50)
+
+    config = load_env_config()
+    connection = pymysql.connect(**config)
+    cursor = connection.cursor()
+
+    # Check total counts by corpora
+    cursor.execute("""
+        SELECT corpora, COUNT(*) as count
+        FROM arc_puzzles
+        GROUP BY corpora
+        ORDER BY corpora
+    """)
+
+    print("\nTotal puzzles by corpora:")
+    for corpora, count in cursor.fetchall():
+        print(f"  {corpora}: {count}")
+
+    # Check ConceptArc entries in detail
+    cursor.execute("""
+        SELECT id, corpora, `set`, difficulty, version
+        FROM arc_puzzles
+        WHERE corpora = 'ConceptArc'
+        ORDER BY `set`, id
+        LIMIT 20
+    """)
+
+    print("\n" + "=" * 50)
+    print("First 20 ConceptArc entries in arc_puzzles:")
+    print("=" * 50)
+    print(f"{'ID':<20} {'Corpora':<15} {'Set':<20} {'Difficulty':<12} {'Version':<10}")
+    print("-" * 80)
+
+    rows = cursor.fetchall()
+    if rows:
+        for row in rows:
+            puzzle_id, corpora, set_name, difficulty, version = row
+            print(f"{puzzle_id:<20} {corpora:<15} {set_name:<20} {str(difficulty):<12} {str(version):<10}")
+    else:
+        print("No ConceptArc entries found!")
+
+    # Count by set
+    cursor.execute("""
+        SELECT `set`, COUNT(*) as count
+        FROM arc_puzzles
+        WHERE corpora = 'ConceptArc'
+        GROUP BY `set`
+        ORDER BY `set`
+    """)
+
+    print("\n" + "=" * 50)
+    print("ConceptArc puzzles by category:")
+    print("=" * 50)
+    for set_name, count in cursor.fetchall():
+        print(f"  {set_name}: {count}")
+
+    connection.close()
+
+if __name__ == '__main__':
+    main()