def __init__(self, config):
self.config = config
self._RDQ = ReceiveDataQueue()
self._UseTooManyResources = TargetUnit(
[
i18n.UseTooManyResources,
],
screens.Target.UseTooManyResources,
exceptions_=exceptions.UseTooManyResources())
log.show_value(self.config, log.level.INFO, [
i18n.connect_core,
], i18n.Init)
def send(self,
msg: str,
target_list: list,
screen_timeout: int = 0,
refresh: bool = True,
secret: bool = False) -> int:
def clean_screen(recv_screen: str, NoColor: bool = True) -> str:
if not recv_screen:
return recv_screen
# http://asf.atmel.com/docs/latest/uc3l/html/group__group__avr32__lib_utils__print__funcs.html#ga024c3e2852fe509450ebc363df52ae73
# screen = re.sub('\[[\d+;]*m', '', screen)
recv_screen = re.sub(r'[\r]', '', recv_screen)
# recv_screen = re.sub(r'[\x00-\x08]', '', recv_screen)
recv_screen = re.sub(r'[\x00-\x07]', '', recv_screen)
# print(recv_screen)
recv_screen = re.sub(r'[\x0b\x0c]', '', recv_screen)
# screen = re.sub(r'[\x0e-\x1f]', '', screen)
recv_screen = re.sub(r'[\x0e-\x1A]', '', recv_screen)
recv_screen = re.sub(r'[\x1C-\x1F]', '', recv_screen)
recv_screen = re.sub(r'[\x7f-\xff]', '', recv_screen)
recv_screen = screens.vt100(recv_screen)
return recv_screen
if not all(isinstance(T, TargetUnit) for T in target_list):
raise ValueError('Item of TargetList must be TargetUnit')
if self._UseTooManyResources not in target_list:
target_list.append(self._UseTooManyResources)
if screen_timeout == 0:
current_screen_timeout = self.config.screen_timeout
else:
current_screen_timeout = screen_timeout
break_detect_after_send = False
break_index = -1
is_secret = secret
use_too_many_res = False
while True:
if refresh and not msg.endswith(command.Refresh):
msg = msg + command.Refresh
try:
msg = msg.encode('big5uao', 'replace')
except AttributeError:
pass
except Exception as e:
traceback.print_tb(e.__traceback__)
print(e)
msg = msg.encode('big5', 'replace')
if is_secret:
log.show_value(self.config, log.level.DEBUG, [i18n.SendMsg],
i18n.HideSensitiveInfor)
else:
log.show_value(self.config, log.level.DEBUG, [i18n.SendMsg],
msg)
if self.config.connect_mode == connect_mode.TELNET:
try:
self._core.read_very_eager()
self._core.write(msg)
except EOFError:
raise exceptions.ConnectionClosed()
else:
try:
asyncio.get_event_loop().run_until_complete(
self._core.send(msg))
except websockets.exceptions.ConnectionClosedError:
raise exceptions.ConnectionClosed()
except RuntimeError:
raise exceptions.ConnectionClosed()
except websockets.exceptions.ConnectionClosedOK:
raise exceptions.ConnectionClosed()
if break_detect_after_send:
return break_index
msg = ''
receive_data_buffer = bytes()
# print(f'0 {use_too_many_res}')
start_time = time.time()
mid_time = time.time()
while mid_time - start_time < current_screen_timeout:
recv_data_obj = RecvData()
if self.config.connect_mode == connect_mode.TELNET:
try:
recv_data_obj.data = self._core.read_very_eager()
except EOFError:
return -1
else:
try:
asyncio.get_event_loop().run_until_complete(
websocket_receiver(self._core,
current_screen_timeout,
recv_data_obj))
except websockets.exceptions.ConnectionClosed:
# print(f'0.1 {use_too_many_res}')
if use_too_many_res:
# print(f'0.2 {use_too_many_res}')
raise exceptions.UseTooManyResources()
# print(f'0.3 {use_too_many_res}')
raise exceptions.ConnectionClosed()
except websockets.exceptions.ConnectionClosedOK:
raise exceptions.ConnectionClosed()
except asyncio.TimeoutError:
return -1
except RuntimeError:
raise exceptions.ConnectionClosed()
receive_data_buffer += recv_data_obj.data
receive_data_temp = receive_data_buffer.decode(
'big5uao', errors='replace')
screen = clean_screen(receive_data_temp)
# qq = receive_data_temp.encode('utf-8')
# b = None
# for q in qq:
# if not b:
# b = f'bytes([{q}'
# else:
# b += f', {q}'
# b += f'])'
# print(b)
find_target = False
for Target in target_list:
condition = Target.is_match(screen)
if condition:
if Target._Handler is not None:
Target._Handler(screen)
if len(screen) > 0:
screens.show(self.config, screen)
self._RDQ.add(screen)
# self._ReceiveDataQueue.append(screen)
if Target == self._UseTooManyResources:
# print('!!!!!!!!!!!!!!!')
use_too_many_res = True
# print(f'1 {use_too_many_res}')
break
Target.raise_exception()
find_target = True
log.show_value(self.config, Target.get_log_level(),
[i18n.PTT, i18n.Msg],
Target.get_display_msg())
end_time = time.time()
log.show_value(self.config, log.level.DEBUG, [
i18n.SpendTime,
], round(end_time - start_time, 3))
if Target.is_break():
return target_list.index(Target)
msg = Target.get_response(screen)
add_refresh = False
if Target.is_refresh():
add_refresh = True
elif refresh:
add_refresh = True
if add_refresh:
if not msg.endswith(command.Refresh):
msg = msg + command.Refresh
is_secret = Target.is_secret()
if Target.is_break_after_send():
break_index = target_list.index(Target)
break_detect_after_send = True
break
# print(f'2 {use_too_many_res}')
if use_too_many_res:
# print(f'3 {use_too_many_res}')
continue
# print(f'4 {use_too_many_res}')
if find_target:
break
if len(screen) > 0:
screens.show(self.config, screen)
self._RDQ.add(screen)
# self._ReceiveDataQueue.append(screen)
mid_time = time.time()
if not find_target:
# raise exceptions.NoMatchTargetError(self._RDQ)
return -1
# raise exceptions.NoMatchTargetError(self._RDQ)
return -1
def fast_send(self,
msg: str,
target_list: list,
refresh: bool = True) -> int:
break_detect_after_send = False
break_index = -1
use_too_many_res = False
while True:
if refresh and not msg.endswith(command.Refresh):
msg = msg + command.Refresh
try:
msg = msg.encode('big5uao', 'replace')
except AttributeError:
pass
except Exception as e:
traceback.print_tb(e.__traceback__)
print(e)
msg = msg.encode('big5', 'replace')
try:
asyncio.get_event_loop().run_until_complete(
self._core.send(msg))
except websockets.exceptions.ConnectionClosedError:
raise exceptions.ConnectionClosed()
except RuntimeError:
raise exceptions.ConnectionClosed()
except websockets.exceptions.ConnectionClosedOK:
raise exceptions.ConnectionClosed()
if break_detect_after_send:
return break_index
msg = ''
receive_data_buffer = bytes()
# print(f'0 {use_too_many_res}')
start_time = time.time()
mid_time = time.time()
while mid_time - start_time < self.config.screen_timeout:
recv_data_obj = RecvData()
try:
asyncio.get_event_loop().run_until_complete(
websocket_receiver(self._core,
self.config.screen_timeout,
recv_data_obj))
except websockets.exceptions.ConnectionClosed:
# print(f'0.1 {use_too_many_res}')
if use_too_many_res:
# print(f'0.2 {use_too_many_res}')
raise exceptions.UseTooManyResources()
# print(f'0.3 {use_too_many_res}')
raise exceptions.ConnectionClosed()
except websockets.exceptions.ConnectionClosedOK:
raise exceptions.ConnectionClosed()
except asyncio.TimeoutError:
return -1
except RuntimeError:
raise exceptions.ConnectionClosed()
receive_data_buffer += recv_data_obj.data
screen = receive_data_buffer.decode('big5uao',
errors='replace')
find_target = False
for Target in target_list:
condition = Target.is_match(screen)
if condition:
if len(screen) > 0:
screens.show(self.config, screen)
self._RDQ.add(screen)
# self._ReceiveDataQueue.append(screen)
if Target == self._UseTooManyResources:
# print('!!!!!!!!!!!!!!!')
use_too_many_res = True
# print(f'1 {use_too_many_res}')
break
Target.raise_exception()
find_target = True
if Target.is_break():
return target_list.index(Target)
msg = Target.get_response(screen)
add_refresh = False
if Target.is_refresh():
add_refresh = True
elif refresh:
add_refresh = True
if add_refresh:
if not msg.endswith(command.Refresh):
msg = msg + command.Refresh
if Target.is_break_after_send():
break_index = target_list.index(Target)
break_detect_after_send = True
break
# print(f'2 {use_too_many_res}')
if use_too_many_res:
# print(f'3 {use_too_many_res}')
continue
# print(f'4 {use_too_many_res}')
if find_target:
break
if len(screen) > 0:
screens.show(self.config, screen)
self._RDQ.add(screen)
# self._ReceiveDataQueue.append(screen)
mid_time = time.time()
if not find_target:
# raise exceptions.NoMatchTargetError(self._RDQ)
return -1
# raise exceptions.NoMatchTargetError(self._RDQ)
return -1
