Source.Python forums

from random import randint

from engines.server import global_vars
from effects.base import TempEntity
from entities.entity import Entity
from filters.entities import EntityIter
from filters.players import PlayerIter
from filters.weapons import WeaponClassIter
from listeners.tick import Delay, Repeat
from mathlib import Vector
from messages import SayText2
from translations.strings import LangStrings

from . import Hero, Skill
from . import events

## TRANSLATIONS

strings = LangStrings('heroes/flame_predator')

_knifeonly = set(weapon.name for weapon in WeaponClassIter()) - {'weapon_knife', 'weapon_c4'}


class Flame_Predator(Hero):
    name = 'Flame Predator'
    description = 'Recoded Flame Predator from original WCS (Kryptonite).'
    category = 'KNIFE'
    cost = 0
    required_level = 100

    @classmethod
    def meets_requirements(cls, player):
        if global_vars.map_name in ('aim_deagle_simpsons', 'aim_r3volver', 'aim_deagle_revolution', 'scoutzknivez_oz20',):
            return False    
        return True 

@Flame_Predator.skill
class Predator(Skill):
    name = 'Predator'
    description = 'Predator only knife.'

    @events('player_spawn')
    def player_spawn(self, player, **eargs):
        if player.team <= 1:
            return
        player.restrict_weapons(*_knifeonly)
        for weapon in player.weapons(not_filters='knife'):
            player.drop_weapon(weapon.pointer, None, None)


@Flame_Predator.skill
class Berserk(Skill):
    name = 'Berserk'
    description = 'More health and speed.'
    max_level = 8

    @events('player_spawn')
    def _on_player_spawn(self, player, **eargs):
        if self.level == 0:
            return
        player.health += 5 * self.level
        player.speed = 1.4 + (0.05 * self.level)


@Flame_Predator.skill
class Cloak_of_Invisibility(Skill):
    name = 'Cloak of Invisibility'
    description = 'Put on your cloak to be 20-90% invisible.'
    max_level = 8

    @events('player_spawn')
    def _on_player_spawn_give_invis(self, player, **eargs):
        if self.level == 0:
            return
        color = player.color
        color.a = 255 - (30 * self.level)
        player.color = color


@Flame_Predator.skill
class Levitation(Skill):
    name = 'Levitation'
    description = 'Reduce your current gravity and damage taken when jumping.'
    max_level = 8

    @events('player_spawn')
    def _on_player_spawn(self, player, **eargs):
        player.gravity = 1 - 0.08 * self.level

    @events('player_pre_victim')
    def _on_player_pre_victim(self, attacker, victim, take_damage_info, **eargs):
        if victim.ground_entity == -1 and self.level != 0:
            take_damage_info.damage *= 1 - 0.06 * self.level
            SayText2(
                strings['Levitation Message 1'].get_string(
                    victim.language[:2], name=attacker.name)
            ).send(victim.index)
            ricochet = TempEntity('Armor Ricochet', direction=victim.view_vector, position=victim.origin)
            ricochet.create()


@Flame_Predator.skill
class Claw_Attack(Skill):
    name = 'Claw Attack'
    description = 'Force an enemy to drop his weapon when you attack him. 35%'
    max_level = 8

    @events('player_attack')
    def _on_player_attack(self, attacker, victim, **eargs):
        if randint(1, 100) > 35 or victim.active_weapon is None or self.level == 0:
            return

        victim.drop_weapon(victim.active_weapon)

        SayText2(strings['Claw Attack Message 1'].get_string(attacker.language[:2], name=victim.name)
            ).send(attacker.index)
        SayText2(strings['Claw Attack Message 2'].get_string(victim.language[:2], name=attacker.name)
            ).send(victim.index)


@Flame_Predator.skill
class Burning_Blade(Skill):
    name = 'Burning Blade'
    description = 'Set an enemy on fire when attacking him. 30%'
    max_level = 8

    @events('player_attack')
    def _on_player_attack(self, attacker, victim, **eargs):
        if randint(1, 100) > 30 or self.level == 0:
            return

        victim.ignite_lifetime(3)

        SayText2(strings['Burning Blade Message 1'].get_string(attacker.language[:2], name=victim.name)
            ).send(attacker.index)
        SayText2(strings['Burning Blade Message 2'].get_string(victim.language[:2], name=attacker.name)
            ).send(victim.index)

@Event('player_disconnect')
def _on_disconnect_save_data(event_data):
    print("[debug] In PLAYER_DISCONNECT event")
    player = players[event_data['userid']]
    print("[debug] Got player")
    save_player_data(player)
    print("[debug] Saved player data")
    save_hero_data(player)
    print("[debug] Saved hero data")
    del players[event_data['userid']]
    print("[debug] Out of PLAYER_DISCONNECT event")