blackteay/DrissionPage
1
0
Fork 0
mirror of https://gitee.com/g1879/DrissionPage.git synced 2024-12-10 04:00:23 +08:00
Code Issues Projects Releases Wiki Activity

完善保存图片有时页面和后台url不一致的问题

Browse Source
This commit is contained in:
g1879 2023-02-28 20:36:55 +08:00
parent 753834e971
commit 56539b2182
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
DrissionPage/chromium_element.py
View File

@ -403,7 +403,7 @@ class ChromiumElement(DrissionElement):
def get_src(self): def get_src(self):
"""返回元素src资源，base64的会转为bytes返回，其它返回str""" """返回元素src资源，base64的会转为bytes返回，其它返回str"""
src = self.attr('src') src = self.prop('currentSrc')
if not src: if not src:
return None return None
@ -441,7 +441,7 @@ class ChromiumElement(DrissionElement):
raise TypeError('该元素无可保存的内容或保存失败。') raise TypeError('该元素无可保存的内容或保存失败。')
path = path or '.' path = path or '.'
rename = rename or basename(self.attr('src')) rename = rename or basename(self.prop('currentSrc'))
write_type = 'wb' if isinstance(data, bytes) else 'w' write_type = 'wb' if isinstance(data, bytes) else 'w'
Path(path).mkdir(parents=True, exist_ok=True) Path(path).mkdir(parents=True, exist_ok=True)