处理获取data为None情况

2026-08-26 04:47:48 +08:00 · 2019-09-17 14:01:27 +08:00
parent f7c03656bf
commit edcdaa2f05
1 changed files with 5 additions and 3 deletions
@@ -21,9 +21,11 @@ class VirusTotalAPI(Query):
        resp = self.get(self.addr, params)
        if not resp:
            return
-        data = resp.json()
-        subdomains = set(data.get('subdomains'))
-        self.subdomains = self.subdomains.union(subdomains)
+        json = resp.json()
+        data = json.get('subdomains')
+        if data:
+            subdomains = set(data)
+            self.subdomains = self.subdomains.union(subdomains)

    def run(self):
        """