from __future__ import annotations

import os
from pathlib import Path

from dotenv import load_dotenv

from shared.docstore import connect_mongodb, count_documents

if __name__ == '__main__':
    # prepare env vars
    env_path = Path(__file__).parent.parent / 'local.env'
    assert env_path.exists()
    load_dotenv(env_path)
    os.environ['MONGO_HOST'] = 'localhost'
    # connect to database
    collection, db, client = connect_mongodb()
    # get visual communication
    num_docs = count_documents(
        collection=collection,
        only_with_annotation=False,
    )
    print(f"total number of documents in database: {num_docs}")