collect all results and display them all at the end

Author: Wulian233

Lib/mimetypes.py

@@ -718,8 +718,6 @@ def _parse_args(args):
 
 def _main(args=None):
     """Run the mimetypes command-line interface and return a text to print."""
-    import sys
-
     args, help_text = _parse_args(args)
 
     results = []
@@ -729,17 +727,21 @@ def _main(args=None):
             if guess:
                 results.append(str(guess))
             else:
-                sys.exit(f"error: unknown type {gtype}")
-        return '\n'.join(results)
+                results.append(f"error: unknown type {gtype}")
+        return results
     else:
         for gtype in args.type:
             guess, encoding = guess_type(gtype, not args.lenient)
             if guess:
                 results.append(f"type: {guess} encoding: {encoding}")
             else:
-                sys.exit(f"error: media type unknown for {gtype}")
-        return '\n'.join(results)
+                results.append(f"error: media type unknown for {gtype}")
+        return results
 
 
 if __name__ == '__main__':
-    print(_main())
+    import sys
+
+    results = _main()
+    print("\n".join(results))
+    sys.exit(any(result.startswith("error: ") for result in results))

Lib/test/test_mimetypes.py

@@ -471,19 +471,20 @@ def test_parse_args(self):
         self.assertEqual(args.type, ["foo.pic"])
 
     def test_multiple_inputs(self):
-        result = mimetypes._main(shlex.split("foo.pdf foo.png"))
+        result = "\n".join(mimetypes._main(shlex.split("foo.pdf foo.png")))
         self.assertEqual(
             result,
             "type: application/pdf encoding: None\n"
             "type: image/png encoding: None"
         )
 
     def test_multiple_inputs_error(self):
-        with self.assertRaises(SystemExit) as err:
-            mimetypes._main(shlex.split("foo.pdf foo.bar_ext"))
-
-        self.assertNotEqual(err.exception.code, 0)
-        self.assertIn("error: media type unknown for foo.bar_ext", str(err.exception))
+        result = "\n".join(mimetypes._main(shlex.split("foo.pdf foo.bar_ext")))
+        self.assertEqual(
+            result,
+            "type: application/pdf encoding: None\n"
+            "error: media type unknown for foo.bar_ext"
+        )
 
 
     def test_invocation(self):
@@ -492,16 +493,17 @@ def test_invocation(self):
             ("-e image/jpeg", ".jpg"),
             ("-l foo.webp", "type: image/webp encoding: None"),
         ]:
-            self.assertEqual(mimetypes._main(shlex.split(command)), expected)
+            result = "\n".join(mimetypes._main(shlex.split(command)))
+            self.assertEqual(result, expected)
 
     def test_invocation_error(self):
         for command, expected in [
             ("-e image/jpg", "error: unknown type image/jpg"),
             ("foo.bar_ext", "error: media type unknown for foo.bar_ext"),
         ]:
             with self.subTest(command=command):
-                with self.assertRaisesRegex(SystemExit, expected):
-                    mimetypes._main(shlex.split(command))
+                result = "\n".join(mimetypes._main(shlex.split(command)))
+                self.assertEqual(result, expected)
 
 
 if __name__ == "__main__":