PyPtt 2.3.6__tar.gz → 2.3.7__tar.gz

{pyptt-2.3.6 → pyptt-2.3.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PyPtt
-Version: 2.3.6
+Version: 2.3.7
 Summary: PyPtt github: https://github.com/PyPtt/PyPtt
 Author-email: CodingMan <pttcodingman@gmail.com>
 License: GNU Lesser General Public License v3 (LGPLv3)

{pyptt-2.3.6 → pyptt-2.3.7}/PyPtt/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '2.3.6'
+__version__ = '2.3.7'
 __author__ = 'CodingMan'
 __email__ = 'pttcodingman@gmail.com'
 

{pyptt-2.3.6 → pyptt-2.3.7}/PyPtt/_api_call_status.py

@@ -1,3 +1,5 @@
+import re
+
 from . import command
 from . import connect_core
 from . import data_type
@@ -6,6 +8,18 @@ from . import log
 from . import screens
 
 
+# 狀態列有 '[呼叫器]打開' 與 '呼叫器關閉' 兩種寫法, 中括號不保證存在。
+_call_status_pattern = re.compile(r'呼叫器[\]\s]*(打開|拔掉|防水|好友|關閉)')
+
+_call_status_map = {
+    '打開': data_type.CallStatus.ON,
+    '拔掉': data_type.CallStatus.UNPLUG,
+    '防水': data_type.CallStatus.WATERPROOF,
+    '好友': data_type.CallStatus.FRIEND,
+    '關閉': data_type.CallStatus.OFF,
+}
+
+
 def get_call_status(api) -> data_type.CallStatus:
 
     cmd_list = []
@@ -17,34 +31,21 @@ def get_call_status(api) -> data_type.CallStatus:
     cmd = ''.join(cmd_list)
 
     target_list = [
-        connect_core.TargetUnit('[呼叫器]打開', log_level=log.DEBUG, break_detect=True),
-        connect_core.TargetUnit('[呼叫器]拔掉', log_level=log.DEBUG, break_detect=True),
-        connect_core.TargetUnit('[呼叫器]防水', log_level=log.DEBUG, break_detect=True),
-        connect_core.TargetUnit('[呼叫器]好友', log_level=log.DEBUG, break_detect=True),
-        connect_core.TargetUnit('[呼叫器]關閉', log_level=log.DEBUG, break_detect=True),
+        connect_core.TargetUnit(screens.Target.MainMenu, log_level=log.DEBUG, break_detect=True),
         connect_core.TargetUnit('★', log_level=log.DEBUG, response=cmd),
     ]
 
-    for i in range(2):
-        index = api.connect_core.send(cmd, target_list)
-        if index < 0:
-            if i == 0:
-                continue
-            raise exceptions.UnknownError('UnknownError')
-
-    if index == 0:
-        return data_type.CallStatus.ON
-    if index == 1:
-        return data_type.CallStatus.UNPLUG
-    if index == 2:
-        return data_type.CallStatus.WATERPROOF
-    if index == 3:
-        return data_type.CallStatus.FRIEND
-    if index == 4:
-        return data_type.CallStatus.OFF
+    for _ in range(2):
+        if api.connect_core.send(cmd, target_list) == 0:
+            break
+    else:
+        raise exceptions.UnknownError('UnknownError')
 
     ori_screen = api.connect_core.get_screen_queue()[-1]
-    raise exceptions.UnknownError(ori_screen)
+    result = _call_status_pattern.search(ori_screen)
+    if result is None:
+        raise exceptions.UnknownError(ori_screen)
+    return _call_status_map[result.group(1)]
 
 
 def set_call_status(api, call_status) -> None:

{pyptt-2.3.6 → pyptt-2.3.7}/PyPtt/_api_get_time.py

@@ -38,7 +38,8 @@ def get_time(api) -> str:
     # 0:00
 
     for line in line_list:
-        if '星期' in line and '線上' in line and '我是' in line:
+        # 短格式的狀態列寫 '週三' 不是 '星期六', 所以不能比對 '星期'。
+        if '線上' in line and '我是' in line:
             result = pattern.search(line)
             if result is not None:
                 return result.group(0)

{pyptt-2.3.6 → pyptt-2.3.7}/PyPtt/screens.py

@@ -63,10 +63,14 @@ def _str_pos_at_cells(line: str, cells: int) -> int:
 
 
 class Target:
+    # 正式站狀態列有兩種寫法, 短的那種會把逗號後的空格與中括號吃掉:
+    #   [5/23 星期六 16:40] [ 射手時 ]  線上27866人, 我是CodingMan   [呼叫器]打開
+    #   8/19週三22:35   [ 七夕 ]   線上30721人,我是DeepLearning 呼叫器關閉  (h)說明
+    # 只比對兩種格式共同的最短關鍵詞。
     MainMenu = [
         '離開，再見',
-        '人, 我是',
-        '[呼叫器]',
+        '我是',
+        '呼叫器',
     ]
 
     MainMenu_Exiting = [

{pyptt-2.3.6 → pyptt-2.3.7}/PyPtt.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PyPtt
-Version: 2.3.6
+Version: 2.3.7
 Summary: PyPtt github: https://github.com/PyPtt/PyPtt
 Author-email: CodingMan <pttcodingman@gmail.com>
 License: GNU Lesser General Public License v3 (LGPLv3)

{pyptt-2.3.6 → pyptt-2.3.7}/PyPtt.egg-info/SOURCES.txt

@@ -97,5 +97,6 @@ tests/test_search_user.py
 tests/test_service.py
 tests/test_set_board_title.py
 tests/test_set_signature_file.py
+tests/test_status_line.py
 tests/test_uao_codec.py
 tests/test_vt100_parser.py

pyptt-2.3.7/tests/test_status_line.py

@@ -0,0 +1,41 @@
+"""PTT 主選單狀態列有長短兩種寫法, 兩種都必須認得。
+
+長: [5/23 星期六 16:40] [ 射手時 ]    線上27866人, 我是CodingMan      [呼叫器]打開
+短: 8/19週三22:35   [ 七夕 ]    線上30721人,我是DeepLearning 呼叫器關閉     (h)說明
+
+短的那種把逗號後的空格與呼叫器的中括號都吃掉了, 日期也從 '星期六' 縮成 '週三'。
+"""
+import re
+
+from PyPtt import data_type
+from PyPtt._api_call_status import _call_status_map, _call_status_pattern
+from PyPtt._api_get_time import pattern as time_pattern
+from PyPtt.screens import Target
+
+LONG = '[5/23 星期六 16:40] [ 射手時 ]    線上27866人, 我是CodingMan      [呼叫器]打開 '
+SHORT = '8/19週三22:35   [ 七夕 ]    線上30721人,我是DeepLearning 呼叫器關閉     (h)說明'
+
+MENU_BODY = '                      (G)oodbye         離開，再見…'
+
+
+def test_main_menu_markers_match_both_status_lines():
+    for status in (LONG, SHORT):
+        screen = MENU_BODY + '\n' + status
+        assert all(t in screen for t in Target.MainMenu), status
+
+
+def test_get_time_line_filter_and_pattern():
+    for status, expected in ((LONG, '16:40'), (SHORT, '22:35')):
+        assert '線上' in status and '我是' in status
+        assert time_pattern.search(status).group(0) == expected
+
+
+def test_call_status_pattern():
+    assert _call_status_map[_call_status_pattern.search(LONG).group(1)] == data_type.CallStatus.ON
+    assert _call_status_map[_call_status_pattern.search(SHORT).group(1)] == data_type.CallStatus.OFF
+
+
+def test_call_status_pattern_covers_every_state():
+    for word, expected in _call_status_map.items():
+        for form in (f'[呼叫器]{word}', f'呼叫器{word}'):
+            assert _call_status_map[_call_status_pattern.search(form).group(1)] == expected