Source.Python forums

import es
import corelib
import gamethread

ent_dict = {}
remove_list = {}

def es_map_start(ev):
	gamethread.delayed(1.0,do_it)
	
def do_it():
	#Put all entities in a dictionary and assign the location to its index. Structure: Classname -> Index -> Location
	remove_list = {}
	ents = es.createentitylist()
	for ent in ents:
		class_name = ents[ent]['classname']
		if class_name not in ent_dict:
			ent_dict[class_name] = {}
		loc = es.getindexprop(ent, 'CBaseEntity.m_vecOrigin')
		if str(loc) == "0.000000,0.000000,0.000000":
			loc = es.getindexprop(ent, 'CBaseEntity.m_Collision.m_vecMins')
		ent_dict[class_name][ent] = loc
	

	#Put entities in the remove_list only if the location  is not used yet.
	for class_name in ent_dict:
		#ent_dict[class_name] = entity index
		
		#Check if the class_name was assigned to the remove_list before
		if class_name not in remove_list:
		
			#If not, assign it an empty dictionary to keep all entities with the same name but with different location
			remove_list[class_name] = []
		
		#Loop through all the locations for that given classname
		for loc in ent_dict[class_name]:
			
			#If the location is not in the remove_dict already add it, if it is, remove the entity
			if ent_dict[class_name][loc] not in remove_list[class_name]:
				remove_list[class_name].append(ent_dict[class_name][loc])
			else:
				if str(ent_dict[class_name][loc]) != "0.000000,0.000000,0.000000":
					es.remove(loc)

# =============================================================================
# >> IMPORTS
# =============================================================================
import shlex

from memory import get_virtual_function
from memory.hooks import PreHook
from engines.server import engine_server
from engines.server import server_game_dll


# =============================================================================
# >> CONFIGURATION
# =============================================================================
# Tested in CS:S on cs_office

def handle_entity(props):
    """Return ``True`` to add the entity, ``False`` to remove it.

    You can also modify the entity's properties. Don't forget to return ``True``
    in that case as well.
    """
    # EXAMPLE: Removing entities
    # Window?
    if get_property(props, 'classname') == 'func_breakable_surf':
        # Don't add windows
        return False

    # EXAMPLE: Modifying entities
    # CT spawn?
    if get_property(props, 'classname') == 'info_player_counterterrorist':
        # Get the current origin, increase the Z axis by 100 units and set it
        origin = list(map(float, get_property(props, 'origin').split(' ')))
        origin[2] += 100
        set_property(props, 'origin', ' '.join(map(str, origin)))

    # Add all other entities
    return True

def add_entities(entities):
    """Add entities here."""

    # EXAMPLE:
    # Add a new entity by parsing a string. It doesn't care about whitespaces.
    entities.append(
        parse_map_entities_string(
            """
            {
                "origin" "-1080.348633 -1792.366089 -224.470856"
                "classname" "weapon_awp"
            }
            """
        )[0]
    )

    # EXAMPLE:
    # Add an entity by using the list style
    entities.append(
        [
            ('classname', 'weapon_m4a1'),
            ('origin', '-1080.348633 -1792.366089 -224.470856')
        ]
    )


# =============================================================================
# >> OTHER STUFF
# =============================================================================
new_map_entities_string = ''

@PreHook(get_virtual_function(engine_server, 'GetMapEntitiesString'))
def post_get_map_entities_string(args):
    return new_map_entities_string

@PreHook(get_virtual_function(server_game_dll, 'LevelInit'))
def pre_level_init(args):
    entities = []

    for entity in parse_map_entities_string(args[2]):
        if handle_entity(entity):
            entities.append(entity)

    # Ask for new/own entities
    add_entities(entities)

    # Save the string, so the memory is not invalidated
    global new_map_entities_string
    new_map_entities_string = build_map_entities_string(entities)
    args[2] = new_map_entities_string

def parse_map_entities_string(buffer):
    """Parse a "map entities"-like string and return it as a nested list."""
    entities = []
    curr_ent = None
    for line in buffer.split('\n'):
        line = line.strip()

        # Skip empty lines
        if not line:
            continue

        if line.startswith('{'):
            curr_ent = []
        elif line.startswith('}'):
            entities.append(curr_ent)
            curr_ent = None
        else:
            curr_ent.append(shlex.split(line))

    return entities

def build_map_entities_string(entities):
    """Create a "map entities"-like string."""
    buffer = ''
    for ent in entities:
        ent_buffer = ''
        for key, value in ent:
            ent_buffer += f'"{key}" "{value}"\n'

        buffer += '{\n%s}\n'% ent_buffer
    return buffer

def get_property(props, name):
    """Return the first found property with the given name. An empty string is
    returned if the property wasn't found.
    """
    for key, value in props:
        if key == name:
            return value

    return ''

def set_property(props, name, value):
    """Change the first found property or add it if missing."""
    assert isinstance(value, str)

    for prop in props:
        if prop[0] == name:
            prop[1] = value
            break
    else:
        # Not found? No problem. Just add it
        props.append((name, value))