refactor: move code to lib

pages.py → lib/pages.py

@@ -1,9 +1,9 @@
 from datetime import datetime
 from time import sleep
-from typing import List, Literal
+from typing import List
 from playwright.sync_api import Page
 
-from log_parser import WebItem
+from lib.log_parser import WebItem
 
 
 class TranslatorPage:

report.py → lib/report.py

@@ -2,8 +2,8 @@ from typing import List
 from dataclasses import dataclass, astuple
 
 from tabulate import tabulate
-from log_parser import LogTag, LogItem, WebItem
-from utils import save_csv, run_textdistance, highlight_diff
+from lib.log_parser import LogTag, LogItem, WebItem
+from lib.utils import save_csv, run_textdistance, highlight_diff
 
 @dataclass
 class Row:

main.py

@@ -1,13 +1,5 @@
-from report import LogReport
-from audio import play_audio
 import time
-import os
-import subprocess
-from playwright.sync_api import sync_playwright
-from utils import cmd
-from environment import APP_PATH, DEBUG_PORT
-from page_runner import PageRunner
-from audio import play_audio_until_end
+from lib.audio import play_audio_until_end
 from tests.test_data import test_audios
 
 if __name__ == '__main__':

scripts/test_pages.py

@@ -1,8 +1,6 @@
-import re
 import time
-from time import sleep
-from playwright.sync_api import Page, expect
-from audio import play_audio, get_length
+from playwright.sync_api import Page
+from lib.audio import play_audio, get_length
 
 
 def test_has_title(page: Page, context):

tests/conftest.py

@@ -1,16 +1,11 @@
-import subprocess
-import time
-
-from numpy.core.numeric import True_
 from pytest import fixture
-from playwright.sync_api import Page, expect
 from playwright.sync_api import sync_playwright
 
 from environment import *
-from app_runner import AppRunner
-from page_runner import PageRunner
-from report import AccuracyReport, DelayReport, LogReport
-from utils import get_time_str
+from lib.app_runner import AppRunner
+from lib.page_runner import PageRunner
+from lib.report import AccuracyReport, DelayReport, LogReport
+from lib.utils import get_time_str
 
 @fixture(scope="session")
 def log_file():

tests/test_accuracy.py

@@ -2,10 +2,10 @@ import time
 from pathlib import Path
 import pytest
 
-from audio import play_audio_until_end, get_length
-from pages import TranslatorPage
+from lib.audio import play_audio_until_end, get_length
+from lib.pages import TranslatorPage
 from tests.test_data import test_audios, audio_texts
-from report import AccuracyReport, AccuracyItem
+from lib.report import AccuracyReport, AccuracyItem
 
 @pytest.mark.parametrize("audio", test_audios.get("zh"))
 def test_accuracy_zh2en(app, page: TranslatorPage, accuracy_report: AccuracyReport, audio:Path):

tests/test_delay.py

@@ -1,9 +1,9 @@
 from pathlib import Path
 import pytest
 
-from audio import get_length, play_audio
-from pages import TranslatorPage
-from report import DelayReport, DelayItem, LogReport
+from lib.audio import get_length, play_audio
+from lib.pages import TranslatorPage
+from lib.report import DelayReport, DelayItem, LogReport
 from tests.test_data import test_audios
 
 @pytest.mark.parametrize("audio", test_audios.get("zh"))

tests/test_logfile.py

@@ -1,10 +1,6 @@
 from pathlib import Path
 import pytest
 
-from report import LogReport, DelayItem
-from environment import REPORTS_DIR
-from utils import get_time_str
-
 test_files = [
     # Path("/Users/jeqin/work/code/Translator/translator.log"),
     Path("/Users/jeqin/Downloads/translator.log"),