import csv
import os
import random
import pandas as pd
from collections import defaultdict
from pymongo import MongoClient
import numpy as np
from tqdm import tqdm

client = MongoClient('mongodb+srv://miami:miami6033626@cluster0.lodffed.mongodb.net/?retryWrites=true&w=majority')
db = client['miami-database']
csv_collection = db['csv']
record_collection = db['record']
csv_list = [f'list_{str(i + 1)}.csv' for i in range(697)]

query0 = record_collection.count_documents({"count": {'$eq': 0}})
query1 = record_collection.count_documents({"count": {'$eq': 1}})
query2 = record_collection.count_documents({"count": {'$eq': 2}})
query3 = record_collection.count_documents({"count": {'$eq': 3}})
query4 = record_collection.count_documents({"count": {'$eq': 4}})
query5 = record_collection.count_documents({"count": {'$eq': 5}})
query6 = record_collection.count_documents({"count": {'$eq': 6}})
query7 = record_collection.count_documents({"count": {'$eq': 7}})
query8 = record_collection.count_documents({"count": {'$eq': 8}})
query9 = record_collection.count_documents({"count": {'$eq': 9}})
print(query0 * 10 + query1 * 9 + query2 * 8 + query3 * 7 +
      query4 * 6 + query5 * 5 + query6 * 4 + query7 * 3 + query8 * 2 + query9 * 1)

# print(query0 * 9 + query1 * 8 + query2 * 7 + query3 * 6 +
#       query4 * 5 + query5 * 4 + query6 * 3 + query7 * 2 + query8 * 1)