Exemplo n.º 1
0
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"