Beispiel #1
0
def setup(rootdir):
    """
    Attempt to configure environment by locating 'bootstrap.py' in ``rootdir``
    or by searching parent folders relative to ``rootdir``.
    """
    rootdir = os.path.abspath(rootdir)
    if os.path.isfile(os.path.join(rootdir, "bootstrap.py")):
        sys.path.append(rootdir)
        import bootstrap
        bootstrap.setup(rootdir, "hgt.log")
    else:
        head, tail = os.path.split(rootdir)
        if head and tail:
            setup(head)
        else:
            raise RuntimeError("Could not locate bootstrap.py")
Beispiel #2
0
def load_from_file(path, **kw):
	
	try:

		kw = bootstrap.setup(**kw)

	except Exception as e:

		raise e

	else:

		# initialize the redis connection
		redis_conn = kw.get("redis_conn")

		# first unpickle the file into a list of objects
		dump_data = pickle.load(open(path, "rb"))	

		for key in dump_data:

			load_key(key, redis_conn)
			
		bootstrap.breakdown(**kw)
Beispiel #3
0
def dump(**kw):

	try:
		# set up the connection wit the passed parameters
		kw = bootstrap.setup(**kw)
	
	except Exception as e:

		raise e
	else:

		redis_conn = kw.get("redis_conn")	

		dump_data = []

		# now loop through each key and remove from the database
		for key in redis_conn.keys():

			dump_data.append(get_key(key, redis_conn))

		# breakdown  test environment here
		bootstrap.breakdown(**kw)

		return dump_data