fix bugs

CEDA/__init__.py

@@ -1,7 +1,7 @@
 import sys
 import os
 
-__version__ = "0.3.0"
+__version__ = "0.3.2"
 __author__ = "Terence Lau"
 
 

CEDA/economic/XinHua.py

@@ -15,12 +15,14 @@ url = {
 }
 
 class XHData(object):
-    def __init__(self):
+    def __init__(self, country:str=None):
+        self.country = country
         pass
 
     
     def toc(self):
         urls, tid, titles, titles_en = [], [], [], []
+        if self.country == "CN":
             for i in tqdm(range(12005, 12100)):
                 url = "https://api.cnfin.com/roll/charts/getContent?ids={}".format(i)
                 r = requests.get(url)
@@ -34,6 +36,20 @@ class XHData(object):
                         title = json.loads(data["data"]["list"][0]["modelCode"])["title"]["text"]
                         titles.append(title)
                         titles_en.append(translate(text=title))
+        elif self.country == "USA":
+            for i in tqdm(range(6361, 6394)):
+                url = "https://api.cnfin.com/roll/charts/getContent?ids={}".format(i)
+                r = requests.get(url)
+                if r.ok:
+                    data = r.json()
+                    if data["data"] == "图表数据不存在":
+                        pass
+                    else:
+                        urls.append(url)
+                        tid.append(i)
+                        title = json.loads(data["data"]["list"][0]["modelCode"])["title"]["text"]
+                        titles.append(title)
+                        titles_en.append(translate(text=title))
 
         return pd.DataFrame({"urls":urls, "id":tid, "title_zh":titles, "title_en":titles_en})
     
@@ -49,6 +65,6 @@ class XHData(object):
             return ValueError("Something went wrong, try again later")
 
 if __name__ == "__main__":
-    xhdata = XHData()
+    xhdata = XHData(country="CN")
     toc = xhdata.toc()
     data = xhdata.download_data(iid=12006) # GDP

CEDA/market/duka.py

@@ -1,11 +1,5 @@
-import re
-import io
 import requests
-import demjson
 import pandas as pd
-from bs4 import BeautifulSoup
-from datetime import datetime
-from urllib.parse import quote, urlencode
 from fake_useragent import UserAgent
 
 url = {

CEDA/market/marketwatch.py

@@ -1,9 +1,6 @@
-import re
 import io
 import requests
-import demjson
 import pandas as pd
-from bs4 import BeautifulSoup
 from datetime import datetime
 from urllib.parse import quote, urlencode
 from fake_useragent import UserAgent

requirements.txt

@@ -1,6 +1,5 @@
 certifi==2020.12.5
 chardet==4.0.0
-demjson
 fake-useragent==0.1.11
 idna==2.10
 numpy==1.21.0

setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 import os
 setup(
     name = "CEDApy",
-    version = "0.3.0",
+    version = "0.3.2",
     keywords = "quantitative economic data",
     long_description = open(
     os.path.join(
@@ -19,7 +19,6 @@ setup(
         "numpy>=1.15.4",
         "pandas>=0.25",
         "requests>=2.22.0",
-        "demjson>=2.2.4",
         "html5lib>=1.0.1",
         "xlrd==1.2.0",
         "bs4",