def test_flag_format_env_variable(): """Test flag formatting and env variable.""" class OtherFlags(Enum): """Some flags to be used on the assertions below.""" MULTI_WORD = 1 SOME_OPTION = 2 assert NitpickApp.format_flag( OtherFlags.MULTI_WORD) == "--nitpick-multi-word" os.environ["NITPICK_SOME_OPTION"] = "something" assert NitpickApp.format_env( OtherFlags.SOME_OPTION) == "NITPICK_SOME_OPTION" assert NitpickApp.get_env(OtherFlags.SOME_OPTION) == "something" assert NitpickApp.get_env(OtherFlags.MULTI_WORD) == ""
def parse_options(option_manager: OptionManager, options, args): # pylint: disable=unused-argument """Create the Nitpick app, set logging from the verbose flags, set offline mode. This function is called only once by flake8, so it's a good place to create the app. """ log_mapping = {1: logging.INFO, 2: logging.DEBUG} logging.basicConfig( level=log_mapping.get(options.verbose, logging.WARNING)) NitpickApp.create_app( offline=bool(options.nitpick_offline or NitpickApp.get_env(NitpickApp.Flags.OFFLINE))) LOGGER.info("Offline mode: %s", NitpickApp.current().offline)