fix hooks may not been assignment #148

2019-04-02 16:47:45 +08:00
parent 656dbe87d9
commit 4935582b19
1 changed files with 3 additions and 2 deletions
--- a/py12306/helpers/request.py
+++ b/py12306/helpers/request.py
@@ -33,8 +33,9 @@ class Request(HTMLSession):
        return response

    def add_response_hook(self, hook):
-        exist_hooks = self.hooks['response']
-        if not isinstance(exist_hooks, list): hooks = [exist_hooks]
+        hooks = self.hooks['response']
+        if not isinstance(hooks, list):
+            hooks = [hooks]
        hooks.append(hook)
        self.hooks['response'] = hooks
        return self