def addtodb(page, lookupurl, classes): ''' Add page to MongoDB :param page: info about page :type page: dict :param lookupurl: URL of parent site :type lookupurl: str :param classes: result of classification for given page :type classes: array of ints ''' parent = Site.objects(url=lookupurl).first() nov = Page(url=page['url'], btext=page['btext']) if parent: nov.parent = parent for c in classes: label = Label.objects(id=c).first() nov.label_model.append(label) print "label added" nov.save() add_to_lookup(page['url']) print "Page added to db"