A simple mongoimport for importing csv files with python and pymongo

pandas_labeled_csv_import_to_mongo.py

import pandas as pd
from pymongo import MongoClient
import json

def mongoimport(csv_path, db_name, coll_name, db_url='localhost', db_port=27000)
    """ Imports a csv file at path csv_name to a mongo colection
    returns: count of the documants in the new collection
    """
    client = MongoClient(db_url, db_port)
    db = client[db_name]
    coll = db[coll_name]
    data = pd.read_csv(csv_path)
    payload = json.loads(data.to_json(orient='records'))
    coll.remove()
    coll.insert(payload)
    return coll.count()

Perfect ... Just what I needed ... thanks ..

Amazing thanks a lot

my whole header became one and all the data became a line
How to define each header with its specific column, I'm new to programming
Example of how it looks {a, b, c, d: "e, f, g, h"}
As it should be {a: "e", b: "f", c: "g", d: "h"}

Awesome, thank you soo much!!!

.remove() seems to be deprecated

Both remove() and insert() methods are deprecated in mongosh. Check the Compatibility Changes with Legacy mongo Shell for alternative options.

import pandas as pd
from pymongo import MongoClient
import json

def mongoimport(csv_path, db_name, coll_name, db_url='localhost', db_port=27017):
    """ Imports a csv file at path csv_name to a mongo colection
    returns: count of the documants in the new collection
    """
    client = MongoClient(db_url, db_port)
    db = client[db_name]
    coll = db[coll_name]
    data = pd.read_csv(csv_path)
    payload = json.loads(data.to_json(orient='records'))
    if coll_name in db.list_collection_names():
        coll.drop()

    coll.insert_many(payload)
    count = coll.count_documents({})
    
    client.close()
    return count