ftl/python/main.py

# Copyright 2017 Google Inc. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""A binary for constructing images from a source context."""

import sys
import argparse

from containerregistry.tools import patched

from ftl.common import args
from ftl.common import logger
from ftl.common import context
from ftl.common import ftl_util
from ftl.common import ftl_error

from ftl.python import builder as python_builder

version = args.version_parser()
parser = args.base_parser()
python_parser = argparse.ArgumentParser(
    add_help=False,
    parents=[parser],
    description='Construct python images from source.')
args.extra_args(python_parser, args.python_flgs)


def main(cli_args):
    try:
        version.parse_known_args(cli_args)
        builder_args = python_parser.parse_args(cli_args)
        logger.setup_logging(builder_args)
        logger.preamble("python", builder_args)
        with ftl_util.Timing("full build"):
            with ftl_util.Timing("builder initialization"):
                python_ftl = python_builder.Python(
                    context.Workspace(builder_args.directory), builder_args)
            with ftl_util.Timing("build process for FTL image"):
                python_ftl.Build()
    except ftl_error.UserError as e:
        ftl_error.UserErrorHandler(
            e, builder_args.builder_output_path, builder_args.fail_on_error)
    except ftl_error.InternalError as e:
        ftl_error.InternalErrorHandler(
            e, builder_args.builder_output_path, builder_args.fail_on_error)


if __name__ == '__main__':
    with patched.Httplib2():
        main(sys.argv[1:])