add shape classes with area method

Certification/02. Python (Basic)/01. Shape Classes with Area Method/Solution.py

@@ -0,0 +1,37 @@
+import os
+from math import pi
+class Rectangle:
+    def __init__(self,l,w):
+        self.l = l
+        self.w = w
+
+    def area(self):
+        return (self.l*self.w)
+
+
+class Circle:
+    def __init__(self,r):
+        self.r = r
+
+    def area(self):
+        return (pi*self.r*self.r)
+
+
+
+if __name__ == '__main__':
+    fptr = open(os.environ['OUTPUT_PATH'], 'w')
+    q = int(input())
+    queries = []
+    for _ in range(q):
+        args = input().split(' ')
+        shape_name, params = args[0], map(int, args[1:])
+        if shape_name == "rectangle":
+            a, b = params[0], params[1]
+            shape = Rectangle(a, b)
+        elif shape_name == "circle":
+            r = params[0]
+            shape = Circle(r)
+        else:
+            raise ValueError("invalid shape type")
+        fptr.write("%.2f\n" % shape.area())
+    fptr.close()