From 5b086b36d3c4f2dda14fb5e42e0f72ad1eaaa799 Mon Sep 17 00:00:00 2001 From: verovaleros Date: Tue, 12 Nov 2024 16:13:26 +0000 Subject: [PATCH] Validate data and raise exception --- lib/aip/models/alpha.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/lib/aip/models/alpha.py b/lib/aip/models/alpha.py index db19e79..44ce497 100644 --- a/lib/aip/models/alpha.py +++ b/lib/aip/models/alpha.py @@ -46,11 +46,14 @@ def run(self, for_date=date.today()): # Get all the attackers IPs attacks = get_attacks(start, end, usecols=['orig']) attacks = pd.concat(attacks).drop_duplicates() - attacks = attacks.rename(columns={'orig':'ip'}) + if not attacks.empty: + attacks = attacks.rename(columns={'orig':'ip'}) - self.blocklist = attacks + self.blocklist = attacks - # Remove IPs from do_not_block_these_ips.csv - self.sanitize() + # Remove IPs from do_not_block_these_ips.csv + self.sanitize() - return self.blocklist + return self.blocklist + else: + raise ValueError("Please check data availability and try again.")