with conn.cursor() as cur: cur.execute("SELECT 1") result = cur.fetchone() if result == (1, ): print(f"success!") return else: print(f"weird, {args} did not return 1: {result}") except Exception as e: ui.progress(".") error = e ui.progress(finish=True) raise Failed(f"Never got correct result for {args}: {error}") @contextlib.contextmanager def cd(path: Path) -> Any: """Execute block within path, and then return""" orig_path = os.getcwd() try: os.chdir(path) yield finally: os.chdir(orig_path) if __name__ == "__main__": with error_handler(say): cli(auto_envvar_prefix="MZ")
self.add_argument("-f", "--file") self.add_argument("--project-directory") self.add_argument("command", nargs="?") self.add_argument("first_command_arg", nargs="?") self.add_argument("remainder", nargs=argparse.REMAINDER) def parse_known_args( self, args: Optional[Sequence[Text]] = None, namespace: Optional[argparse.Namespace] = None, ) -> Tuple[argparse.Namespace, List[str]]: ns = argparse.Namespace() try: (pargs, unknown_args) = super().parse_known_args(args, namespace=ns) if pargs.mz_build_mode not in ["dev", "release"]: raise errors.BadSpec( f'unknown build mode {pargs.mz_build_mode!r} (expected "dev" or "release")' ) return (pargs, unknown_args) except ValueError: return (ns, []) def error(self, message: str) -> NoReturn: raise ValueError(message) if __name__ == "__main__": with errors.error_handler(lambda *args: print(*args, file=sys.stderr)): sys.exit(main(sys.argv[1:]))