fix .head for errors and add test

2013-07-20 22:25:45 -07:00
parent 5eaea6a449
commit 899d063071
2 changed files with 11 additions and 1 deletions
--- a/modules/head.py
+++ b/modules/head.py
@@ -47,7 +47,10 @@ def head(phenny, input):
        info = web.head(uri)
        info['status'] = '200'
    except web.HTTPError as e:
-        return phenny.say(str(e.code))
+        if hasattr(e, 'code'):
+            return phenny.say(str(e.code))
+        else:
+            return phenny.say(str(e.response.status_code))
    except web.ConnectionError:
        return phenny.say("Can't connect to %s" % uri)