Merge pull request #4 from DD6181S03/master

add ip and dns lookup

worker/ip.py

@@ -0,0 +1,32 @@
+from zzcore import StdAns
+import requests
+
+class Ans(StdAns):
+    def GETMSG(self):
+        try:
+            ip = self.parms[0]
+        except:
+            ip = ''
+        url = 'https://api.ip.sb/geoip/'
+        if ip == '' or ip == 'help':
+            ret = '您可以使用 "/ip ip地址" 进行查找'
+        else:
+            res = requests.get(url + ip)
+            rsc = res.status_code
+            rem = res.json()
+            if rsc == 400:
+                ret = '不是正确的ip地址，您可以使用 "/ip ip地址" 进行查找'
+            else:
+                rlt = []
+                try:
+                    rlt.append('IP： ' + ip)
+                    rlt.append('国家/地区： ' + rem['country'])
+                    rlt.append('ISP： ' + rem['isp'])
+                    rlt.append('ASN： ' + str(rem['asn']))
+                    rlt.append('ASN组织： ' + rem['asn_organization'])
+                    rlt[1] += ', ' + rem['region']
+                    rlt[1] += ', ' + rem['city']
+                except:
+                    pass
+                ret = '\n'.join(rlt)
+        return ret

worker/ns.py

@@ -0,0 +1,9 @@
+from zzcore import StdAns
+from subprocess import Popen
+
+class Ans(StdAns):
+    def GETMSG(self):
+        domain = self.raw_msg['message'][4:]
+        p = Popen(["nslookup ",domain])
+        msg = p.read()
+        return msg

worker/yiyan.py

@@ -12,11 +12,10 @@ class Ans(StdAns):
             cmd = ''
         if cmd == '帮助':
             msg = '您可以使用以下参数：\n   动画，漫画，游戏，文学，原创，来自网络，其他，影视，诗词，网易云，哲学，抖机灵'
+        elif cmd == '':
+            msg = requests.get(url).text
+        elif cmd in cmdlst:
+            msg = requests.get(url+cmdlst[cmd]).text
         else:
-            if cmd == '':
+            msg = '我不知道 ' + cmd + ' 这个参数，你可以使用"/yiyan 帮助"来获取帮助'
-                msg = requests.get(url).text
-            elif cmd in cmdlst:
-                msg = requests.get(url+cmdlst[cmd]).text
-            else:
-                msg = '我不知道 ' + cmd + ' 这个参数，你可以使用"/yiyan 帮助"来获取帮助'
         return msg