def transform_fn()

in containers/Layout/2vs3/host.py [0:0]

• 41 lines of code
• 15 McCabe index (conditional complexity)

def transform_fn(model, request_body, content_type, accept_type):
    try:
        print(request_body)
        input_object=json.loads(request_body)

        height=input_object["height"]
        width=input_object["width"]
        ships=input_object["ruleSet"]["shipCells"]

        layout=[[0]*width for x in range(height)]
        size=int(choices([2,3]))
        print(size) 
        num_ships=math.ceil(ships/2)
        extra=ships-size*num_ships
       
        while True: #iterate till we generate a valid layout
            layout=[[0]*width for x in range(height)]

            #generate ships of size 2
            cords=[(random.randint(0,width-size),random.randint(0,width-size)) 
                for z in range(num_ships)]
        
            if size==2:
                for (x,y) in cords:
                    layout[x][y]=1
                    layout[x][y+1]=1
            else:
                for (x,y) in cords:
                    layout[x][y]=1
                    layout[x][y+1]=1
                    layout[x][y+2]=1

            #gerenate ships of size 1 incase ships is not a multiple of 2 or 3
            cords=[(random.randint(0,width-1),random.randint(0,width-1)) 
                for z in range(extra)]
            
            for (x,y) in cords:
                layout[x][y]=1
            
            if sum([sum(x) for x in layout])==ships: #test for valid layout
                break

        return bytearray(json.dumps({
            "layout":layout,
            "session":{
                "type":"random-2vs3",
                "size":size
            }
        }),'utf-8'),accept_type
    except Exception as e:
        print(traceback.format_exc())
        print(e)