entrouvert
/
misc-fred
17
0
Fork 0
Code Issues Pull Requests Releases Activity

eoptasks: add configuration file/do not hardcode my path

This commit is contained in:
Frédéric Péters 2018-12-12 08:28:44 +01:00
parent 7e808fb2ae
commit 969f9277be
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
eoptasks/eoptasks.py
View File

@ -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))