Ejemplo n.º 1
0
def start_htm(server_port, settings_id):
    setjson = RunSettings.select().where(RunSettings.id == settings_id).get()
    f = SettingsForm.from_json(jsonpickle.json.loads(setjson.json_string), skip_unknown_keys=True)
    inset = f.getInputSettings()
    spset = f.getSpatialSettings()
    tpset = f.getTemporalSettings()

    htm = HTMCore(inset, spset, tpset)
    htm.move()
    # десиреализовать объект из строки - jsonpickle.decode(s)

    server = SocketServer(server_port)

    while True:
        data = server.receive_message()
        message = SystemMessages.get_keys_in_text(data)
        data = SystemMessages.clear_keys_in_text(data)
        if SystemMessages.GET_DATA in message:
            # сериализуем нужные нам части в объекте HTMSerialization
            htm_serialization = HTMSerialization(htm)
            server.send_message(jsonpickle.encode(htm_serialization))
            continue

        if SystemMessages.MOVE in message:
            # сериализуем нужные нам части в объекте HTMSerialization

            htm.move()
            htm_serialization = HTMSerialization(htm)
            server.send_message(jsonpickle.encode(htm_serialization))
            continue

        if SystemMessages.MOVE100 in message:
            # сериализуем нужные нам части в объекте HTMSerialization
            for i in range(100):
                htm.move()
            htm_serialization = HTMSerialization(htm)

            server.send_message(jsonpickle.encode(htm_serialization))
            continue

        if SystemMessages.MOVE1000 in message:
            # сериализуем нужные нам части в объекте HTMSerialization
            for i in range(1000):
                htm.move()
            htm_serialization = HTMSerialization(htm)

            server.send_message(jsonpickle.encode(htm_serialization))
            continue

        if SystemMessages.RESTART_WITH_SETTINGS:
            port, settings_id = data.replace('(', '', 1).replace(')', '', 1).split(',')
            setjson = RunSettings.select().where(RunSettings.id == settings_id).get()
            f = SettingsForm.from_json(jsonpickle.json.loads(setjson.json_string), skip_unknown_keys=True)
            inset = f.getInputSettings()
            spset = f.getSpatialSettings()
            tpset = f.getTemporalSettings()

            htm = HTMCore(inset, spset, tpset)
            htm.move()
            server.send_message("ok")
            continue

        if SystemMessages.STOP in message:
            server.send_message("ok")
            server.close()
            break

        print("Сообщение не обработано, нет ключей или .п.")
Ejemplo n.º 2
0

if __name__ == "__main__":
    #  In order to start using WTForms-JSON, you need to first initialize the extension.
    #  This monkey patches some classes and methods within WTForms and adds JSON handling support
    #  wtf ????
    wtforms_json.init()

    active_ports = set()

    running_port = 10100
    runner_server = SocketServer(running_port)
    print(" * Running on port %d (Press CTRL+C to quit)" % running_port)
    while True:
        data = runner_server.receive_message()
        message = SystemMessages.get_keys_in_text(data)
        data = SystemMessages.clear_keys_in_text(data)

        if SystemMessages.TURN_ON_HTM_WITH_SETTINGS in message:
            (port, settings_id) = data.replace('(', '', 1).replace(')', '', 1).split(',')
            (port, settings_id) = (int(port), int(settings_id))
            if port in active_ports:
                client = SocketClient(port)
                client.request((port, settings_id), SystemMessages.RESTART_WITH_SETTINGS)
                client.close()
            else:
                active_ports.add(port)
                thread = threading.Thread(target=start_htm, args=(port, settings_id,))
                thread.daemon = True
                thread.start()