refactor: Add URL shortening functionality
This commit is contained in:
parent
c0533a4772
commit
a4c7f76216
14
crawler.py
14
crawler.py
@ -31,6 +31,18 @@ def save_to_redis(article):
|
|||||||
redis_db.set(key, json.dumps(article), ex=3600*24*7)
|
redis_db.set(key, json.dumps(article), ex=3600*24*7)
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
def url_shorten(url):
|
||||||
|
api_url = "https://s.tunpok.com/api/v2/links"
|
||||||
|
api_key = "HJAs1~1sjd5mX66_Ydg8~9oSuqgkd75VeLhjjni6"
|
||||||
|
headers = {
|
||||||
|
'x-api-key': api_key,
|
||||||
|
}
|
||||||
|
resp = requests.post(api_url, headers=headers, json={"target": url})
|
||||||
|
if resp.status_code == 201:
|
||||||
|
return resp.json()['link']
|
||||||
|
else:
|
||||||
|
return url
|
||||||
|
|
||||||
def crawler():
|
def crawler():
|
||||||
# get article list in html div class name = "acon cl"
|
# get article list in html div class name = "acon cl"
|
||||||
home_url = 'https://www.chiphell.com/'
|
home_url = 'https://www.chiphell.com/'
|
||||||
@ -108,7 +120,7 @@ def crawler():
|
|||||||
'date': date,
|
'date': date,
|
||||||
'category': category,
|
'category': category,
|
||||||
'content': content,
|
'content': content,
|
||||||
'url': url,
|
'url': url_shorten(url),
|
||||||
'article_id': article_id
|
'article_id': article_id
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user