24 lines
645 B
Python
24 lines
645 B
Python
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}")
|