20 lines
538 B
Python
20 lines
538 B
Python
from pathlib import Path
|
|
from dotenv import load_dotenv
|
|
import os
|
|
|
|
from src.database import (
|
|
connect,
|
|
total_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()
|
|
# get visual communication
|
|
num_docs = total_documents(collection)
|
|
print(f"total number of documents in database: {num_docs}") |