Browse Source

update: enhance fetch script

master
Dnomd343 2 years ago
parent
commit
fae554fe25
  1. 15
      src/wxsy.net/fetch.py

15
src/wxsy.net/fetch.py

@ -7,6 +7,7 @@ Download raw html content as `.html` files.
USAGE: python3 fetch.py [CATALOG] [OUTPUT_DIR] USAGE: python3 fetch.py [CATALOG] [OUTPUT_DIR]
""" """
import os
import sys import sys
import json import json
import time import time
@ -20,21 +21,21 @@ userAgent = ( # default user agent
) )
def httpRequest(url: str, fileName: str) -> bool: # save html content def httpRequest(fileUrl: str, fileName: str) -> bool: # save html content
try: try:
logger.debug('Http request `%s` -> %s' % (url, fileName)) logger.debug('Http request `%s` -> `%s`' % (fileUrl, fileName))
request = requests.get(url, timeout = 30, # timeout -> 30s request = requests.get(fileUrl, timeout = 30, # timeout -> 30s
headers = { headers = {
'user-agent': userAgent, # with fake user-agent 'user-agent': userAgent, # with fake user-agent
} }
) )
if request.status_code not in range(200, 300): # http status code 2xx if request.status_code not in range(200, 300): # http status code 2xx
logger.warning('Http request failed -> %s' % url) logger.warning('Http request failed -> `%s`' % fileUrl)
return False return False
logger.debug('Http request success -> %s' % url) logger.debug('Http request success -> `%s`' % fileUrl)
with open(fileName, 'w') as fileObj: # save html content with open(fileName, 'w') as fileObj: # save html content
fileObj.write(request.text) fileObj.write(request.text)
logger.debug('File save success -> %s' % fileName) logger.debug('File save success -> `%s`' % fileName)
except: except:
return False return False
return True return True
@ -44,7 +45,7 @@ catalog = json.loads(open(sys.argv[1]).read()) # load catalog
for _, chapterId in catalog.items(): # traverse all chapters for _, chapterId in catalog.items(): # traverse all chapters
pageUrl = 'https://www.wxsy.net/novel/57104/read_%s.html' % chapterId pageUrl = 'https://www.wxsy.net/novel/57104/read_%s.html' % chapterId
pageFile = '%s/%s.html' % (sys.argv[2], chapterId) pageFile = os.path.join(sys.argv[2], '%s.html' % chapterId)
if httpRequest(pageUrl, pageFile): # save html content if httpRequest(pageUrl, pageFile): # save html content
logger.info('Page request success -> %s' % pageUrl) logger.info('Page request success -> %s' % pageUrl)
else: else:

Loading…
Cancel
Save