pep8

oneforall/modules/check/axfr.py

@@ -1,11 +1,11 @@
 """
-查询域名的NS记录(域名服务器记录，记录该域名由哪台域名服务器解析)，检查查出的域名服务器是否开启
-DNS域传送，如果开启且没做访问控制和身份验证便加以利用获取域名的所有记录
+查询域名的NS记录(域名服务器记录，记录该域名由哪台域名服务器解析)，检查查出的域名服务器是
+否开启DNS域传送，如果开启且没做访问控制和身份验证便加以利用获取域名的所有记录。
 
-DNS域传送(DNS zone transfer)指的是一台备用域名服务器使用来自主域名服务器的数据刷新自己的域
-数据库，目的是为了做冗余备份，防止主域名服务器出现故障时 dns 解析不可用。
-当主服务器开启DNS域传送同时又对来请求的备用服务器未作访问控制和身份验证便可以利用此漏洞获取某个
-域的所有记录。
+DNS域传送(DNS zone transfer)指的是一台备用域名服务器使用来自主域名服务器的数据刷新自己
+的域数据库，目的是为了做冗余备份，防止主域名服务器出现故障时 dns 解析不可用。
+当主服务器开启DNS域传送同时又对来请求的备用服务器未作访问控制和身份验证便可以利用此漏洞获
+取某个域的所有记录。
 """
 import dns.resolver
 import dns.zone

oneforall/modules/check/cdx.py

@@ -10,7 +10,6 @@ class CheckCDX(Module):
     """
     检查crossdomain.xml文件收集子域名
     """
-
     def __init__(self, domain: str):
         Module.__init__(self)
         self.domain = self.register(domain)

oneforall/modules/check/cert.py

@@ -31,8 +31,8 @@ class CheckCert(Module):
         except Exception as e:
             logger.log('ERROR', e)
             return
-        subdomains_find = utils.match_subdomain(self.domain, str(cert_dict))
-        self.subdomains = self.subdomains.union(subdomains_find)
+        subdomains = utils.match_subdomain(self.domain, str(cert_dict))
+        self.subdomains = self.subdomains.union(subdomains)
 
     def run(self):
         """

oneforall/modules/check/csp.py

@@ -25,8 +25,10 @@ class CheckCSP(Module):
         正则匹配响应头中的内容安全策略字段以发现子域名
         """
         if not self.header:
-            urls = [f'http://{self.domain}', f'https://{self.domain}',
-                    f'http://www.{self.domain}', f'https://www.{self.domain}']
+            urls = [f'http://{self.domain}',
+                    f'https://{self.domain}',
+                    f'http://www.{self.domain}',
+                    f'https://www.{self.domain}']
             response = None
             for url in urls:
                 self.header = self.get_header()