Skip to content
This repository has been archived by the owner on Feb 19, 2024. It is now read-only.

Commit

Permalink
Create collect.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Vahid Farid committed Apr 10, 2023
1 parent 88141fe commit be057c6
Showing 1 changed file with 71 additions and 0 deletions.
71 changes: 71 additions & 0 deletions collect.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,71 @@
import time
import json
import dns
import dns.resolver
from datetime import datetime

def main():
result = collect()

with open('list.json', 'w') as json_file:
json_file.write(json.dumps(result, indent=4))

with open('list.txt', 'w') as text_file:
text_file.write(f"Last Update: {result['last_update']}\n\nIPv4:\n")
for el in result["ipv4"]:
text_file.write(f" - {el['ip']:15s} {el['operator']:5s} {el['provider']} {el['created_at']}\n")


def collect():
result = {
"last_update": "",
"last_timestamp": 0,
"ipv4": [],
"ipv6": []
}

providers = json.load(open('providers.json'))
existing_ips = json.load(open('list.json'))
last_update = 0

resolver = dns.resolver.Resolver()
resolver.timeout = 2
resolver.lifetime = 2

for provider in providers:
# IPv4
try:
ipv4_result = resolver.resolve(provider, "A")
for ipv4 in ipv4_result:
ip = ipv4.to_text()
prev = next((el for el in existing_ips["ipv4"] if el["ip"] == ip), None)
created_at = prev["created_at"] if prev else int(time.time())
last_update = created_at if created_at > last_update else last_update
result["ipv4"].append({
"ip": ip,
"operator": providers[provider],
"provider": '.'.join(provider.split('.')[1:]),
"created_at": created_at
})
except:
pass

result["last_update"] = datetime.fromtimestamp(last_update).__str__()
result["last_timestamp"] = last_update

result["ipv4"].sort(key=lambda el: el["created_at"], reverse=True)
result["ipv4"].sort(key=lambda el: el["operator"])
return result

class IP:
def __init__(self, ip, operator, provider, created_at):
self.ip = ip
self.operator = operator
self.provider = provider
self.created_at = created_at

def __repr__(self):
return self.toJson()

if __name__ == '__main__':
main()

0 comments on commit be057c6

Please sign in to comment.