1

console.py

@@ -5,10 +5,32 @@
 @Time    :   2024/04/26
 @License :   Copyright © 2017-2022 liuyuqi. All Rights Reserved.
 @Desc    :   windows console mode
+
 """
+import argparse
+import os
+import sys
+from searchdomain import SearchDomain
+
+print('''
+      Search Domain 域名搜索工具 v1.0
+      author: liuyuqi
+      email: liuyuqi.gov@msn.cn
+      ''')
+
+comnand = input('''
+        1. 域名批量生成
+        2. 域名批量查询（查询是否可以注册）
+        3. 退出
+        请选择操作: 
+        ''')
 
-def main():
-    pass
+if comnand == '1':
+    domain = input('请输入域名: ')
+    search = SearchDomain(domain)
+    search.run()
+elif comnand == '2':
+    sys.exit(0)
 
 if __name__=='__main__':
     pass

scripts/start.bat

@@ -4,8 +4,48 @@ REM @Contact :   liuyuqi.gov@msn.cn
 REM @Time    :   2024/04/26 16:02:11
 REM @Version :   1.0
 REM @License :   (C)Copyright 2019 liuyuqi.
-REM @Desc    :   None
+REM @Desc    :   调用 search_domain.exe 查询域名，本脚本提示用户输入相应的参数启动程序
 REM %1 - ext_name
 REM %2 - characters replaced
 REM %3 - new characters
 REM ***************************************************************************
+
+
+REM 1. 打印help
+:help
+echo 请输入相应的参数启动程序
+echo 1. 生成域名
+echo 2. 检索域名
+set /p input=请输入相应的参数:
+if "%input%"=="1" goto make_domain
+if "%input%"=="2" goto search_domain
+goto help
+
+REM 2. 生成域名
+:make_domain
+echo 请输入域名长度:
+set /p length=请输入域名长度:
+echo 请输入域名后缀:
+set /p ext_name=请输入域名后缀:
+echo 请输入替换字符:
+set /p characters=请输入替换字符:
+echo 请输入新字符:
+set /p new_characters=请输入新字符:
+echo 生成域名中...
+search_domain.exe make %ext_name% %characters% %new_characters% %length%
+pause
+goto help
+
+REM 3. 检索域名
+:search_domain
+echo 请输入域名后缀:
+set /p ext_name=请输入域名后缀:
+echo 请输入替换字符:
+set /p characters=请输入替换字符:
+echo 请输入新字符:
+set /p new_characters=请输入新字符:
+echo 检索域名中...
+search_domain.exe search %ext_name% %characters% %new_characters%
+
+
+pause

searchdomain/generate_domain.py

@@ -23,6 +23,7 @@ class GenerateDomain(object):
         self.composeDomain=[]
 
     def run(self):
+        """ 批量生成域名, 保存到 domain.txt """
         with open(("data/a.csv"), "r", encoding="utf-8") as f:
             csv_data = csv.reader(f)
             for row in csv_data:

searchdomain/searchdomain.py

@@ -58,6 +58,7 @@ class SearchDomain(object):
                 self.saveRes(domain, res)
 
     def saveRes(self, domain: str, res: bool):
+        """ save result to file """
         # db.Mysql().save()
         db.File().save(self.output, domain + "    " + str(res))