From 5a0e2e9158da0837684bea686b3915e3eb8b24a3 Mon Sep 17 00:00:00 2001
From: ayasyrev <a.yasyrev@gmail.com>
Date: Fri, 30 Aug 2024 16:41:07 +0300
Subject: [PATCH] nbcheck add `--warn` argument - check for warnings

---
 src/nbmetaclean/app_check.py | 28 ++++++++++++++++++++++------
 1 file changed, 22 insertions(+), 6 deletions(-)

diff --git a/src/nbmetaclean/app_check.py b/src/nbmetaclean/app_check.py
index 63e9b5b..6944451 100644
--- a/src/nbmetaclean/app_check.py
+++ b/src/nbmetaclean/app_check.py
@@ -1,6 +1,6 @@
 import argparse
 
-from .check import check_nb_ec, check_nb_errors
+from .check import check_nb_ec, check_nb_errors, check_nb_warnings
 from .helpers import get_nb_names_from_list, read_nb
 
 parser = argparse.ArgumentParser(
@@ -23,6 +23,11 @@
     action="store_true",
     help="Check errors in outputs.",
 )
+parser.add_argument(
+    "--warn",
+    action="store_true",
+    help="Check warnings in outputs.",
+)
 parser.add_argument(
     "--not_strict",
     action="store_true",
@@ -73,15 +78,26 @@ def app_check() -> None:
                 print("- ", nb)
 
     if cfg.err:
-        wrong_err = []
+        nb_errors = []
         for nb in nb_files:
             result = check_nb_errors(read_nb(nb))
             if not result:
-                wrong_err.append(nb)
+                nb_errors.append(nb)
+
+        if nb_errors:
+            print(f"{len(nb_errors)} notebooks with some errors in outputs:")
+            for nb in nb_errors:
+                print("- ", nb)
+    if cfg.warn:
+        nb_warnings = []
+        for nb in nb_files:
+            result = check_nb_warnings(read_nb(nb))
+            if not result:
+                nb_warnings.append(nb)
 
-        if wrong_err:
-            print(f"{len(wrong_err)} notebooks with some errors in outputs:")
-            for nb in wrong_err:
+        if nb_warnings:
+            print(f"{len(nb_warnings)} notebooks with some warnings in outputs:")
+            for nb in nb_warnings:
                 print("- ", nb)