## GENERATED WITH BLOCKMAN2D v0.2.5.0
name = 'enanojump'
version = ''
author = 'Lostmotel'
description = ''
extensions = { 'water_damage' : 25 }

from pyspades.constants import *
from pyspades.server import ServerConnection
import random
 
def get_entity_location(team, entity_id):
     if entity_id == BLUE_FLAG:
         return (239, 238, 8)
     if entity_id == GREEN_FLAG:
         return (239, 238, 8)
     if entity_id == BLUE_BASE:
         return (154, 204, 8)
     if entity_id == GREEN_BASE:
         return (155, 203, 8)
 
SPAWN_RECTANGLES = [[(156,156,206,205)], [(152,153,206,207)]]
 
def get_spawn_location(connection):
     if SPAWN_RECTANGLES is not None:
         if connection.team is connection.protocol.green_team:
             x1, x2, y1, y2 = random.choice(SPAWN_RECTANGLES[0])
         elif connection.team is connection.protocol.blue_team:
             x1, x2, y1, y2 = random.choice(SPAWN_RECTANGLES[1])
         x, y = random.randint(x1,x2), random.randint(y1,y2)
         z = connection.protocol.map.get_z(x, y)
     if x is not None and y is not None and z is not None:
         return ServerConnection.set_location_safe(connection, (x, y, z))
     else:
         return ServerConnection.get_spawn_location(connection)