import pprint
from pymongo import MongoClient
import datetime


password = 'XxXXxxxxXXXxxx'
connection_string = f'mongodb+srv://yourusername:{password}@cluster0.abcdef.mongodb.net/test'

client = MongoClient(connection_string)

# reading ----------------------------------------------------

# making print output nicer
printer = pprint.PrettyPrinter()


# get all databases
def get_databases(value='full'):
    if value == 'names':
        dbs_names = client.list_database_names()
        return dbs_names

    else:
        dbs = client.list_databases()
        return [i for i in dbs]


def get_database_collections(db_name):
    collections = db_name.list_collection_names()
    return collections


def get_document_in_a_collection(db_name, collection_name):
    collection = db_name[collection_name]
    cursor = collection.find({})
    for i in cursor:
        printer.pprint(i)


def get_document_by_id(db_name, collection_name, doc_id):
    # from bson.objectid import ObjectId
    #
    # _id = ObjectId(doc_id)
    document = db_name[collection_name].find_one({"_id": doc_id})
    printer.pprint(document)


def get_document_by_object_id(db_name, collection_name, doc_id):
    from bson.objectid import ObjectId
    _id = ObjectId(doc_id)
    document = db_name[collection_name].find_one({"_id": _id})
    printer.pprint(document)


def get_project_columns(db_name, collection_name, columns):
    document = db_name[collection_name].find({}, columns)
    for i in document:
        print(i)
        

def get_age_range(db_name, collection_name, min_date, max_date):
    query = {"$and": [
        {"birthdate": {"$gte": min_date}},
        {"birthdate": {"$lte": max_date}}
    ]}

    document = db_name[collection_name].find(query).sort("birthdate")
    for i in document:
        printer.pprint(i)



# # # TEST

# # print just names as list
# print(get_databases('names'))
#
# # print collections inside top10board database
# print(get_database_collections(client.top10board))
#
# # print values in a collection
# get_document_in_a_collection(client.top10board, 'items')
#
# # print document by id (which is manually generated)
# get_document_by_id(client.top10board, 'items', 1)

# # print document by id (with ObjectId property)
# get_document_by_object_id(client.sample_analytics, 'customers', '5ca4bbcea2dd94ee58162a68')

# # printing values only by username, name as specified. 1 means show.
# # id=0 means don't print the objectId / ID otherwise it will be shown by default.
# get_project_columns(client.sample_analytics, 'customers', {"_id": 0, "username": 1, "name": 1})

# find between years 1990 ~ 1991 dates
get_age_range(client.sample_analytics, 'customers', datetime.datetime(1990, 1, 1, 0, 0, 0), datetime.datetime(1991, 1, 1, 0, 0, 0))