From cc362bedfbf0558156e90e053404b6b8d88f5b2f Mon Sep 17 00:00:00 2001
From: TerenceLau <Terenceliu1012@gmail.com>
Date: Thu, 14 Apr 2022 03:07:41 +0000
Subject: [PATCH] add monthly, quarterly, annual dbcode

---
 CEDA/economic/NBSC.py | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/CEDA/economic/NBSC.py b/CEDA/economic/NBSC.py
index c6d9fd5..822aca3 100644
--- a/CEDA/economic/NBSC.py
+++ b/CEDA/economic/NBSC.py
@@ -85,18 +85,28 @@ class NBSCData(object):
                                 "name":self.name, "pid":self.pid, "wdcode":self.wdcode})
         return data
     
-    def download_data(self, nid:str=None, sj="1978-"):
+
+    
+    def download_data(self, nid:str=None, sj="1978-", period:str="monthly"):
+
+        if period == "monthly":
+            dbcode="hgyd"
+        elif period == "quarterly":
+            dbcode="hgjd"
+        elif period == "annual":
+            dbcode="hgnd"
+
         params = {
-        "dbcode": "hgnd",
-        "rowcode": "zb",
         "m": "QueryData",
+        "dbcode": dbcode,
+        "rowcode": "zb",
         "colcode": "sj",
         "wds": "[]",
         "dfwds": '[{"wdcode":"zb","valuecode":"'
         + nid
         + '"},{"wdcode":"sj","valuecode":"'
-        + sj
         + '"}]',
+        "sj": sj
         }
         r = requests.get(self.url, params=params, verify=False, headers=self.generate_header())
         if r.ok: