helper_functions.py

import csv 
#Loading the malicious url from the csv file.
def load_malicious_urls_from_csv(filename):
    urls = []
    print(f"--- Loading malicious URLs from '{filename}'... ---")
    try:
        with open(filename, 'r', encoding='utf-8') as f:
            reader = csv.reader(f)
            next(reader) 
            for row in reader:
                if row:  
                    urls.append(row[1])
    except FileNotFoundError:
        print(f"   [Error] The file '{filename}' was not found.")
        print("   Please download it from PhishTank and save it in the same directory.")
        return None
    except Exception as e:
        print(f"   [Error] An error occurred while reading the file: {e}")
        return None
        
    print(f"--- Loaded {len(urls)} URLs successfully ---\n")
    return urls
#Loading the ips from the text file.
def load_ips_from_text_file(filename):
    ips = []
    print(f"--- Loading malicious IPs from '{filename}' ---")
    try:
        with open(filename,'r') as f:
            reader = f.readlines()
        for i in reader:
            ips.append(i.rstrip("\n"))
    except:   
        print("Error loading the file")
    print(f"--- Loaded {len(ips)} URLs successfully ---\n")
    return ips