determine charset from response header
This commit is contained in:
parent
36b407e86c
commit
5818b0e096
36
nevernote.py
36
nevernote.py
@ -1,15 +1,43 @@
|
|||||||
#!/usr/bin/python3
|
#!/usr/bin/python3
|
||||||
|
|
||||||
import argparse
|
import argparse
|
||||||
|
import http.client
|
||||||
import sys
|
import sys
|
||||||
import urllib.request
|
|
||||||
|
|
||||||
|
from urllib.parse import urlparse
|
||||||
|
|
||||||
def get_page(url):
|
def get_page(url):
|
||||||
'''download page and decode it to utf-8'''
|
'''download page and decode it to utf-8'''
|
||||||
u = urllib.request.urlopen(url)
|
charset = 'utf-8'
|
||||||
page_binary = u.read(100)
|
|
||||||
page = page_binary.decode()
|
up = urlparse(url)
|
||||||
|
|
||||||
|
headers = {
|
||||||
|
"Host": up.netloc,
|
||||||
|
"Content-Type": "text/html; charset=utf-8",
|
||||||
|
"Connection": "keep-alive",
|
||||||
|
}
|
||||||
|
|
||||||
|
if up.scheme == 'http':
|
||||||
|
conn = http.client.HTTPConnection(up.netloc)
|
||||||
|
elif up.scheme == 'https':
|
||||||
|
conn = http.client.HTTPSConnection(up.netloc)
|
||||||
|
else:
|
||||||
|
print("ERROR: invalid protocol set in '{0}'".format(url))
|
||||||
|
return False
|
||||||
|
|
||||||
|
conn.request("GET", up.path, None, headers)
|
||||||
|
response = conn.getresponse()
|
||||||
|
|
||||||
|
# determine page charset
|
||||||
|
contenttype = response.getheader('Content-Type')
|
||||||
|
if contenttype:
|
||||||
|
charset = contenttype.split('; ')[1].split('=')[1]
|
||||||
|
|
||||||
|
page_binary = response.read()
|
||||||
|
page = page_binary.decode(charset)
|
||||||
|
|
||||||
|
return page
|
||||||
|
|
||||||
|
|
||||||
def write_file(page):
|
def write_file(page):
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user