import sys, json, os, traceback
sys.path.insert(0, r'F:\projects\gogo2')
res={}
try:
    from utils.database_manager import get_database_manager
    db=get_database_manager()
    def db_count(name):
        try:
            lst = db.list_checkpoints(name)
            return len(lst) if lst is not None else 0
        except Exception as e:
            print("DB error for %s: %s" % (name, str(e)))
            return -1
    res.setdefault('db', {})['dqn_agent']=db_count('dqn_agent')
    res['db']['enhanced_cnn']=db_count('enhanced_cnn')
except Exception as e:
    res['db']={'error': str(e)}
try:
    from utils.checkpoint_manager import get_checkpoint_manager
    cm=get_checkpoint_manager()
    def fs_count(name):
        try:
            lst = cm.get_all_checkpoints(name)
            return len(lst) if lst is not None else 0
        except Exception as e:
            print("FS error for %s: %s" % (name, str(e)))
            return -1
    res.setdefault('fs', {})['dqn_agent']=fs_count('dqn_agent')
    res['fs']['enhanced_cnn']=fs_count('enhanced_cnn')
except Exception as e:
    res['fs']={'error': str(e)}
print(json.dumps(res))