mirror of
https://github.com/ail-project/ail-framework.git
synced 2024-11-10 08:38:28 +00:00
79 lines
2.7 KiB
Python
Executable file
79 lines
2.7 KiB
Python
Executable file
#!/usr/bin/env python2
|
|
# -*-coding:UTF-8 -*
|
|
|
|
import ConfigParser
|
|
from ConfigParser import RawConfigParser
|
|
import os
|
|
|
|
|
|
def main():
|
|
|
|
configfile = os.path.join(os.environ['AIL_BIN'], 'packages/config.cfg')
|
|
if not os.path.exists(configfile):
|
|
raise Exception('Unable to find the configuration file. \
|
|
Did you set environment variables? \
|
|
Or activate the virtualenv.')
|
|
configfileSample = os.path.join(os.environ['AIL_BIN'], 'packages/config.cfg.sample')
|
|
|
|
cfg = ConfigParser.ConfigParser()
|
|
cfg.read(configfile)
|
|
cfgSample = ConfigParser.ConfigParser()
|
|
cfgSample.read(configfileSample)
|
|
|
|
sections = RawConfigParser.sections(cfg)
|
|
sectionsSample = RawConfigParser.sections(cfgSample)
|
|
|
|
missingSection = []
|
|
dicoMissingSection = {}
|
|
missingItem = []
|
|
dicoMissingItem = {}
|
|
|
|
for sec in sectionsSample:
|
|
if sec not in sections:
|
|
missingSection += [sec]
|
|
dicoMissingSection[sec] = RawConfigParser.items(cfgSample, sec)
|
|
else:
|
|
setSample = set(RawConfigParser.options(cfgSample, sec))
|
|
setNormal = set(RawConfigParser.options(cfg, sec))
|
|
if setSample != setNormal:
|
|
missing_items = list(setSample.difference(setNormal))
|
|
missingItem += [sec]
|
|
list_items = []
|
|
for i in missing_items:
|
|
list_items.append( (i, cfgSample.get(sec, i)) )
|
|
dicoMissingItem[sec] = list_items
|
|
|
|
if len(missingSection) == 0 and len(missingItem) == 0:
|
|
print("Configuration up-to-date")
|
|
return
|
|
print("/!\\ Configuration not complete. Missing following configuration: /!\\")
|
|
print("+--------------------------------------------------------------------+")
|
|
for section in missingSection:
|
|
print("["+section+"]")
|
|
for item in dicoMissingSection[section]:
|
|
print(" - "+item[0])
|
|
for section in missingItem:
|
|
print("["+section+"]")
|
|
for item in dicoMissingItem[section]:
|
|
print(" - "+item[0])
|
|
print("+--------------------------------------------------------------------+")
|
|
resp = raw_input("Do you want to auto fix it? [y/n] ")
|
|
|
|
if resp != 'y':
|
|
return
|
|
else:
|
|
for section in missingSection:
|
|
cfg.add_section(section)
|
|
for item, value in dicoMissingSection[section]:
|
|
cfg.set(section, item, value)
|
|
for section in missingItem:
|
|
for item, value in dicoMissingItem[section]:
|
|
cfg.set(section, item, value)
|
|
|
|
with open(configfile, 'w') as f:
|
|
cfg.write(f)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|