From 969f9277be757a2fd65d3c707819bf2b5c446f83 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fr=C3=A9d=C3=A9ric=20P=C3=A9ters?= Date: Wed, 12 Dec 2018 08:28:44 +0100 Subject: [PATCH] eoptasks: add configuration file/do not hardcode my path --- eoptasks/eoptasks.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) 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))