from twisted.internet.task import LoopingCall
from commands import add
from pyspades.constants import *

def apply_script(protocol, connection, config):
    class djumpConnection(connection):
        def on_login(self, name):
            self.fly = True
            return connection.on_login(self, name)
        def on_spawn(self, pos):
            for contained in connection.loader_recieved:
                if player.world_object.acceleration.z == 0:
                    player.fly = True
                elif player.world_object.acceleration.z != 0:
                    player.fly = False
                return connection.loader_recieved
            return connection.on_spawn(self, pos)

    return protocol, djumpConnection