将api加入到插拔式配置

This commit is contained in:
Jing Ling
2020-08-14 13:00:39 +08:00
parent 091167acb2
commit e16b7dada0
23 changed files with 135 additions and 57 deletions
+3 -3
View File
@@ -1,4 +1,4 @@
from config import api
from config import settings
from common.query import Query
@@ -9,8 +9,8 @@ class RiskIQ(Query):
self.module = 'Intelligence'
self.source = 'RiskIQAPIQuery'
self.addr = 'https://api.passivetotal.org/v2/enrichment/subdomains'
self.user = api.riskiq_api_username
self.key = api.riskiq_api_key
self.user = settings.riskiq_api_username
self.key = settings.riskiq_api_key
def query(self):
"""
+2 -2
View File
@@ -1,4 +1,4 @@
from config import api
from config import settings
from common.query import Query
@@ -9,7 +9,7 @@ class ThreatBookAPI(Query):
self.module = 'Intelligence'
self.source = 'ThreatBookAPIQuery'
self.addr = 'https://api.threatbook.cn/v3/domain/sub_domains'
self.key = api.threatbook_api_key
self.key = settings.threatbook_api_key
def query(self):
"""
+2 -2
View File
@@ -1,4 +1,4 @@
from config import api
from config import settings
from common.query import Query
@@ -9,7 +9,7 @@ class VirusTotalAPI(Query):
self.module = 'Intelligence'
self.source = 'VirusTotalAPIQuery'
self.addr = 'https://www.virustotal.com/vtapi/v2/domain/report'
self.key = api.virustotal_api_key
self.key = settings.virustotal_api_key
def query(self):
"""