compass-core/bin/chef/addroles.py
xiaodongwang 7f080e4e51 fix pep8 warning and move major util code from bin to compass dir.
Change-Id: I7baa536888db1ca46cadcdf07ef9cc0a1a3a12fb
2014-02-19 22:25:45 -08:00

35 lines
713 B
Python
Executable File

#!/usr/bin/env python
"""script to import roles to chef server"""
import logging
import os
import os.path
from compass.utils import flags
from compass.utils import logsetting
flags.add('roles_dir',
help='chef roles directory',
default='/var/chef/roles')
def main():
"""main entry"""
flags.init()
logsetting.init()
rolelist = []
roles_dir = flags.OPTIONS.roles_dir
for item in os.listdir(roles_dir):
role_file = os.path.join(roles_dir, item)
rolelist.append(role_file)
for role in rolelist:
logging.info('add role %s', role)
cmd = "knife role from file %s" % role
os.system(cmd)
if __name__ == '__main__':
main()