jianyu_auto_api/lib/linkapi.py

from hytest import *
from playwright.sync_api import sync_playwright
from PIL import Image, ImageChops
from bs4 import BeautifulSoup
import requests
class APILink():
    #打开链接，返回title
    def obtain_url_title(self,url):
        response = requests.get(url)
        response.encoding = 'utf-8'  # 设置编码为gbk
        soup = BeautifulSoup(response.text, 'html.parser')
        title = soup.title.string
        return title

    #打开链接，status=200，返回true
    def open_url_status(self,url):
        response = requests.get(url)
        status_code = response.status_code
        if status_code==200:
            return True
        else:
            return False

    def fetch_page_source(self,url):
        try:
            response = requests.get(url)
            response.raise_for_status()  # 检查请求是否成功
            return response.text  # 返回网页源代码
        except requests.RequestException as e:
            print(f"Error fetching the URL: {e}")
            return None

    def check_value_in_source(self,url,value):
        page_source = self.fetch_page_source(url)
        if page_source:
            if value in page_source:
                # print(f"'{value_to_check}' exists in the page source.")
                return True
            else:
                # print(f"'{value_to_check}' does not exist in the page source.")
                return False
        else:
            # print("Failed to retrieve the page source.")
            return False


    def setup(self):
        # 初始化 Playwright
        self.playwright = sync_playwright().start()
        self.browser = self.playwright.chromium.launch(headless=True)
        self.page = self.browser.new_page()

    def teardown(self):
        # 关闭浏览器
        self.browser.close()
        self.playwright.stop()

    def obtain_element_text(self,url,element):
        self.page.goto(url)
        # 等待元素出现
        self.page.wait_for_selector(element)
        # 查找元素
        element_handle = self.page.locator(element)

        # 获取元素的文本或属性值
        element_text = element_handle.text_content()
        # element_attribute = element_handle.get_attribute('属性名')
        # 设置超时时间
        self.page.wait_for_timeout(3000)
        return element_text
        # print(f'元素的属性值： {element_attribute}')

    #网页截图模糊遮罩方法，适用于网页有动态元素，进行遮罩比较
    def save_screenshot_mask(self,url, output_path, elements, clip=None):
        locs = []
        self.page.goto(url)
        for element in elements:
            loc = self.page.locator(element)
            locs.append(loc)
        self.page.screenshot(path=output_path,mask=locs, clip=clip)

    #网页直接截图方法，适用于网页元素不变，可直接截图比较
    def save_screenshot(self,url, output_path, clip=None):
        self.page.goto(url)
        self.page.screenshot(path=output_path, clip=clip)

    def compare_images(self, image1_path, image2_path):
        image1 = Image.open(image1_path)
        image2 = Image.open(image2_path)
        diff = ImageChops.difference(image1, image2)
        if diff.getbbox() is None:
            INFO("两张图片完全相同")
            return True
        else:
            INFO("两张图片存在差异")
            return False

#对比样本快照与当前快照
    def contrast_snapshot(self,url,expected_screenshot,current_screenshot,clip):
        # 如果不存在样本照片，生成样本照片
        if not os.path.exists(expected_screenshot):
            self.save_screenshot(url, expected_screenshot,clip)
            INFO(f"样本快照已保存：{expected_screenshot}")
        #生成对比照片
        apilink.save_screenshot(url, current_screenshot,clip)
        INFO(f"对比快照已保存：{expected_screenshot}")
        #返回对比结果
        result = apilink.compare_images(current_screenshot, expected_screenshot)
        return result

#对比样本快照与当前快照2
    def contrast_snapshot_mask(self,url,expected_screenshot,current_screenshot,element,clip):
        # 如果不存在样本照片，生成样本照片
        if not os.path.exists(expected_screenshot):
            self.save_screenshot_mask(url, expected_screenshot,element,clip)
            INFO(f"样本快照已保存：{expected_screenshot}")
        #生成对比照片
        apilink.save_screenshot_mask(url, current_screenshot,element,clip)
        INFO(f"对比快照已保存：{expected_screenshot}")
        #返回对比结果
        result = apilink.compare_images(current_screenshot, expected_screenshot)
        return result

apilink = APILink()