forked from ungleich-public/cdist
103 lines
2.9 KiB
Python
Executable file
103 lines
2.9 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# 2010-2016 Nico Schottelius (nico-cdist at schottelius.org)
|
|
# 2016 Darko Poljak (darko.poljak at gmail.com)
|
|
#
|
|
# This file is part of cdist.
|
|
#
|
|
# cdist is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# cdist is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with cdist. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
#
|
|
|
|
import logging
|
|
import os
|
|
import sys
|
|
|
|
# try to import cdist and if that fails, then add this file's parent dir to
|
|
# module search path and try again. additionally check if this file is
|
|
# symlinked, so user can symlink this file to, for example, ~/.local/bin.
|
|
try:
|
|
import cdist
|
|
except ModuleNotFoundError:
|
|
cdist_bin = __file__
|
|
if os.path.islink(cdist_bin):
|
|
cdist_bin = os.readlink(cdist_bin)
|
|
cdist_dir = os.path.abspath(os.path.join(os.path.dirname(cdist_bin), os.pardir))
|
|
sys.path.insert(0, cdist_dir)
|
|
import cdist
|
|
|
|
import cdist.argparse
|
|
import cdist.banner
|
|
import cdist.config
|
|
import cdist.install
|
|
import cdist.shell
|
|
import cdist.inventory
|
|
|
|
|
|
def commandline():
|
|
"""Parse command line"""
|
|
|
|
# preos subcommand hack
|
|
if len(sys.argv) > 1 and sys.argv[1] == 'preos':
|
|
return cdist.preos.PreOS.commandline(sys.argv[1:])
|
|
parser, cfg = cdist.argparse.parse_and_configure(sys.argv[1:])
|
|
args = cfg.get_args()
|
|
|
|
# Work around python 3.3 bug:
|
|
# http://bugs.python.org/issue16308
|
|
# http://bugs.python.org/issue9253
|
|
|
|
# FIXME: catching AttributeError also hides
|
|
# real problems.. try a different way
|
|
|
|
# FIXME: we always print main help, not
|
|
# the help of the actual parser being used!
|
|
try:
|
|
getattr(args, "func")
|
|
except AttributeError:
|
|
parser['main'].print_help()
|
|
sys.exit(0)
|
|
|
|
args.func(args)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if sys.version < cdist.MIN_SUPPORTED_PYTHON_VERSION:
|
|
print('Python >= {} is required on the source host.'.format(
|
|
cdist.MIN_SUPPORTED_PYTHON_VERSIO), file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
exit_code = 0
|
|
|
|
try:
|
|
import re
|
|
import os
|
|
|
|
if re.match("__", os.path.basename(sys.argv[0])):
|
|
import cdist.emulator
|
|
emulator = cdist.emulator.Emulator(sys.argv)
|
|
emulator.run()
|
|
else:
|
|
commandline()
|
|
|
|
except KeyboardInterrupt:
|
|
exit_code = 2
|
|
|
|
except cdist.Error as e:
|
|
log = logging.getLogger("cdist")
|
|
log.error(e)
|
|
exit_code = 1
|
|
|
|
sys.exit(exit_code)
|