⭐(sina) filter space in response data

close shidenggui#75
znss2012 · Mar 12, 2020 · 3aafbe2 · 3aafbe2
1 parent e0ea2b5
commit 3aafbe2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/easyquotation/sina.py b/easyquotation/sina.py
@@ -29,6 +29,7 @@ def stock_api(self) -> str:
     def format_response_data(self, rep_data, prefix=False):
         stocks_detail = "".join(rep_data)
         stocks_detail = self.del_null_data_stock.sub('', stocks_detail)
+        stocks_detail = stocks_detail.replace(' ', '')
         grep_str = self.grep_detail_with_prefix if prefix else self.grep_detail
         result = grep_str.finditer(stocks_detail)
         stock_dict = dict()

diff --git a/tests/test_easyquotation.py b/tests/test_easyquotation.py
@@ -9,15 +9,15 @@ def test_stock_code_with_prefix(self):
         cases = ["sina", "qq"]
         for src in cases:
             q = easyquotation.use(src)
-            data = q.all_market
+            data = q.market_snapshot(prefix=True)
             for k in data.keys():
                 self.assertRegex(k, r"(sh|sz)\d{6}")
 
     def test_all(self):
         cases = ["sina", "qq"]
         for src in cases:
             q = easyquotation.use(src)
-            data = q.all
+            data = q.market_snapshot()
             for k in data.keys():
                 self.assertRegex(k, r"\d{6}")