lqg 2 years ago
parent
commit
be0b15796c
4 changed files with 56 additions and 10 deletions
  1. 12 2
      main.py
  2. 8 0
      searchdomain/__init__.py
  3. 17 0
      searchdomain/db.py
  4. 19 8
      searchdomain/searchdomain.py

+ 12 - 2
main.py

@@ -1,6 +1,16 @@
 
-from searchdomain.searchdomain import crawl
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+'''
+@Contact :   liuyuqi.gov@msn.cn
+@Time    :   2023/03/08 17:39:34
+@License :   Copyright © 2017-2022 liuyuqi. All Rights Reserved.
+@Desc    :   enter point
+'''
+
+from searchdomain import SearchDomain
 
 
 if __name__== "__main__":
-    crawl()
+    serachdomain = SearchDomain() 
+    serachdomain.run()

+ 8 - 0
searchdomain/__init__.py

@@ -0,0 +1,8 @@
+from .searchdomain import SearchDomain
+
+def main(parameter_list):
+    """
+    param parameter_list:
+    return:
+    """
+    pass

+ 17 - 0
searchdomain/db.py

@@ -0,0 +1,17 @@
+import os,sys,re,json
+import mysql
+
+class Db():
+    '''Db接口'''
+    def __init__(self):
+        pass
+    def save(domain:str, res:bool):
+        pass
+
+class File(Db):
+    def save(domain: str, res: bool):
+        return super().save(res)
+
+class Mysql(Db):
+    def save(domain: str, res: bool):
+        return super().save(res)

+ 19 - 8
searchdomain/searchdomain.py

@@ -8,22 +8,33 @@ class SearchDomain(object):
     def __init__(self):
         super(SearchDomain,self).__init__()
 
-    def crawl(self, domain:str)->bool:
+    def crawl(self, domain:str)->None:
+        '''
+        检测域名是否可用
+        :params domain 域名:
+        :return true or false'''
+        res=False
         try:
             whi = whois.whois(domain)
-            return false
+            res= False
         except Exception as e:
             if(str(e).index("No match")==0):
-                return true
+                res= True
             else:
-                return False
+                res= False
+        self.saveRes(domain,res)
+
+    def saveRes(damin:str, res:bool):
+        # mysql.save()
+        # file.save()
+        pass
 
     def run(self):
         with open("res/res.json","w",encoding="utf8") as file:
             pool=ThreadPoolExecutor(max_workers=10)
             for i in range(100):
-                pool.submit(crawl, domain)
+                pool.submit(self.crawl, domain)
 
-if ___name__ == '__main__':
-    ss = SearchDomain()
-    ss.run()
+if __name__ == '__main__':
+    sd = SearchDomain()
+    sd.run()