cleanup prints

examples/test_extract.py

@@ -1,35 +1,38 @@
-import krec
-import argparse
-from pathlib import Path
-
-
 """
+Extract KRec data from an MKV file and optionally save it to a new file.
+
+Usage:
 # Just extract and view
 python examples/test_extract.py input.mkv -v
 
 # Extract and save
 python examples/test_extract.py input.mkv -o output.krec -v
 """
 
+import argparse
+import logging
+from pathlib import Path
+
+import colorlogging
+
+import krec
+
 
 def main(args):
-    print(f"Extracting from: {args.input_file}")
+    logging.info("Extracting from: %s", args.input_file)
 
     try:
         extracted_krec = krec.extract_from_video(args.input_file, verbose=args.verbose)
-        print("Extraction successful")
-        print(f"Extracted KRec has {len(extracted_krec)} frames")
+        logging.info("Extraction successful")
+        logging.info("Extracted KRec has %d frames", len(extracted_krec))
 
         if args.output_file:
             output_path = Path(args.output_file)
             extracted_krec.save(str(output_path))
-            print(f"Saved to: {output_path}")
+            logging.info("Saved to: %s", output_path)
 
     except Exception as e:
-        print(f"Error: {e}")
-        return 1
-
-    return 0
+        logging.error("Error: %s", e)
 
 
 if __name__ == "__main__":
@@ -38,5 +41,7 @@ def main(args):
     parser.add_argument("-o", "--output-file", type=str, help="Output KRec file path (optional)")
     parser.add_argument("-v", "--verbose", action="store_true", help="Enable verbose output")
 
+    colorlogging.configure()
+
     args = parser.parse_args()
-    exit(main(args))
+    main(args)

requirements.txt

@@ -0,0 +1 @@
+colorlogging

setup.py

@@ -22,6 +22,9 @@
 version: str = version_re.group(1)
 
 
+with open("requirements.txt") as f:
+    requirements: list[str] = f.read().splitlines()
+
 class RustBuildExt(build_ext):
     def run(self) -> None:
         # Generate the stub file
@@ -60,6 +63,7 @@ def run(self) -> None:
         ),
     ],
     setup_requires=["setuptools-rust"],
+    install_requires=requirements,
     zip_safe=False,
     long_description=long_description,
     long_description_content_type="text/markdown",