fix issue 174 175

refraction-ray · Jul 11, 2023 · 6dceaa1 · 6dceaa1
1 parent 06a161f
commit 6dceaa1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -35,7 +35,7 @@
         "requests",
         "pyecharts==1.7.1",  # broken api between 0.x and 1.x
         "beautifulsoup4>=4.9.0",
-        "sqlalchemy",
+        "sqlalchemy<2.0",
         "pysocks",  # sock5 proxy support
     ],
     tests_require=["pytest"],

diff --git a/xalpha/info.py b/xalpha/info.py
@@ -540,6 +540,8 @@ def __init__(
                 "This code seems to be a mfund, use ``mfundinfo`` instead"
             )
         code = code.zfill(6)  # 1234 is the same as 001234
+        assert code.isdigit(), "fund code must be a strin of six digits"
+        assert len(code) == 6, "fund code must be a strin of six digits"
         self._url = (
             "http://fund.eastmoney.com/pingzhongdata/" + code + ".js"
         )  # js url api for info of certain fund