tinygrad/extra/dump_cache.py

import sys, sqlite3, pickle
from tinygrad.helpers import CACHEDB

if __name__ == "__main__":
  fn = sys.argv[1] if len(sys.argv) > 1 else CACHEDB
  conn = sqlite3.connect(fn)
  cur = conn.cursor()
  cur.execute("SELECT name FROM sqlite_master WHERE type='table'")
  for f in cur.fetchall():
    table = f[0]
    cur2 = conn.cursor()
    cur2.execute(f"SELECT COUNT(*) FROM {table}")
    cnt = cur2.fetchone()[0]
    print(f"{table:20s} : {cnt}")

    cur3 = conn.cursor()
    cur3.execute(f"SELECT * FROM {table} LIMIT 10")
    for f in cur3.fetchall():
      v = pickle.loads(f[-1])
      print("   ", len(f[0]) if isinstance(f[0], str) else f[0], f[1:-1], str(v)[0:50])
      #print(f"{len(k):10d}, {sk} -> {v}")