From 572b6dfca96502584ec6302f5477f40f1602a63b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Saugat=20Pachhai=20=28=E0=A4=B8=E0=A5=8C=E0=A4=97=E0=A4=BE?=
 =?UTF-8?q?=E0=A4=A4=29?= <suagatchhetri@outlook.com>
Date: Fri, 20 May 2022 13:08:27 +0545
Subject: [PATCH] logging: show logs from dvc-objects

---
 dvc/cli/__init__.py |  4 ++--
 dvc/logger.py       | 18 ++++++++++++++++--
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/dvc/cli/__init__.py b/dvc/cli/__init__.py
index bb64d7bbb..f94000311 100644
--- a/dvc/cli/__init__.py
+++ b/dvc/cli/__init__.py
@@ -143,7 +143,7 @@ def main(argv=None):  # noqa: C901
     from dvc._debug import debugtools
     from dvc.config import ConfigError
     from dvc.exceptions import DvcException, NotDvcRepoError
-    from dvc.logger import disable_other_loggers
+    from dvc.logger import disable_other_loggers, set_loggers_level
 
     # NOTE: stderr/stdout may be closed if we are running from dvc.daemon.
     # On Linux we directly call cli.main after double forking and closing
@@ -171,7 +171,7 @@ def main(argv=None):  # noqa: C901
             level = logging.TRACE
 
         if level is not None:
-            logger.setLevel(level)
+            set_loggers_level(level)
 
         logger.trace(args)
 
diff --git a/dvc/logger.py b/dvc/logger.py
index 849e9b360..6c32376e7 100644
--- a/dvc/logger.py
+++ b/dvc/logger.py
@@ -207,7 +207,12 @@ def disable_other_loggers():
             logger.disabled = True
 
 
-def setup(level=logging.INFO):
+def set_loggers_level(level: int = logging.INFO) -> None:
+    for name in ["dvc", "dvc_objects"]:
+        logging.getLogger(name).setLevel(level)
+
+
+def setup(level: int = logging.INFO) -> None:
     colorama.init()
 
     if level >= logging.DEBUG:
@@ -269,7 +274,16 @@ def setup(level=logging.INFO):
                         "console_trace",
                         "console_errors",
                     ],
-                }
+                },
+                "dvc_objects": {
+                    "level": level,
+                    "handlers": [
+                        "console_info",
+                        "console_debug",
+                        "console_trace",
+                        "console_errors",
+                    ],
+                },
             },
             "disable_existing_loggers": False,
         }