Browse Source

元博网 - update

dongzhaorui 3 years ago
parent
commit
3bf86a8a97
1 changed files with 9 additions and 5 deletions
  1. 9 5
      ybw/utils/execptions.py

+ 9 - 5
ybw/utils/execptions.py

@@ -1,11 +1,15 @@
 class YbwCrawlError(Exception):
 
     def __init__(self, *args, **kwargs):
-        if 'code' not in kwargs and 'reason' not in kwargs:
-            kwargs['code'] = 10000
-            kwargs['reason'] = '元博网采集未知错误,请手动处理'
-        for key, val in kwargs.items():
-            setattr(self, key, val)
+        self.code = kwargs.get('code', 10000)
+        self.reason = kwargs.get('reason', '元博网采集未知错误,请手动处理')
+
+        if 'code' not in kwargs:
+            kwargs['code'] = self.code
+        if 'reason' not in kwargs:
+            kwargs['reason'] = self.reason
+
+        [setattr(self, key, val) for key, val in kwargs.items()]
         super(YbwCrawlError, self).__init__(*args, kwargs)