From 70d4e5b2b439e6599cba8dd10e8b3ebbc70c2fdb Mon Sep 17 00:00:00 2001
From: Joost Lekkerkerker <joostlek@outlook.com>
Date: Wed, 21 Feb 2024 17:53:33 +0100
Subject: [PATCH] Fix T201 (#244)

---
 pyproject.toml           |  1 -
 roombapy/entry_points.py | 13 ++++++++-----
 2 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 33f3ea2..fb654ae 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -81,7 +81,6 @@ ignore = [
     "RET507",
     "SIM102",
     "SLF001",
-    "T201",
 ]
 select = ["ALL"]
 
diff --git a/roombapy/entry_points.py b/roombapy/entry_points.py
index 11b981b..d997b03 100644
--- a/roombapy/entry_points.py
+++ b/roombapy/entry_points.py
@@ -1,21 +1,24 @@
+import logging
 import sys
 
 from roombapy import RoombaFactory
 from roombapy.discovery import RoombaDiscovery
 from roombapy.getpassword import RoombaPassword
 
+LOGGER = logging.getLogger(__name__)
+
 
 def discovery():
     roomba_ip = _get_ip_from_arg()
 
     roomba_discovery = RoombaDiscovery()
     if roomba_ip is not None:
-        print(roomba_discovery.find(roomba_ip))
+        LOGGER.info(roomba_discovery.find(roomba_ip))
         return
 
     robots_info = roomba_discovery.find()
     for robot in robots_info:
-        print(robot)
+        LOGGER.info(robot)
 
 
 def password():
@@ -30,7 +33,7 @@ def password():
     roomba_password = RoombaPassword(roomba_ip)
     found_password = roomba_password.get_password()
     roomba_info.password = found_password
-    print(roomba_info)
+    LOGGER.info(roomba_info)
 
 
 def connect():
@@ -47,7 +50,7 @@ def connect():
     roomba = RoombaFactory.create_roomba(
         roomba_info.ip, roomba_info.blid, roomba_password
     )
-    roomba.register_on_message_callback(lambda msg: print(msg))
+    roomba.register_on_message_callback(lambda msg: LOGGER.info(msg))
     roomba.connect()
 
     while True:
@@ -70,7 +73,7 @@ def _validate_roomba_info(roomba_info):
 
 
 def _wait_for_input():
-    print(
+    LOGGER.info(
         "Roomba have to be on Home Base powered on.\n"
         "Press and hold HOME button until you hear series of tones.\n"
         "Release button, Wi-Fi LED should be flashing"