Merge pull request #5 from hatamiarash7/fix-problem

Fix problem
This commit is contained in:
Arash Hatami 2023-04-17 19:11:05 +03:30 committed by GitHub
commit 6705e510af
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -23,11 +23,8 @@ def get_ips() -> List[str]:
ips = [] ips = []
# Load IP addresses from TOML file # Load IP addresses from TOML file
lists = toml.load(ip_list) lists = toml.load(ip_list)
# Iterate over groups of IP addresses # Iterate over list
for section in lists: ips = [value for section in lists for value in lists[section].values()]
# Iterate over individual IP addresses
for endpoint_ip in lists[section]:
ips.extend(list(lists[section][endpoint_ip].values()))
# Return a list of unique IP addresses # Return a list of unique IP addresses
return list(set(ips)) return list(set(ips))