From b136bf7568bf50091591be2c92048f309bfbb9db Mon Sep 17 00:00:00 2001
From: Benjamyn Love <benjamyn.love@gmail.com>
Date: Sun, 9 Sep 2018 05:31:54 +1000
Subject: [PATCH] Added IP lookup functionality to the lib

---
 .gitignore                          |   1 +
 __pycache__/pybusedb.cpython-37.pyc | Bin 843 -> 983 bytes
 pybusedb.py                         |  11 +++++++++--
 test.py                             |  12 ++++++++++++
 4 files changed, 22 insertions(+), 2 deletions(-)
 create mode 100644 .gitignore
 create mode 100644 test.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..bee8a64
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+__pycache__
diff --git a/__pycache__/pybusedb.cpython-37.pyc b/__pycache__/pybusedb.cpython-37.pyc
index ab9b25eabcb3a7ff94ea23167e6085a58005807e..b60b8fb557dfa074d593296a5d8c3a2a3750a63e 100644
GIT binary patch
delta 346
zcmX@jcAcHiiI<m)0SMHTr$n<eZ{)kqSkD6Fg0M3X7smjJ8paxic!m<j1xzVGRu;oT
zrlJra%?!jW;tU`WafTEoAk8WUWC7K(m9PVOY|V^}3?&>3I6>@%jI~TPObfUcGSo8H
zFb6ZFfG|TagC>h#l~PPfwV{r2ZHyj>;Q(hLGpcy>vx@Wc?6XrVZB>h_xIFX95_2+B
z6g&eoxo<IL7TjVgN{wR8O)N>ym~77^8GnnTD7COOwYa4C7IS)P$t@O;ZcUaVcA$TX
zxPini#-b<=kf2LqNn#NPNEl3Tg5=nfGg6bYJp%-QoMI5cz{tkL$0*0h#mK|R!N|fi
dxtqz6QE>8ECIkK=exP&_FNoj)5|ag)tpIQAQAPj&

delta 201
zcmcc4ewvNXiI<m)0SK;2Oo{%^xRLKVV?7g)%M8TMKwNADBx)FI7~&af7_xwL7Lz!L
z#URd*0^%`?0a-v<mJ(JVkENNBks*a4m_d`tuZl-Mt2jT;K0CG2R<&4@?G{sJK@khk
zv?5jz0kR;9IW)+phz-cnWGZ3@iE;yp$*-AYxj=#pjBJc@jFV-U9T|Bh$1xj-M6rNO
XjN&LtEi6qfE-AjnRFpb-F0&N?ulXs(

diff --git a/pybusedb.py b/pybusedb.py
index f13f225..1e37480 100644
--- a/pybusedb.py
+++ b/pybusedb.py
@@ -15,5 +15,12 @@ class APIConnect(object):
 
     def checkIP(self, ip):
         self.ip = ip
-        self.URL = self.baseAPIURL + self.checkIPURL + self.ip + '/json' '?key=%s' % self.APIKEY
-        print(self.URL)
+        #checks to see if we supplied a valid IPv4
+        if re.match('\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}', self.ip):
+            self.URL = self.baseAPIURL + self.checkIPURL + self.ip + '/json' '?key=%s' % self.APIKEY
+            print(self.URL)
+            r = requests.get(self.URL)
+            jsonData = r.json()
+            return jsonData
+        else:
+            return "Invalid IP"
diff --git a/test.py b/test.py
new file mode 100644
index 0000000..7c03ae4
--- /dev/null
+++ b/test.py
@@ -0,0 +1,12 @@
+#!/usr/bin/python3
+
+import pybusedb
+import json
+from pprint import pprint
+
+APIKEY = '213456'
+IP = '178.210.90.90'
+APIConnector = pybusedb.APIConnect(APIKEY)
+
+data = APIConnector.checkIP(IP)
+pprint(data)