def test_register_save():
registry.register('/tmp')
registry.save_registry()
with open(util.get_registry_filename(), 'r') as file:
_registry = load(file, Loader=Loader) or default
assert 'projects' in _registry
_projects = _registry['projects']
assert isinstance(_projects, dict)
assert '/tmp' in _projects
registry.unregister('/tmp')
def test_register_save():
registry.register('/tmp')
registry.save_registry()
with open(util.get_registry_filename(), 'r') as file:
_registry = load(file, Loader=Loader) or default
assert 'projects' in _registry
_projects = _registry['projects']
assert isinstance(_projects, dict)
assert '/tmp' in _projects
registry.unregister('/tmp')
def test_register_alias_save():
registry.register('/tmp', 'foobar')
registry.save_registry()
with open(util.get_registry_filename(), 'r') as file:
_registry = load(file, Loader=Loader) or default
assert 'projects' in _registry
_projects = _registry['projects']
assert isinstance(_projects, dict)
assert '/tmp' in _projects
assert 'alias' in _projects['/tmp']
assert 'foobar' == _projects['/tmp']['alias']
def test_register_alias_save():
registry.register('/tmp', 'foobar')
registry.save_registry()
with open(util.get_registry_filename(), 'r') as file:
_registry = load(file, Loader=Loader) or default
assert 'projects' in _registry
_projects = _registry['projects']
assert isinstance(_projects, dict)
assert '/tmp' in _projects
assert 'alias' in _projects['/tmp']
assert 'foobar' == _projects['/tmp']['alias']
def execute(self):
# pdb.set_trace()
name = self.options.name
project = getattr(self.options, 'project', None)
if project:
proj_path = registry.pathfor(project)
if not proj_path:
print "* Project", project, "not found."
return
elif self.root:
proj_path = self.root
else:
print "Must provide -p/--project option, if outside of pin project."
return
if name in registry._aliases:
if proj_path != registry._aliases[name]:
print "Alias `{alias}' already exists for other project:".format(
alias=name)
print registry._aliases[name]
return
else:
print "Alias `{alias}' already set.".format(alias=name)
return
else:
current_alias = registry._projects[proj_path].get('alias')
if current_alias:
prompt = "Alias `{alias}' already set, overwrite?".format(
alias=current_alias)
answer = option_select(['y', 'n'], prompt)
if answer == 'n':
print "Aborted."
return
_projects[proj_path]['alias'] = name
_aliases[name] = proj_path
registry.save_registry()
print "Alias `{alias}' has been set for, {path}".format(
alias=name, path=proj_path)
return True
def execute(self):
# pdb.set_trace()
name = self.options.name
project = getattr(self.options, 'project', None)
if project:
proj_path = registry.pathfor(project)
if not proj_path:
print "* Project", project, "not found."
return
elif self.root:
proj_path = self.root
else:
print "Must provide -p/--project option, if outside of pin project."
return
if name in registry._aliases:
if proj_path != registry._aliases[name]:
print "Alias `{alias}' already exists for other project:".format(alias=name)
print registry._aliases[name]
return
else:
print "Alias `{alias}' already set.".format(alias=name)
return
else:
current_alias = registry._projects[proj_path].get('alias')
if current_alias:
prompt = "Alias `{alias}' already set, overwrite?".format(alias=current_alias)
answer = option_select(['y','n'], prompt)
if answer == 'n':
print "Aborted."
return
_projects[proj_path]['alias'] = name
_aliases[name] = proj_path
registry.save_registry()
print "Alias `{alias}' has been set for, {path}".format(alias=name,
path=proj_path)
return True
