From 53fe00ed58f3401e7bb1da5964d397e8966288a8 Mon Sep 17 00:00:00 2001 From: Pierre-Olivier Mercier Date: Thu, 30 Aug 2018 07:59:41 +0200 Subject: [PATCH] tools/web: new function to retrieve only headers --- nemubot/tools/web.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/nemubot/tools/web.py b/nemubot/tools/web.py index 73a3807..1a4fbd7 100644 --- a/nemubot/tools/web.py +++ b/nemubot/tools/web.py @@ -138,6 +138,21 @@ def _URLConn(cb, url, body=None, timeout=7, header=None, follow_redir=True): finally: conn.close() + +def getURLHeaders(url, body=None, timeout=7, header=None, follow_redir=True): + """Return page headers corresponding to URL or None if any error occurs + + Arguments: + url -- the URL to get + body -- Data to send as POST content + timeout -- maximum number of seconds to wait before returning an exception + """ + + def next(res): + return res.status, res.getheaders() + return _URLConn(next, url=url, body=body, timeout=timeout, header=header, follow_redir=follow_redir) + + def getURLContent(url, body=None, timeout=7, header=None, decode_error=False, max_size=524288): """Return page content corresponding to URL or None if any error occurs