Allow to set different log level

trino/logging.py

@@ -10,13 +10,22 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import os
 import logging
+from typing import Optional
 
-LEVEL = logging.INFO
+_level = os.environ.get('TRINO_LOGLEVEL', 'INFO')
+if _level.upper() == _level and hasattr(logging, _level) and isinstance(getattr(logging, _level), int):
+    LEVEL = getattr(logging, _level)
+elif _level == 'NONE':
+    LEVEL = None
+else:
+    LEVEL = logging.INFO
 
 
 # TODO: provide interface to use ``logging.dictConfig``
-def get_logger(name: str, log_level: int = LEVEL) -> logging.Logger:
+def get_logger(name: str, log_level: Optional[int] = LEVEL) -> logging.Logger:
     logger = logging.getLogger(name)
-    logger.setLevel(log_level)
+    if log_level is not None:
+        logger.setLevel(log_level)
     return logger