Separate shodanapi to manage calls to shodan

stavinski · Mar 16, 2022 · 04f6426 · 04f6426
1 parent 7b3487e
commit 04f6426
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/shodanapi.py b/shodanapi.py
@@ -0,0 +1,18 @@
+from shodan import Shodan, APIError
+
+class ShodanAPI(object):
+
+    def __init__(self, api_key):
+        self.api = Shodan(api_key)
+
+    def test(self):
+        try:
+            return True, self.api.info()
+        except APIError as e:
+            return False, e
+
+    def host(self, ip):
+        try:
+            return True, ip, self.api.host(ip)
+        except APIError as e:
+            return False, ip, e