consumer_details={} consumer_details['js_consumer_key'] = args.js_consumer_key if args.js_consumer_key != None else None consumer_details['all_consumer_key'] = args.all_consumer_key if args.all_consumer_key != None else None consumer_details['all_consumer_secret'] = args.all_consumer_secret if args.all_consumer_secret != None else None filename = os.path.join(dir, args.props) if not os.path.exists(filename): print "Properties file doesn't exist at", filename, ", please create it before running this script" exit(1) with open(filename) as data_file: data = json.load(data_file) lowercase_clients(data) for section in requiredSections: if not section in data: print "Must have section",section,"in config JSON" exit(1) zk = KazooClient(hosts=args.zookeeper) zk.start() try: seldon.dbSetup(zk,data["db"],propertyToZkNode["db"]) seldon.memcachedSetup(zk,data["memcached"],propertyToZkNode["memcached"]) seldon.clientSetup(zk,data["clients"],data["db"],propertyToZkNode["clients"], consumer_details) print "Finished succesfully" except KeyError as e: print "Property missing from config json:",e traceback.print_exc()
#!/usr/bin/env python import argparse, os, json import seldon_utils as seldon from kazoo.client import KazooClient dir = os.path.dirname(os.path.abspath(__file__)) parser = argparse.ArgumentParser(prog="add_client", description="Adds a new client to the Seldon Server") parser.add_argument("client",nargs=1,help="The client to add") parser.add_argument("--json",help="extra JSON configuration for the client") parser.add_argument("--props",help="Relative path to the file with the props", default='../server_config.json') parser.add_argument("--db",help="The name of the DB to use (from the config file), default is to use the first one mentioned.") parser.add_argument("--zookeeper",help="Location of zookeeper (hosts)", default="localhost") args = parser.parse_args() filename = os.path.join(dir, args.props) zk = KazooClient(hosts=args.zookeeper) zk.start() if not os.path.exists(filename): print "Properties file doesn't exist at", filename, ", please create it before running this script" exit(1) with open(filename) as data_file: data = json.load(data_file) seldon.clientSetup(zk,[{"name":args.client[0],"db":args.db}],data["db"],"/all_clients") print "Finished successfully"
dir = os.path.dirname(os.path.abspath(__file__)) parser = argparse.ArgumentParser(prog="PROG",description="Sets up the Seldon Server") parser.add_argument("--props",help="Relative path to the file with the props", default='../server_config.json') parser.add_argument("--zookeeper",help="Location of zookeeper (hosts)", default="localhost") args = parser.parse_args() filename = os.path.join(dir, args.props) if not os.path.exists(filename): print "Properties file doesn't exist at", filename, ", please create it before running this script" exit(1) with open(filename) as data_file: data = json.load(data_file) for section in requiredSections: if not section in data: print "Must have section",section,"in config JSON" exit(1) zk = KazooClient(hosts=args.zookeeper) zk.start() try: seldon.dbSetup(zk,data["db"],propertyToZkNode["db"]) seldon.memcachedSetup(zk,data["memcached"],propertyToZkNode["memcached"]) seldon.clientSetup(zk,data["clients"],data["db"],propertyToZkNode["clients"]) print "Finished succesfully" except KeyError as e: print "Property missing from config json:",e traceback.print_exc()
parser = argparse.ArgumentParser( prog="add_client", description="Adds a new client to the Seldon Server") parser.add_argument("client", nargs=1, help="The client to add") parser.add_argument("--json", help="extra JSON configuration for the client") parser.add_argument("--props", help="Relative path to the file with the props", default='../server_config.json') parser.add_argument( "--db", help= "The name of the DB to use (from the config file), default is to use the first one mentioned." ) parser.add_argument("--zookeeper", help="Location of zookeeper (hosts)", default="localhost") args = parser.parse_args() filename = os.path.join(dir, args.props) zk = KazooClient(hosts=args.zookeeper) zk.start() if not os.path.exists(filename): print "Properties file doesn't exist at", filename, ", please create it before running this script" exit(1) with open(filename) as data_file: data = json.load(data_file) seldon.clientSetup(zk, [{ "name": args.client[0], "db": args.db }], data["db"], "/all_clients") print "Finished successfully"