diff --git a/abuseipdb/__init__.py b/abuseipdb/__init__.py index 997c060..5db7803 100644 --- a/abuseipdb/__init__.py +++ b/abuseipdb/__init__.py @@ -102,10 +102,11 @@ def __init__(self, ipAddress, abuseConfidenceScore): class BlacklistedIP: - def __init__(self, ipAddress, totalReports, abuseConfidenceScore): + def __init__(self, ipAddress, countryCode, abuseConfidenceScore, lastReportedAt): self.ipAddress = ipAddress - self.totalReports = totalReports + self.countryCode = countryCode self.abuseConfidenceScore = abuseConfidenceScore + self.lastReportedAt = lastReportedAt class AbuseIPDB: @@ -208,8 +209,9 @@ def get_blacklisted_ips(self, countMinimum=15, maxAgeInDays=60, confidenceMinimu blacklisted_ips.append( BlacklistedIP( ipAddress=d['ipAddress'], - totalReports=d['totalReports'], - abuseConfidenceScore=d['abuseConfidenceScore'] + countryCode=d['countryCode'], + abuseConfidenceScore=d['abuseConfidenceScore'], + lastReportedAt=d['lastReportedAt'] ) ) return blacklisted_ips