more exceptions, testing exceptions

Kepka · Kepka · commit 4fdb41d62c28 · 2022-05-19T12:05:02.000+02:00
diff --git a/Game/ducks.py b/Game/ducks.py
@@ -7,21 +7,21 @@ def fly(self):
         if self.ratio > 1:
             print("Weee, this is fun")
         elif self.ratio == 1:
-            print("This is hard word, but I'm flying")
+            print("This is hard work, but I'm flying")
         else:
             print("I think I'll just walk")
 
 
 class Duck(object):
 
-    def __int__(self):
+    def __init__(self):
         self._wing = Wing(1.8)
 
     def walk(self):
         print("Waddle, waddle, waddle")
 
     def swim(self):
-        print("Come on it, th water's lovely")
+        print("Come on in, the water's lovely")
 
     def quack(self):
         print("Quack quack")
@@ -32,24 +32,60 @@ def fly(self):
 
 class Penguin(object):
 
+    def __init__(self):
+        self.fly = self.aviate
+
     def walk(self):
         print("Waddle, waddle, I waddle too")
 
     def swim(self):
-        print("Come on in, it's a bit chilly this far South")
+        print("Come on in, but it's a bit chilly this far South")
 
     def quack(self):
-        print("Are you 'avin' a larf? I'm penguin!")
+        print("Are you 'avin' a larf? I'm a penguin!")
+
+    def aviate(self):
+        print("I won the lottery and bought a learjet")
 
 
 # def test_duck(duck):
 #     duck.walk()
 #     duck.swim()
 #     duck.quack()
 
+class Mallard(Duck):
+    pass
+
+
+class Flock(object):
+
+    def __init__(self):
+        self.flock = []
+
+    def add_duck(self, duck: Duck) -> None:
+        fly_method = getattr(duck, 'fly', None)
+        # if isinstance(duck, Duck):
+        if callable(fly_method):
+            self.flock.append(duck)
+        else:
+            raise TypeError("Cannot add duck, are you sure it's not a " + str(type(duck).__name__))
+
+    def migrate(self):
+        problem = None
+        for duck in self.flock:
+            try:
+                duck.fly()
+                raise AttributeError("Testing exception handler in migrate")    # TODO remove before release
+            except AttributeError as e:
+                print('One duck down')
+                problem = e
+        if problem:
+            raise problem
+
 
 if __name__ == '__main__':
     donald = Duck()
+    donald.fly()
 
     # percy = Penguin()
     # test_duck(percy)
diff --git a/Game/migration.py b/Game/migration.py
@@ -0,0 +1,22 @@
+import ducks
+
+flock = ducks.Flock()
+donald = ducks.Duck()
+daisy = ducks.Duck()
+duck3 = ducks.Duck()
+duck4 = ducks.Duck()
+duck5 = ducks.Duck()
+duck6 = ducks.Duck()
+duck7 = ducks.Duck()
+percy = ducks.Penguin()
+
+flock.add_duck(donald)
+flock.add_duck(daisy)
+flock.add_duck(duck3)
+flock.add_duck(duck4)
+flock.add_duck(percy)
+flock.add_duck(duck5)
+flock.add_duck(duck6)
+flock.add_duck(duck7)
+
+flock.migrate()
