Merge pull request #9 from louigigr/win-autodays_patch

Win autodays patch

Author: louigigr

README.md

@@ -28,8 +28,11 @@ e.g. ``` abuseipdb -f alert.log ```
 
 > Optional: To save to a file you can apend > {filename} after the command e.g ``` abuseipdb -f alert.log > ThreatReport.txt ```
 
-- Specify the number of days you wish to search and display alerts for.
+- Specify the number of days you wish to search and display alerts for by using the -d or --days switch.
+e.g. ``` abuseipdb -f myip.txt -d 60 ```
+
   - Accepted range is 1 to 365
+  - If no switch is specified the script uses a default value of _30_
   - Contains a really simple failsafe so the script won't break, really helps with the binary version
 - To get help type abuseipdb -h
 
@@ -63,6 +66,8 @@ python3 abuseipdb.py -f file_to_parse.txt
 -t      outputs items in tab seperated values (Default)
 
 -c      outputs items in comma seperated values
+
+-d      specifies number of days
 ```
 
 

abuseipdb.py

@@ -54,18 +54,18 @@ def get_reg(name):
 	help="parses IP Addresses from a single given file",
 	action="store",
 	required=False)
-# required.add_argument(
-	# "-d",
-	# "--days",
-	# help="Number of days to look back in history for alerts",
-	# action="store",
-	# required=False)
+required.add_argument(
+	"-d",
+	"--days",
+	help="Number of days to look back in history for alerts",
+	action="store",
+	required=False)
 
 
 parser.add_argument("-t", "--tsv", help="outputs items in tab seperated values (Default)", action="store_true")
 parser.add_argument("-c", "--csv", help="outputs items in comma seperated values",  action="store_true")
 parser.add_argument("-a", dest="APIKEY", help="stores new API key in registry",  action="store")
-# parser.add_argument("-d", dest="DAYS", help="Number of days to look back in history for alerts",  action="store")
+#parser.add_argument("-d", dest="DAYS", help="Number of days to look back in history for alerts",  action="store")
 
 args = parser.parse_args()
 
@@ -105,7 +105,11 @@ def get_cat(x):
 
 
 def get_report(IP):
-	noofdays=input("Please enter number of days to look back in history for alerts: ")
+	if not args.days:
+		print("Setting days to default: 30")
+		noofdays = 30
+	else:
+		noofdays=args.days
 	try:
 		val=int(noofdays)
 	except ValueError:
@@ -114,6 +118,7 @@ def get_report(IP):
 	if (int(noofdays) <= int(0) or int(noofdays) > int(365)):
 		print("\nYou must specify an integer between 1 and 365.\n")
 		sys.exit()
+
 	#request = 'https://www.abuseipdb.com/check/%s/json?key=%s&days=%s' % (IP, api_key, str(argv[4]))
 	request = 'https://www.abuseipdb.com/check/%s/json?key=%s&days=%s' % (IP, api_key, noofdays)
 	# DEBUG