From 6714ed116e94545ead2c2870ca6254394e6f6521 Mon Sep 17 00:00:00 2001 From: g1879 Date: Fri, 6 Nov 2020 18:04:53 +0800 Subject: [PATCH] =?UTF-8?q?DrissionElement=E5=A2=9E=E5=8A=A0page=E5=B1=9E?= =?UTF-8?q?=E6=80=A7?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- DrissionPage/common.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/DrissionPage/common.py b/DrissionPage/common.py index 1ea1229..a27221a 100644 --- a/DrissionPage/common.py +++ b/DrissionPage/common.py @@ -10,20 +10,19 @@ from re import split as re_SPLIT from shutil import rmtree from typing import Union -from lxml.etree import _Element -# from lxml.html import HtmlElement -# from requests_html import Element +from lxml.html import HtmlElement from selenium.webdriver.remote.webelement import WebElement class DrissionElement(object): """SessionElement和DriverElement的基类""" - def __init__(self, ele: Union[WebElement, _Element]): + def __init__(self, ele: Union[WebElement, HtmlElement], page=None): self._inner_ele = ele + self.page = page @property - def inner_ele(self) -> Union[WebElement, _Element]: + def inner_ele(self) -> Union[WebElement, HtmlElement]: return self._inner_ele @property