acidvegas/ptrstream
1
Fork 0
Code Issues Pull Requests Activity

Easier theming of colors via variables

Browse Source
This commit is contained in:
Dionysus 2023-11-23 04:39:04 -05:00
parent 1b3fb6b722
commit 581efb8f6d
Signed by: acidvegas
GPG Key ID: EF4B922DB85DC9DE
1 changed files with 26 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
ptrstream.py
View File

@ -14,6 +14,17 @@ except ImportError:
raise ImportError('missing required \'aiodns\' library (pip install aiodns)') raise ImportError('missing required \'aiodns\' library (pip install aiodns)')
# Colors
class colors:
ip = '\033[35m'
ip_match = '\033[96m'
ptr = '\033[93m'
spooky = '\033[31m'
invalid = '\033[90m'
reset = '\033[0m'
separator = '\033[90m'
def get_dns_servers() -> list: def get_dns_servers() -> list:
'''Get a list of DNS servers to use for lookups.''' '''Get a list of DNS servers to use for lookups.'''
source = urllib.request.urlopen('https://public-dns.info/nameservers.txt') source = urllib.request.urlopen('https://public-dns.info/nameservers.txt')
@ -87,27 +98,27 @@ async def main():
for task in done: for task in done:
ip, result = task.result() ip, result = task.result()
if result: if result:
if result in ('127.0.0.1','localhost'): if result in ('127.0.0.1', 'localhost'):
print(f'\033[35m{ip.ljust(15)}\033[0m \033[90m-> {result}\033[0m') print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}-> {result}{colors.reset}')
elif ip in result: elif ip in result:
result = result.replace(ip, f'\033[96m{ip}\033[93m') result = result.replace(ip, f'{colors.ip_match}{ip}{colors.ptr}')
elif (daship := ip.replace('.', '-')) in result: elif (daship := ip.replace('.', '-')) in result:
result = result.replace(daship, f'\033[96m{daship}\033[93m') result = result.replace(daship, f'{colors.ip_match}{daship}{colors.ptr}')
print(f'\033[35m{ip.ljust(15)}\033[0m \033[90m->\033[0m \033[93m{result}\033[0m') print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}->{colors.reset} {colors.ptr}{result}{colors.reset}')
elif (revip := '.'.join(ip.split('.')[::-1])) in result: elif (revip := '.'.join(ip.split('.')[::-1])) in result:
result = result.replace(revip, f'\033[96m{revip}\033[93m') result = result.replace(revip, f'{colors.ip_match}{revip}{colors.ptr}')
print(f'\033[35m{ip.ljust(15)}\033[0m \033[90m->\033[0m \033[93m{result}\033[0m') print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}->{colors.reset} {colors.ptr}{result}{colors.reset}')
elif result.endswith('.gov') or result.endswith('.mil'): elif result.endswith('.gov') or result.endswith('.mil'):
result = result.replace('.gov', f'\033[31m.gov\033[0m') result = result.replace('.gov', f'{colors.spooky}.gov{colors.reset}')
result = result.replace('.mil', f'\033[31m.gov\033[0m') result = result.replace('.mil', f'{colors.spooky}.gov{colors.reset}')
print(f'\033[35m{ip.ljust(15)}\033[0m \033[90m->\033[0m \033[93m{result}\033[0m') print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}->{colors.reset} {colors.ptr}{result}{colors.reset}')
elif '.gov.' in result or '.mil.' in result: elif '.gov.' in result or '.mil.' in result:
result = result.replace('.gov.', f'\033[31m.gov.\033[0m') result = result.replace('.gov.', f'{colors.spooky}.gov.{colors.reset}')
result = result.replace('.mil.', f'\033[31m.mil.\033[0m') result = result.replace('.mil.', f'{colors.spooky}.mil.{colors.reset}')
print(f'\033[35m{ip.ljust(15)}\033[0m \033[90m->\033[0m \033[93m{result}\033[0m') print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}->{colors.reset} {colors.ptr}{result}{colors.reset}')
else: else:
scary = ('.gov') print(f'{colors.ip}{ip.ljust(15)}{colors.reset} {colors.separator}->{colors.reset} {colors.ptr}{result}{colors.reset}')
print(f'\033[35m{ip.ljust(15)}\033[0m \033[90m->\033[0m \033[93m{result}\033[0m')
results_cache.append(f'{ip}:{result}') results_cache.append(f'{ip}:{result}')
if len(results_cache) >= 1000: if len(results_cache) >= 1000:
stamp = time.strftime('%Y%m%d') stamp = time.strftime('%Y%m%d')