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()