Shape Classes with Area Method.py

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()