campaignsConfiguration.py

from utils import mongo_db_url
import optparse
import ssl,pymongo
import json
import sys

parser = optparse.OptionParser()
parser.add_option('--dump',help="dump the whole content in this file",default=None)
parser.add_option('--load',help="synchronize the db with the content of this file", default=None)
parser.add_option('-n','--name')
parser.add_option('-c','--configuration')
parser.add_option('-p','--parameter')
(options,args) = parser.parse_args()


client = pymongo.MongoClient('mongodb://%s/?ssl=true'%mongo_db_url,
                             ssl_cert_reqs=ssl.CERT_NONE)
db = client.unified.campaignsConfiguration

if options.load:
    content = json.loads( open(options.load).read())
    for k,v in content.items():
        up = {'name' : k}
        s = {"$set": v}
        db.update( up, s )
        #up.update( v)
        #db.insert_one( up )
        print k,v
    sys.exit(0)
if options.dump:
    uc = {}
    for content in db.find():
        content.pop("_id")
        uc[content.pop("name")] = content

    open(options.dump,'w').write(json.dumps( uc, indent =2))
    sys.exit(0)


post = {}
if options.configuration:
    post.update(json.loads(options.value))
if options.parameter:
    name,value = options.parameter.split(':',1)
    path = reversed(name.split('.'))
    content = {}
    w = post
    for p in path:
        w[p] = {}
found = db.find_one({"name":options.name})
if found:
    if post:
        ## need to update a value
        up = {'_id':found['_id']}
        s = {"$set": post}
        print "updating",options.name,"with values",post
        db.update( up, s )
    else:
        ## use that to show the value in the database
        # not other headers in the output, so that it can be json loadable
        found.pop('name')
        found.pop('_id')
        print json.dumps(found, indent=2)
else:
    if post:
        ## entering a new value
        post.update( {"name":options.name})
        db.insert_one( post )
    else:
        availables = [o["name"] for o in db.find()]
        print options.name," Not found. Available parameters \n","\n\t".join( sorted( availables))