def __autoreload(self): wait = 3 if os.name == "nt": sub_kwargs = { 'creationflags': subprocess.CREATE_NEW_PROCESS_GROUP } break_signal = signal.CTRL_BREAK_EVENT else: sub_kwargs = {} break_signal = signal.SIGTERM sub_args = [x for x in sys.argv if x != '-r' and x != "--auto-reload"] try: sub_args = [sys.executable] + sub_args process = subprocess.Popen(sub_args, **sub_kwargs) except: Logs.error("Couldn't find a suitable python executable") sys.exit(1) last_mtime = max(self.__file_times(".")) while True: try: max_mtime = max(self.__file_times(".")) if max_mtime > last_mtime: last_mtime = max_mtime Logs.message("Restarting plugin") process.send_signal(break_signal) process = subprocess.Popen(sub_args, **sub_kwargs) time.sleep(wait) except KeyboardInterrupt: process.send_signal(break_signal) break
def __loop(self): to_remove = [] try: while True: if self.__connected == False: elapsed = timer() - self.__disconnection_time if elapsed >= try_reconnection_time: Logs.message("Trying to reconnect...") if self.__connect() == False: self.__disconnection_time = timer() continue else: time.sleep(try_reconnection_time - elapsed) continue if self._network.receive() == False: self.__connected = False self.__disconnect() continue if timer() - self.__last_keep_alive >= keep_alive_time_interval: self.__last_keep_alive = timer() packet = Network._Packet() packet.set(_Plugin._plugin_id, Network._Packet.packet_type_keep_alive, 0) self._network.send(packet) to_remove.clear() for id, session in self._sessions.items(): if session._read_from_plugin() == False: session.close_pipes() to_remove.append(id) for id in to_remove: self._sessions[id]._send_disconnection_message(_Plugin._plugin_id) del self._sessions[id] self._process_manager._update() except KeyboardInterrupt: self.__exit()
def _on_packet_received(self, packet): if packet.packet_type == Network._Packet.packet_type_message_to_plugin: session_id = packet.session_id if session_id in self._sessions: # packet.decompress() self._sessions[session_id]._on_packet_received(packet.payload) return # If we don't know this session_id, try to register it first if _Plugin.__serializer.try_register_session( packet.payload) == True: received_version_table, _, _ = _Plugin.__serializer.deserialize_command( packet.payload, None) version_table = _TypeSerializer.get_best_version_table( received_version_table) self.__on_client_connection(session_id, version_table) # Doesn't register? It's an error else: Logs.warning("Received a command from an unregistered session", session_id) elif packet.packet_type == Network._Packet.packet_type_plugin_connection: _Plugin._plugin_id = packet.plugin_id Logs.message("Registered with plugin ID", _Plugin._plugin_id, "\n=======================================\n") elif packet.packet_type == Network._Packet.packet_type_plugin_disconnection: if _Plugin._plugin_id == -1: if self._description['auth'] == None: Logs.error( "Connection refused by NTS. Are you missing a security key file?" ) else: Logs.error( "Connection refused by NTS. Your security key file might be invalid" ) sys.exit(1) else: Logs.debug("Connection ended by NTS") sys.exit(0) elif packet.packet_type == Network._Packet.packet_type_client_disconnection: try: id = packet.session_id self._sessions[id].signal_and_close_pipes() del self._sessions[id] Logs.debug("Session", id, "disconnected") except: pass elif packet.packet_type == Network._Packet.packet_type_keep_alive: pass else: Logs.warning("Received a packet of unknown type", packet.packet_type, ". Ignoring")
def __parse_args(self): Logs._set_verbose(False) for i in range(1, len(sys.argv)): if sys.argv[i] == "-h": Logs.message("Usage:", sys.argv[1],"[-h] [-a ADDRESS] [-p PORT]") Logs.message(" -h display this help") Logs.message(" -a connects to a NTS at the specified IP address") Logs.message(" -p connects to a NTS at the specified port") Logs.message(" -k specifies a key file to use to connect to NTS") Logs.message(" -n name to display for this plugin in Nanome") Logs.message(" -v enable verbose mode, to display Logs.debug") Logs.message(" -r, --auto-reload restart plugin automatically if a .py or .json file in current directory changes") Logs.message(" --ignore to use with auto-reload. All paths matching this pattern will be ignored, " \ "use commas to specify several. Supports */?/[seq]/[!seq]") sys.exit(0) elif sys.argv[i] == "-a": if i >= len(sys.argv): Logs.error("Error: -a requires an argument") sys.exit(1) self.__host = sys.argv[i + 1] i += 1 elif sys.argv[i] == "-p": if i >= len(sys.argv): Logs.error("Error: -p requires an argument") sys.exit(1) try: self.__port = int(sys.argv[i + 1]) except ValueError: Logs.error("Error: -p argument has to be an integer") sys.exit(1) i += 1 elif sys.argv[i] == "-k": if i >= len(sys.argv): Logs.error("Error: -k requires an argument") sys.exit(1) self.__key_file = sys.argv[i + 1] i += 1 elif sys.argv[i] == "-n": if i >= len(sys.argv): Logs.error("Error: -n requires an argument") sys.exit(1) self._description['name'] = sys.argv[i + 1] i += 1 elif sys.argv[i] == "-v": self.__has_verbose = True Logs._set_verbose(True) elif sys.argv[i] == "-r" or sys.argv[i] == "--auto-reload": self.__has_autoreload = True elif sys.argv[i] == "--ignore": if i >= len(sys.argv): Logs.error("Error: --ignore requires an argument") sys.exit(1) split = sys.argv[i + 1].split(",") self.__to_ignore.extend(split)