def get_mumjolandia_update_package(self, file_name='mumjolandia.tar.gz'):
self.socket_client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.socket_client.connect((ConfigLoader.get_config().server_address,
int(ConfigLoader.get_config().server_port)))
m_to_send = MessageFactory().get('update')
self.socket_client.send(m_to_send.get())
bytes_received = self.__receive_message().get_bytes()
if os.path.isfile(file_name):
os.remove(file_name)
with open(file_name, 'wb') as f:
f.write(bytes_received)
self.socket_client.close()
return os.path.abspath(file_name)
def get_file(self, file_name):
return_value = None
if type(file_name) is not str:
return None
self.socket_client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.socket_client.connect((ConfigLoader.get_config().server_address,
int(ConfigLoader.get_config().server_port)))
m_to_send = MessageFactory().get('get ' + file_name)
self.socket_client.send(m_to_send.get())
bytes_received = self.__receive_message().get_bytes()
if len(bytes_received) > 0:
if os.path.isfile(file_name):
os.remove(file_name)
with open(file_name, 'wb+') as f:
f.write(bytes_received)
return_value = os.path.abspath(file_name)
self.socket_client.close()
return return_value
def send(self, message):
self.__send_message_object(MessageFactory.get(message), self.connect,
self.address)
def send(self, message):
logging.debug('Sending: ' + str(self.address_client) + ':' + str(self.port_client))
m_to_send = MessageFactory().get(message)
self.socket_client.send(m_to_send.get())
return self.__receive_message().get_string()
def receive(self):
received_message = MessageFactory.get(
self.__receive_message(self.connect))
logging.debug(received_message.get_string())
return received_message.get_string()