Add check_color() API

* Modified check_color() API

* Changed type of error

Author: BCKM

neopia/neosoco.py

@@ -450,6 +450,49 @@ def convert_scale(self, port='in1', omin=0, omax=255, tmin=0, tmax=100):
         else:
             raise TypeError
 
+    def check_color(self, port='in1', color='white'):
+        if isinstance(port and color, str):
+            if port.lower() =='in1':
+                c_value = self.read(Neosoco.INPUT_1)
+            elif port.lower() =='in2':
+                c_value = self.read(Neosoco.INPUT_2)
+            elif port.lower() =='in3':
+                c_value = self.read(Neosoco.INPUT_3)
+            else:
+                raise ValueError("Wrong value of port")
+
+            if not color.lower() in Neosoco._COLORS.keys():
+                raise ValueError('Wrong value of color')
+            if (c_value >= 10 and c_value <= 50):
+                if (color.lower() == self.COLOR_NAME_WHITE): 
+                    return True
+                else: 
+                    return False
+            elif (c_value >= 51 and c_value <= 90):
+                if (color.lower() == self.COLOR_NAME_RED): 
+                    return True
+                else: 
+                    return False
+            elif (c_value >= 91 and c_value <= 130):
+                if (color.lower() == self.COLOR_NAME_YELLOW): 
+                    return True
+                else: 
+                    return False
+            elif (c_value >= 131 and c_value <= 170):
+                if (color.lower() == self.COLOR_NAME_GREEN): 
+                    return True
+                else: 
+                    return False
+            elif (c_value >= 171 and c_value <= 210):
+                if (color.lower() == self.COLOR_NAME_BLUE): 
+                    return True
+                else: 
+                    return False
+            else:
+                return False 
+        else:
+            raise TypeError
+
     def led_on(self, port='out1', brightness='100'):
         percent_cvt = {
             '100': 255,
@@ -746,8 +789,9 @@ def buzzer_by_port(self, port='in1'):
             value = self._convert_scale_from_input_port(port, 65)
             self.write(Neosoco.NOTE, value)
         else:
+
             raise TypeError
-    
+
     def buzzer_stop(self):
         self.write(Neosoco.NOTE, 0)
 
@@ -761,4 +805,4 @@ def remote_button(self, button='1'):
             else:
                 raise ValueError('Wrong value of button')
         else:
-            raise TypeError
+            raise TypeError

test.py

@@ -119,4 +119,13 @@
   #   n.servo_rotate('out2', 'forward', '50')
   #   wait(2000)
   #   n.servo_rotate('out2', 'backward', '50')
-  #   wait(2000)
+  #   wait(2000)
+
+# case 17) Turn on LED when sensor detects green color
+# while True:
+#     print(n.get_value('in1'))
+#     wait(500)
+#     if n.check_color('in1', 'Green'):
+#         n.led_on('out1', '100')
+#     else:
+#         n.led_off('out1')