eoptasks: use servers.yaml instead of deprecated servergroups.yaml
This commit is contained in:
parent
e1dda1cc35
commit
3f2f478631
|
@ -16,7 +16,7 @@
|
||||||
#
|
#
|
||||||
# Configuration: ~/.config/eoptasks.ini
|
# Configuration: ~/.config/eoptasks.ini
|
||||||
# [config]
|
# [config]
|
||||||
# servergroups = /path/to/servergroups.yaml
|
# servers = /path/to/servers.yaml
|
||||||
# ignore = server1, server2
|
# ignore = server1, server2
|
||||||
# stripsuffix = .entrouvert.org
|
# stripsuffix = .entrouvert.org
|
||||||
#
|
#
|
||||||
|
@ -51,12 +51,14 @@ import time
|
||||||
import libtmux
|
import libtmux
|
||||||
import yaml
|
import yaml
|
||||||
|
|
||||||
|
|
||||||
class Server:
|
class Server:
|
||||||
def __init__(self, servername, group='', display_name=''):
|
def __init__(self, servername, tags=[], display_name=''):
|
||||||
self.name = servername
|
self.name = servername
|
||||||
self.display_name = display_name or self.name
|
self.display_name = display_name or self.name
|
||||||
self.keywords = set(re.split(r'[-_ \.]', servername + ' ' + group))
|
self.keywords = set(re.split(r'[-_ \.]', servername))
|
||||||
self.keywords.add(group)
|
for tag in tags:
|
||||||
|
self.keywords.add(tag)
|
||||||
|
|
||||||
# add all possible hostname parts as keywords,
|
# add all possible hostname parts as keywords,
|
||||||
# ex: node1.dev.entrouvert.org will add:
|
# ex: node1.dev.entrouvert.org will add:
|
||||||
|
@ -76,17 +78,16 @@ def get_servers():
|
||||||
servers = []
|
servers = []
|
||||||
config = configparser.ConfigParser()
|
config = configparser.ConfigParser()
|
||||||
config.read(os.path.join(os.path.expanduser('~/.config/eoptasks.ini')))
|
config.read(os.path.join(os.path.expanduser('~/.config/eoptasks.ini')))
|
||||||
servergroup = config.get('config', 'servergroups', fallback=None)
|
serversfile = config.get('config', 'servers', fallback=None)
|
||||||
if servergroup is None:
|
if serversfile is None:
|
||||||
print("You need to create ~/.config/eoptasks.ini with such a content:\n"
|
print("You need to create ~/.config/eoptasks.ini with such a content:\n"
|
||||||
"\n"
|
"\n"
|
||||||
" [config]\n"
|
" [config]\n"
|
||||||
" servergroups = /home/user/src/puppet/data/servergroups.yaml\n")
|
" servers = /home/user/src/puppet/data/servers.yaml\n")
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
ignorelist = [x.strip() for x in config.get('config', 'ignore', fallback='').split(',')]
|
ignorelist = [x.strip() for x in config.get('config', 'ignore', fallback='').split(',')]
|
||||||
|
|
||||||
stripsuffixes = [x.strip() for x in config.get('config', 'stripsuffix', fallback='').split(',')]
|
stripsuffixes = [x.strip() for x in config.get('config', 'stripsuffix', fallback='').split(',')]
|
||||||
servergroups = yaml.safe_load(open(servergroup))['servergroups']
|
|
||||||
|
|
||||||
def get_display_name(x):
|
def get_display_name(x):
|
||||||
for stripsuffix in stripsuffixes:
|
for stripsuffix in stripsuffixes:
|
||||||
|
@ -94,13 +95,15 @@ def get_servers():
|
||||||
return x[:-len(stripsuffix)]
|
return x[:-len(stripsuffix)]
|
||||||
return x
|
return x
|
||||||
|
|
||||||
for group in servergroups:
|
for s in yaml.safe_load(open(serversfile))['servers']:
|
||||||
for servername in servergroups[group]:
|
servername, tags = s.get('name'), s.get('tags', [])
|
||||||
if servername in ignorelist:
|
if servername in ignorelist:
|
||||||
continue
|
continue
|
||||||
servers.append(Server(servername, group, display_name=get_display_name(servername)))
|
servers.append(Server(servername, tags, display_name=get_display_name(servername)))
|
||||||
|
|
||||||
return servers
|
return servers
|
||||||
|
|
||||||
|
|
||||||
def parse_args():
|
def parse_args():
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
parser.add_argument('-l', '--list-servers', action='store_true')
|
parser.add_argument('-l', '--list-servers', action='store_true')
|
||||||
|
|
Loading…
Reference in New Issue