def plugin_installation_failure(self, plugin_name, plugin_version): self.logger.warning('{0} plugin installation failure '.format(plugin_name)) if plugin_name in self.plugin_manager.delayed_profiles.keys(): profile = self.plugin_manager.delayed_profiles[plugin_name] self.logger.warning('An error message sending with related profile properties...') related_policy = self.db_service.select('policy', ['version', 'execution_id'], 'id={0}'.format(profile.get_id())) data = dict() data['message'] = "Profil işletilirken eklenti bulunamadı " "ve eksik olan eklenti kurulmaya çalışırken hata ile karşılaşıldı. " "İlgili eklenti Ahenk'e yüklendiğinde, başarısız olan bu profil " "(Başka bir politika tarafından ezilmedikçe) " "çalıştırılacaktır" " Sorunu çözmek için Lider yapılandırma dosyasındaki eklenti dağıtım " "bilgilerinin doğruluğundan ve belirtilen dizinde geçerli eklenti paketinin " "bulunduğundan emin olun." response = Response(type=MessageType.POLICY_STATUS.value, id=profile.get_id(), code=MessageCode.POLICY_ERROR.value, message="Profil işletilirken eklenti bulunamadı " "ve eksik olan eklenti kurulurken hata oluştu", execution_id=related_policy[0][1], policy_version=related_policy[0][0], data=json.dumps(data), content_type=ContentType.APPLICATION_JSON.value) messenger = Scope.get_instance().get_messenger() messenger.send_direct_message(self.message_manager.policy_status_msg(response)) self.logger.warning( 'Error message was sent about {0} plugin installation failure while trying to run a profile') if plugin_name in self.plugin_manager.delayed_tasks.keys(): task = self.plugin_manager.delayed_tasks[plugin_name] self.logger.warning('An error message sending with related task properties...') data = dict() data['message'] = "Görev işletilirken eklenti bulunamadı " "ve eksik olan eklenti kurulmaya çalışırken hata ile karşılaşıldı. " "İlgili eklenti Ahenk'e yüklendiğinde, başarısız olan bu görev " "çalıştırılacaktır" " Sorunu çözmek için Lider yapılandırma dosyasındaki eklenti dağıtım " "bilgilerinin doğruluğundan ve belirtilen dizinde geçerli eklenti paketinin " "bulunduğundan emin olun." response = Response(type=MessageType.TASK_STATUS.value, id=task.get_id(), code=MessageCode.TASK_ERROR.value, message="Görev işletilirken eklenti bulunamadı " "ve eksik olan eklenti kurulmaya çalışırken oluştu.", data=json.dumps(data), content_type=ContentType.APPLICATION_JSON.value) messenger = Scope.get_instance().get_messenger() messenger.send_direct_message(self.message_manager.task_status_msg(response)) self.logger.warning( 'Error message was sent about {0} plugin installation failure while trying to run a task')
def run(self): while self.keep_run: try: try: item_obj = self.in_queue.get(block=True) obj_name = item_obj.obj_name except Exception as e: self.logger.error( '[Plugin] A problem occurred while executing process. Error Message: {0}' .format(str(e))) if obj_name == "TASK": self.logger.debug('[Plugin] Executing task') command = Scope.get_instance().get_plugin_manager( ).find_command(self.getName(), item_obj.get_plugin().get_version(), item_obj.get_command_cls_id().lower()) self.context.put('task_id', item_obj.get_id()) if item_obj.get_file_server( ) is not None and item_obj.get_file_server() != 'null': self.context.put( 'protocol', json.loads(item_obj.get_file_server())['protocol']) self.context.put( 'parameterMap', json.loads( item_obj.get_file_server())['parameterMap']) task_data = item_obj.get_parameter_map() self.logger.debug( '[Plugin] Sending notify to user about task process') if System.Sessions.user_name() is not None and len( System.Sessions.user_name()) > 0: for user in System.Sessions.user_name(): Util.send_notify( "Lider Ahenk", "{0} eklentisi şu anda bir görev çalıştırıyor." .format(self.getName()), System.Sessions.display(user), user) self.logger.debug('[Plugin] Handling task') command.handle_task(task_data, self.context) if self.context.data is not None and self.context.get( 'responseCode') is not None: self.logger.debug('[Plugin] Creating response') response = Response( type=MessageType.TASK_STATUS.value, id=item_obj.get_id(), code=self.context.get('responseCode'), message=self.context.get('responseMessage'), data=self.context.get('responseData'), content_type=self.context.get('contentType')) if response.get_data() and response.get_content_type( ) != ContentType.APPLICATION_JSON.value: success = False try: file_manager = FileTransferManager( json.loads(item_obj.get_file_server()) ['protocol'], json.loads(item_obj.get_file_server()) ['parameterMap']) file_manager.transporter.connect() md5 = str( json.loads(response.get_data())['md5']) success = file_manager.transporter.send_file( System.Ahenk.received_dir_path() + md5, md5) file_manager.transporter.disconnect() except Exception as e: self.logger.error( '[Plugin] A problem occurred while file transferring. Error Message :{0}' .format(str(e))) self.logger.debug('[Plugin] Sending response') message = self.messaging.task_status_msg(response) if success is False: response = Response( type=MessageType.TASK_STATUS.value, id=item_obj.get_id(), code=MessageCode.TASK_ERROR.value, message= 'Task processed successfully but file transfer not completed. Check defined server conf' ) message = self.messaging.task_status_msg( response) Scope.get_instance().get_messenger( ).send_direct_message(message) else: self.logger.debug('[Plugin] Sending task response') Scope.get_instance().get_messenger( ).send_direct_message( self.messaging.task_status_msg(response)) else: self.logger.error( '[Plugin] There is no Response. Plugin must create response after run a task!' ) elif obj_name == "PROFILE": self.logger.debug('[Plugin] Executing profile') profile_data = item_obj.get_profile_data() policy_module = Scope.get_instance().get_plugin_manager( ).find_policy_module(item_obj.get_plugin().get_name(), item_obj.get_plugin().get_version()) self.context.put('username', item_obj.get_username()) execution_id = self.get_execution_id(item_obj.get_id()) policy_ver = self.get_policy_version(item_obj.get_id()) self.context.put('policy_version', policy_ver) self.context.put('execution_id', execution_id) # if item_obj.get_file_server() is not None and item_obj.get_file_server() !='null': # self.context.put('protocol', json.loads(item_obj.get_file_server())['protocol']) # self.context.put('parameterMap', json.loads(item_obj.get_file_server())['parameterMap']) self.logger.debug( '[Plugin] Sending notify to user about profile process' ) Util.send_notify( "Lider Ahenk", "{0} eklentisi şu anda bir profil çalıştırıyor.". format(self.getName()), System.Sessions.display(item_obj.get_username()), item_obj.get_username()) self.logger.debug('[Plugin] Handling profile') policy_module.handle_policy(profile_data, self.context) if self.context.data is not None and self.context.get( 'responseCode') is not None: self.logger.debug('[Plugin] Creating response') response = Response( type=MessageType.POLICY_STATUS.value, id=item_obj.get_id(), code=self.context.get('responseCode'), message=self.context.get('responseMessage'), data=self.context.get('responseData'), content_type=self.context.get('contentType'), execution_id=execution_id, policy_version=policy_ver) if response.get_data() and response.get_content_type( ) != ContentType.APPLICATION_JSON.value: success = False try: file_manager = FileTransferManager( json.loads(item_obj.get_file_server()) ['protocol'], json.loads(item_obj.get_file_server()) ['parameterMap']) file_manager.transporter.connect() md5 = str( json.loads(response.get_data())['md5']) success = file_manager.transporter.send_file( System.Ahenk.received_dir_path() + md5, md5) file_manager.transporter.disconnect() except Exception as e: self.logger.error( '[Plugin] A problem occurred while file transferring. Error Message :{0}' .format(str(e))) self.logger.debug('[Plugin] Sending response') message = self.messaging.task_status_msg(response) if success is False: response = Response( type=MessageType.POLICY_STATUS.value, id=item_obj.get_id(), code=MessageCode.POLICY_ERROR.value, message= 'Policy processed successfully but file transfer not completed. Check defined server conf' ) message = self.messaging.task_status_msg( response) Scope.get_instance().get_messenger( ).send_direct_message(message) else: self.logger.debug( '[Plugin] Sending profile response') Scope.get_instance().get_messenger( ).send_direct_message( self.messaging.policy_status_msg(response)) else: self.logger.error( '[Plugin] There is no Response. Plugin must create response after run a policy!' ) elif 'MODE' in obj_name: module = Scope.get_instance().get_plugin_manager( ).find_module(obj_name, self.name) if module is not None: if item_obj.obj_name in ('LOGIN_MODE', 'LOGOUT_MODE', 'SAFE_MODE'): self.context.put('username', item_obj.username) try: self.logger.debug( '[Plugin] {0} is running on {1} plugin'.format( str(item_obj.obj_name), str(self.name))) module.handle_mode(self.context) except Exception as e: self.logger.error( '[Plugin] A problem occurred while running {0} on {1} plugin. Error Message: {2}' .format(str(obj_name), str(self.name), str(e))) if item_obj.obj_name is 'SHUTDOWN_MODE': self.logger.debug( '[Plugin] {0} plugin is stopping...'.format( str(self.name))) self.keep_run = False else: self.logger.warning( "[Plugin] Not supported object type: {0}".format( str(obj_name))) self.context.empty_data() except Exception as e: self.logger.error( "[Plugin] Plugin running exception. Exception Message: {0} " .format(str(e)))
def run(self): while self.keep_run: try: try: item_obj = self.in_queue.get(block=True) obj_name = item_obj.obj_name except Exception as e: self.logger.error( "[Plugin] A problem occurred while executing process. Error Message: {0}".format(str(e)) ) if obj_name == "TASK": self.logger.debug("[Plugin] Executing task") command = ( Scope.get_instance() .get_plugin_manager() .find_command( self.getName(), item_obj.get_plugin().get_version(), item_obj.get_command_cls_id().lower() ) ) self.context.put("task_id", item_obj.get_id()) if item_obj.get_file_server() is not None and item_obj.get_file_server() != "null": self.context.put("protocol", json.loads(item_obj.get_file_server())["protocol"]) self.context.put("parameterMap", json.loads(item_obj.get_file_server())["parameterMap"]) task_data = item_obj.get_parameter_map() self.logger.debug("[Plugin] Sending notify to user about task process") if System.Sessions.user_name() is not None and len(System.Sessions.user_name()) > 0: for user in System.Sessions.user_name(): Util.send_notify( "Lider Ahenk", "{0} eklentisi şu anda bir görev çalıştırıyor.".format(self.getName()), System.Sessions.display(user), user, ) self.logger.debug("[Plugin] Handling task") command.handle_task(task_data, self.context) if self.context.data is not None and self.context.get("responseCode") is not None: self.logger.debug("[Plugin] Creating response") response = Response( type=MessageType.TASK_STATUS.value, id=item_obj.get_id(), code=self.context.get("responseCode"), message=self.context.get("responseMessage"), data=self.context.get("responseData"), content_type=self.context.get("contentType"), ) if response.get_data() and response.get_content_type() != ContentType.APPLICATION_JSON.value: success = False try: file_manager = FileTransferManager( json.loads(item_obj.get_file_server())["protocol"], json.loads(item_obj.get_file_server())["parameterMap"], ) file_manager.transporter.connect() md5 = str(json.loads(response.get_data())["md5"]) success = file_manager.transporter.send_file( System.Ahenk.received_dir_path() + md5, md5 ) file_manager.transporter.disconnect() except Exception as e: self.logger.error( "[Plugin] A problem occurred while file transferring. Error Message :{0}".format( str(e) ) ) self.logger.debug("[Plugin] Sending response") message = self.messaging.task_status_msg(response) if success is False: response = Response( type=MessageType.TASK_STATUS.value, id=item_obj.get_id(), code=MessageCode.TASK_ERROR.value, message="Task processed successfully but file transfer not completed. Check defined server conf", ) message = self.messaging.task_status_msg(response) Scope.get_instance().get_messenger().send_direct_message(message) else: self.logger.debug("[Plugin] Sending task response") Scope.get_instance().get_messenger().send_direct_message( self.messaging.task_status_msg(response) ) else: self.logger.error( "[Plugin] There is no Response. Plugin must create response after run a task!" ) elif obj_name == "PROFILE": self.logger.debug("[Plugin] Executing profile") profile_data = item_obj.get_profile_data() policy_module = ( Scope.get_instance() .get_plugin_manager() .find_policy_module(item_obj.get_plugin().get_name(), item_obj.get_plugin().get_version()) ) self.context.put("username", item_obj.get_username()) execution_id = self.get_execution_id(item_obj.get_id()) policy_ver = self.get_policy_version(item_obj.get_id()) self.context.put("policy_version", policy_ver) self.context.put("execution_id", execution_id) # if item_obj.get_file_server() is not None and item_obj.get_file_server() !='null': # self.context.put('protocol', json.loads(item_obj.get_file_server())['protocol']) # self.context.put('parameterMap', json.loads(item_obj.get_file_server())['parameterMap']) self.logger.debug("[Plugin] Sending notify to user about profile process") Util.send_notify( "Lider Ahenk", "{0} eklentisi şu anda bir profil çalıştırıyor.".format(self.getName()), System.Sessions.display(item_obj.get_username()), item_obj.get_username(), ) self.logger.debug("[Plugin] Handling profile") policy_module.handle_policy(profile_data, self.context) if self.context.data is not None and self.context.get("responseCode") is not None: self.logger.debug("[Plugin] Creating response") response = Response( type=MessageType.POLICY_STATUS.value, id=item_obj.get_id(), code=self.context.get("responseCode"), message=self.context.get("responseMessage"), data=self.context.get("responseData"), content_type=self.context.get("contentType"), execution_id=execution_id, policy_version=policy_ver, ) if response.get_data() and response.get_content_type() != ContentType.APPLICATION_JSON.value: success = False try: file_manager = FileTransferManager( json.loads(item_obj.get_file_server())["protocol"], json.loads(item_obj.get_file_server())["parameterMap"], ) file_manager.transporter.connect() md5 = str(json.loads(response.get_data())["md5"]) success = file_manager.transporter.send_file( System.Ahenk.received_dir_path() + md5, md5 ) file_manager.transporter.disconnect() except Exception as e: self.logger.error( "[Plugin] A problem occurred while file transferring. Error Message :{0}".format( str(e) ) ) self.logger.debug("[Plugin] Sending response") message = self.messaging.task_status_msg(response) if success is False: response = Response( type=MessageType.POLICY_STATUS.value, id=item_obj.get_id(), code=MessageCode.POLICY_ERROR.value, message="Policy processed successfully but file transfer not completed. Check defined server conf", ) message = self.messaging.task_status_msg(response) Scope.get_instance().get_messenger().send_direct_message(message) else: self.logger.debug("[Plugin] Sending profile response") Scope.get_instance().get_messenger().send_direct_message( self.messaging.policy_status_msg(response) ) else: self.logger.error( "[Plugin] There is no Response. Plugin must create response after run a policy!" ) elif "MODE" in obj_name: module = Scope.get_instance().get_plugin_manager().find_module(obj_name, self.name) if module is not None: if item_obj.obj_name in ("LOGIN_MODE", "LOGOUT_MODE", "SAFE_MODE"): self.context.put("username", item_obj.username) try: self.logger.debug( "[Plugin] {0} is running on {1} plugin".format(str(item_obj.obj_name), str(self.name)) ) module.handle_mode(self.context) except Exception as e: self.logger.error( "[Plugin] A problem occurred while running {0} on {1} plugin. Error Message: {2}".format( str(obj_name), str(self.name), str(e) ) ) if item_obj.obj_name is "SHUTDOWN_MODE": self.logger.debug("[Plugin] {0} plugin is stopping...".format(str(self.name))) self.keep_run = False else: self.logger.warning("[Plugin] Not supported object type: {0}".format(str(obj_name))) self.context.empty_data() except Exception as e: self.logger.error("[Plugin] Plugin running exception. Exception Message: {0} ".format(str(e)))