Subversion Repositories SmartDukaan

import pymongo
import os
import json
from xml.etree.ElementTree import parse, Element
from xml import etree

con=None

filePath="/var/lib/tomcat6/webapps/export/javascripts/autosuggesthotspot.json"
solrPath = "/var/lib/tomcat6/webapps/export/solr/"

def get_mongo_connection(host='localhost', port=27017):
    global con
    if con is None:
        print "Establishing connection %s host and port %d" %(host,port)
        try:
            con = pymongo.MongoClient(host, port)
        except Exception, e:
            print e
            return None
    return con

def generate_auto_suggest():
    PATH = os.path.expanduser(filePath)
    if os.path.isfile(PATH):
        try:
            os.remove(os.path.expanduser(filePath))
        except OSError:
            pass
    output = open(filePath, 'ab+')
    all = get_mongo_connection().Hotspot.HotspotEntity.find()
    autoSuggestMap = {}
    for a in all:
        autoSuggestMap[a['title']] = [a['title']]
    json.dump(autoSuggestMap, output)

def alter_solr_xml():
    all = get_mongo_connection().Hotspot.HotspotEntity.find()
    for a in all:
        fileName = str(a['_id'])+"_irdata_solr.xml"
        doc = parse(solrPath+fileName)
        root = doc.find('doc')
        co = Element("field", name="P_120537")
        co.text="Yes"
        root.append(co)
        doc.write(solrPath+fileName)
    
def main():
    generate_auto_suggest()
    alter_solr_xml()

if __name__ == '__main__':
    main()