kennethreitz/requests-html
1
0
Fork 0
mirror of https://github.com/kennethreitz/requests-html.git synced 2026-06-05 23:00:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor

Browse Source 
Signed-off-by: Kenneth Reitz <me@kennethreitz.org>
This commit is contained in:
Kenneth Reitz
2018-02-26 10:23:08 -05:00
parent 3375647666
commit 9fb5adba37
1 changed files with 6 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
requests_html.py
+6 -15
View File
@@ -199,27 +199,18 @@ class Element(BaseParser):
class HTML(BaseParser):
"""An HTML document."""
def __init__(self, *, response):
def __init__(self, *, url, html, default_encoding=DEFAULT_ENCODING):
super(HTML, self).__init__(
element=fromstring(response.text),
html=response.text,
url=response.url,
default_encoding=response.encoding
element=fromstring(html),
html=html,
url=url,
default_encoding=default_encoding
)
def __repr__(self):
return "<HTML url={}>".format(repr(self.url))
def _handle_response(response, **kwargs):
"""Requests HTTP Response handler. Attaches .html property to Response
objects.
"""
response.html = HTML(response=response)
return response
def user_agent(style=None):
"""Returns a random user-agent, if not requested one of a specific
style.
@@ -253,7 +244,7 @@ class Session(requests.Session):
if not response.encoding:
response.encoding = DEFAULT_ENCODING
response.html = HTML(response=response)
response.html = HTML(url=response.url, html=response.text, default_encoding=response.encoding)
return response