diff --git a/eoptasks/eoptasks.py b/eoptasks/eoptasks.py index 2fe08aa..39707fd 100755 --- a/eoptasks/eoptasks.py +++ b/eoptasks/eoptasks.py @@ -15,6 +15,7 @@ # Requirements: libtmux and pyyaml. import argparse +import configparser import os import random import re @@ -41,7 +42,17 @@ class Server: servers = [] -servergroups = yaml.load(open('/home/fred/src/eo/puppet/data/servergroups.yaml'))['servergroups'] +config = configparser.ConfigParser() +config.read(os.path.join(os.path.expanduser('~/.config/eoptasks.ini'))) +servergroup = config.get('config', 'servergroups', fallback=None) +if servergroup is None: + print("You need to create ~/.config/eoptasks.ini with such a content:\n" + "\n" + " [config]\n" + " servergroups = /home/user/src/puppet/data/servergroups.yaml\n") + sys.exit(1) + +servergroups = yaml.load(open(servergroup))['servergroups'] for group in servergroups: for servername in servergroups[group]: servers.append(Server(servername, group))