आपको डेटाबेस को सीधे लाने की आवश्यकता है, क्योंकि यह आपकी विधि में पारित हो गया है:
from pymongo import MongoClient
client = MongoClient('localhost:27017')
def my_function(mydb):
db = client.get_database(mydb)
return db.collection.find().count()
print(my_function('my_database'))