database.py

import glob
import json
import re
import xlwt
import uuid
import os
import pandas as pd
from pymongo import MongoClient
count=1

#Mongo Collection creation

mng_client=MongoClient()
db_cm=mng_client.project
collection=db_cm.dataset_test
#coll = mng_client["Project"]["Dataset"]

## Adding rows to Mongo Collection
docs=[]
genre=["action","comedy","drama","horror"]
data=pd.DataFrame(columns=list(['index','name','genre','format']))
prevdir = os.getcwd()
for g in genre:
    path="/Users/administrator/PDFS/MachineLearning/december/movie trailer/midterm_movie_trailer/data_final/train/"+g

    print path
    os.chdir(path)
    for filename in ("*.mp4", "*.webm"):
        title=glob.glob(filename)
        format1=filename.split(".")[1]
        for i in range(0,(len(title))):
            title[(i)] = unicode(title[(i)], errors='ignore')
            data.loc[count]=[int(count),title[(i)],g,format1]
            count=int(count+1)
    os.chdir(prevdir)
print data
data_json=json.loads(data.to_json(orient='records'))

# collection.insert_many(data.to_dict(orient='records'))
# print(list(collection.find()))