Browse Source

fix stdin link importing

Nick Sweeting 5 years ago
parent
commit
7c428f40c8
1 changed files with 7 additions and 2 deletions
  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()
 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()
     subcommands = list_subcommands()
     parser = argparse.ArgumentParser(
     parser = argparse.ArgumentParser(
         prog=__command__,
         prog=__command__,