Create qimai_data.py

Author: Stephen0124

requirements.txt

@@ -22,3 +22,4 @@ tweepy
 # yulewalker  # uncomment once audio_filters/equal_loudness_filter.py is fixed
 typing_extensions
 xgboost
+DrissionPage ; python_version > '4.0.4.12'

web_programming/qimai_data.py

@@ -1,12 +1,10 @@
 """
-Qimai Data (qimai.cn) is a mobile product intelligence and business analysis platform in China.
+Qimai Data (qimai.cn) is a mobile product business analysis platform in China.
 
 Get a dict of the app information for a given typename (free, paid, grossing) from https://www.qimai.cn
 
-DrissionPage is a simple encapsulation of the selenium package, which is continuously updated, and some of the code could be
-not compatible with the latest version of DrissionPage.
-
 p.s. Current version of DrissionPage : 4.0.4.17
+https://github.com/g1879/DrissionPage
 """
 
 from DrissionPage import ChromiumPage
@@ -21,7 +19,7 @@ def login(page: ChromiumPage, username: str, password: str) -> bool:
             page.ele(".submit").click()
             print("Login successfully")
             return True
-        except:
+        except KeyError:
             print("Login failed")
             page.close()
             return False
@@ -30,7 +28,8 @@ def login(page: ChromiumPage, username: str, password: str) -> bool:
         return True
 
 
-def get_app_info(page: ChromiumPage, username: str, password: str, typename: str) -> dict:
+def get_app_info(page: ChromiumPage,
+                 username: str, password: str, typename: str) -> dict:
     # Clear the cache to avoid the login failure
     page.clear_cache(cookies=True)
     if login(page, username, password):
@@ -82,11 +81,10 @@ def get_app_info(page: ChromiumPage, username: str, password: str, typename: str
                     }
 
     else:
-        dic = {"error": "Login failed"}
+        return {'error': 'Login failed'}
 
     return dic
 
-
 if __name__ == "__main__":
     page = ChromiumPage()
     print(get_app_info(page, "YOUR USERNAME", "YOUR PASSWORD", "free"))