Behavorial pattern => State

Author: rezamobaraki

state2.py

@@ -0,0 +1,49 @@
+"""
+    Behavioral pattern:
+        State
+    Example:
+        switch between several state
+"""
+
+
+class State:
+    def operate(self):
+        print(f'Turning Tv {self.status}')
+
+
+class TurnOn(State):
+    def __init__(self, tv):
+        self.tv = tv
+        self.status = 'On'
+
+    def change_state(self):
+        print('Changing stat to off...')
+        self.tv.state = self.tv.off
+
+
+class TurnOff(State):
+    def __init__(self, tv):
+        self.tv = tv
+        self.status = 'Off'
+
+    def change_state(self):
+        print('Changing state to On...')
+        self.tv.state = self.tv.on
+
+
+class Tv:
+    def __init__(self):
+        self.on = TurnOn(self)
+        self.off = TurnOff(self)
+        self.state = self.on
+
+    def press(self):
+        self.state.operate()
+        self.state.change_state()
+
+
+if __name__ == '__main__':
+    t = Tv()
+    t.press()
+    t.press()
+    t.press()