From 6759b9055ceb04c54eb3dc31a64de7854c3e22e1 Mon Sep 17 00:00:00 2001
From: midoks <midoks@163.com>
Date: Fri, 17 Feb 2023 15:01:26 +0800
Subject: [PATCH] ip

---
 plugins/migration_api/index.py | 8 ++++----
 route/__init__.py              | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/plugins/migration_api/index.py b/plugins/migration_api/index.py
index c0aa6eea8..eefa038bd 100755
--- a/plugins/migration_api/index.py
+++ b/plugins/migration_api/index.py
@@ -49,7 +49,7 @@ class classApi:
 
     def __http_post_cookie(self, url, p_data, timeout=1800):
         try:
-            print(url)
+            # print(url)
             res = self._REQUESTS.post(url, p_data, timeout=timeout)
             return res.text
         except Exception as ex:
@@ -64,7 +64,6 @@ class classApi:
         url = self.__MW_PANEL + '/api' + url
         post_data = self.__get_key_data()  # 取签名
         post_data.update(args)
-        print(post_data)
         result = self.__http_post_cookie(url, post_data, timeout)
         try:
             return json.loads(result)
@@ -174,11 +173,12 @@ def stepOne():
     url = args['url']
     token = args['token']
 
-    # api = classApi(url, token)
+    api = classApi(url, token)
     # api = classApi('http://127.0.0.1:7200','HfJNKGP5RPqGvhIOyrwpXG4A2fTjSh9B')
     rdata = api.send('/task/count', {})
+    # print(rdata)
     if type(rdata) != int:
-        return mw.returnJson(False, '验证失败')
+        return mw.returnJson(False, rdata['msg'])
     data = getCfgData()
 
     data['url'] = url
diff --git a/route/__init__.py b/route/__init__.py
index 5311b9deb..5b0d88c70 100755
--- a/route/__init__.py
+++ b/route/__init__.py
@@ -465,7 +465,7 @@ def api(reqClass=None, reqAction=None, reqData=None):
         return mw.returnJson(False, '密钥错误')
 
     if not mw.inArray(data['limit_addr'], request_ip):
-        return mw.returnJson(False, '非法请求')
+        return mw.returnJson(False, '非法请求,请添加IP白名单')
 
     if reqClass == None:
         return mw.returnJson(False, '请指定请求方法类')