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:]))
