Răsfoiți Sursa

fix stdin link importing

Nick Sweeting 5 ani în urmă
părinte
comite
7c428f40c8
1 a modificat fișierele cu 7 adăugiri și 2 ștergeri
  1. 7 2
      archivebox/cli/__init__.py

+ 7 - 2
archivebox/cli/__init__.py

@@ -64,9 +64,14 @@ def run_subcommand(subcommand: str,
 
 SUBCOMMANDS = list_subcommands()
 
+class NotProvided:
+    pass
+
+
+def main(args: Optional[List[str]]=NotProvided, stdin: Optional[IO]=NotProvided, pwd: Optional[str]=None) -> None:
+    args = sys.argv[1:] if args is NotProvided else args
+    stdin = sys.stdin if stdin is NotProvided else stdin
 
-def main(args: Optional[List[str]]=None, stdin: Optional[IO]=None, pwd: Optional[str]=None) -> None:
-    args = sys.argv[1:] if args is None else args
     subcommands = list_subcommands()
     parser = argparse.ArgumentParser(
         prog=__command__,