def ready(self): from indigo.models import initialise, Collection, User, Group logging.basicConfig(level=logging.WARNING) logging.getLogger("models").setLevel(logging.WARNING) logging.getLogger("dse.policies").setLevel(logging.WARNING) logging.getLogger("dse.cluster").setLevel(logging.WARNING) logging.getLogger("dse.cqlengine.management").setLevel(logging.WARNING) cfg = get_config(None) initialise(keyspace=cfg.get('KEYSPACE', 'indigo'), hosts=cfg.get('CASSANDRA_HOSTS', ('127.0.0.1', ))) # Try to get the root. It will create it if it doesn't exist root = Collection.get_root() # TODO: Review that at some point # Check that the graph vertices for users are still there User.check_graph_users()
def main(): __doc__ = args_doc cfg = load_config() ############################ might work args = sys.argv[1:] ##### Test fragment ... remove ## TODO: remove test fragment if len(sys.argv) < 3: args = 'inject --copy --postgres --prefix=/Users/johnburns/PycharmProjects --dataset=resource'.strip().split() args = '''prepare --walk --postgres --prefix=/Users/johnburns/PycharmProjects /Users/johnburns/PycharmProjects'''.split() ### End Remove try : args = docopt.docopt(args_doc_tidied, args) except docopt.DocoptExit as e : print args_doc.encode('utf-8'), print help_text.encode('utf-8') print 'Command Line:\n' + (' '.join(args)) raise except docopt.DocoptLanguageError as e : print e raise if args['prepare']: return prepare(args, cfg) if args['summary']: raise NotImplementedError summary(args, cfg) else: try: cassandra_ip = cfg.get('cassandra', 'CASSANDRA_HOSTS') if cassandra_ip: cassandra_ip = re.split(r'[^\w\.]+',cassandra_ip ) except: cassandra_ip = ['127.0.0.1'] initialise(keyspace='indigo', hosts=cassandra_ip) if args['inject']: return inject(args, cfg) if args['validate']: raise NotImplementedError return validate(args, cfg) print >>sys.stderr, my_arg_doc, "\n Unknown subcommand \n", sys.argv return None
def setUp(self): initialise()
def setup_package(): initialise(keyspace=TEST_KEYSPACE, strategy="SimpleStrategy", repl_factor=1) sync()
def __init__(self): self.terminal = Terminal() initialise()