domain.py

#!/usr/bin/python

# domain.py - ZmEu <zmeu@whitehat.ro>
#	- Don't hate me, hate the code.

# Changelog:
#	version 0.1:
#	- first version. (Sun Sep 30 16:12:34 EEST 2018)
#	version 0.2:
#	- added: time.sleep [slow/fast] check & fixed some lines. (Sun Sep 30 18:20:02 EEST 2018)
#	- added: save domains [registered/reserved]. (Sun Sep 30 10:29:29 CDT 2018)
#	- added: functions [whois/api/delete]. (Sun Sep 30 11:00:25 CDT 2018)
#	- added: delete yesterday domains reserved. (Sun Sep 30 11:23:08 CDT 2018)

import subprocess, sys, time
from datetime import date, timedelta
from random import randint

def save(argument1, argument2):
	with open(argument1, "a+") as outfile:
		outfile.write(argument2+"\n")

def delete(argument1, argument2):
	with open(argument1, "r") as infile:
		newlist = [i for i in infile.read().split() if i != argument2.rstrip("\r\n")]
	with open(argument1, "w") as outfile:
		outfile.write("\n".join(newlist))

def whois(argument):
	cmd = "echo '{}\r\n' | nc -i 1 whois.rotld.ro -s 1.2.3.{} 43".format(argument.rstrip("\r\n"), randint(1, 254))
	ps = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
	return ps.communicate()[0]

def api(argument):
	cmd = "curl https://localhost/api --interface ethX -X POST -F 'domeniu={}' -F 'tld=ro'".format(argument.replace(".ro", "").rstrip("\r\n"))
	ps = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)

while True:
	with open(sys.argv[1]) as fp:
		for line in fp:
			print "+ we check \033[0;36m{}\033[0m".format(line.rstrip("\r\n"))

			today = date.today()
			yesterday = oday = date.today() - timedelta(days=1)
			t = "{:%Y-%m-%d}".format(today)
			y = "{:%Y-%m-%d}".format(yesterday)
			output = whois(line.rstrip("\r\n"))

			if "entries" in output.split():
				api(line.rstrip("\r\n"))
				print "+ we found \033[0;32m{}\033[0m free.".format(line.rstrip("\r\n"))
				save("registered.txt", "{}".format(line.rstrip("\r\n")))

			elif t in output.split() or y in output.split():
				print "- we found \033[0;31m{}\033[0m reserved.".format(line.rstrip("\r\n"))
				delete(sys.argv[1], line.rstrip("\r\n"))
				save("deleted.txt", "{}".format(line.rstrip("\r\n")))
			time.sleep(0.09)

# - END OF LINE -