Fix phishing sync

This commit is contained in:
Zeva Rose 2022-04-23 14:40:57 -06:00
parent 004d8d1091
commit d354211e0c

View file

@ -72,13 +72,21 @@ class Jarvis(Snake):
self.logger.debug(f"Found {len(data)} changes to phishing domains") self.logger.debug(f"Found {len(data)} changes to phishing domains")
add = 0
sub = 0
for update in data: for update in data:
if update["type"] == "add": if update["type"] == "add":
if update["domain"] not in self.phishing_domains: for domain in update["domains"]:
self.phishing_domains.append(update["domain"]) if domain not in self.phishing_domains:
add += 1
self.phishing_domains.append(domain)
elif update["type"] == "delete": elif update["type"] == "delete":
if update["domain"] in self.phishing_domains: for domain in update["domains"]:
self.phishing_domains.remove(update["domain"]) if domain in self.phishing_domains:
sub -= 1
self.phishing_domains.remove(domain)
self.logger.debug(f"{add} additions, {sub} removals")
async def _prerun(self, ctx: Context, *args, **kwargs) -> None: async def _prerun(self, ctx: Context, *args, **kwargs) -> None:
name = ctx.invoked_name name = ctx.invoked_name