siat 3.8.1__py3-none-any.whl → 3.8.2__py3-none-any.whl

siat/economy2.py

@@ -53,7 +53,7 @@ def find_economic_indicator(key_words='GDP',top=10, \
 def indicator_wb(key_words='GDP',top=20,note=False,translate=False):
    """
    ============================================================================
-   功能：在世界银行数据库中查找宏观经济指标的代码
+   功能：在WB/IMF/FRED数据库中查找宏观经济指标的代码
    参数：
    key_words：可包括多个关键词，使用空格隔开，不区分大小写
    top：输出相似度最高的，默认5个
@@ -73,7 +73,7 @@ def indicator_wb(key_words='GDP',top=20,note=False,translate=False):
        try:
            df_tmp=wb.search(word)
        except:
-           print("  Sorry, currently source rejected connection, try again later")
+           print("  Sorry, data source rejected connection, try again later")
            return None
        
        # 合并
@@ -94,6 +94,10 @@ def indicator_wb(key_words='GDP',top=20,note=False,translate=False):
    df2=fuzzy_search_wb(df,key_words=key_words,column='name2',top=top)
    
    # 遍历输出
+   if len(df2)==0:
+       print(f"Sorry, no indicator found with key words: {key_words}")
+       return None
+       
    #print('') #空一行
    for row in df2.itertuples():
        
@@ -189,6 +193,50 @@ def indicator_name_wb(indicator='NY.GDP.MKTP.KN'):
     return indicator_name
 
 
+#==============================================================================
+if __name__ =="__main__":
+    ticker='CN'; show_name=True
+    check_country_code('ZWE',show_name=True)
+    check_country_code('ZAF',show_name=True)
+    check_country_code('cn',show_name=True)
+    
+def check_country_code(ticker='CN',show_name=False):
+    """
+    ===========================================================================
+    功能：检查国家代码是否支持
+    ticker：国家代码
+    show_name：是否显示国家名称，默认否False
+    
+    返回值：若国家代码在列表中，True；否则，False
+    """
+    country_codes=wb.country_codes
+    
+    elements_to_remove = ['all','ALL','All']
+    country_code_list = [x for x in country_codes if x not in elements_to_remove]
+    
+    result=False
+    if ticker in country_code_list:
+        result=True
+        
+    if show_name:
+        if result:
+            indicator='NY.GDP.MKTP.KN'
+            df=economy_indicator_wb(ticker=ticker,indicator=indicator, \
+                                  start='2000',graph=False)
+            if not (df is None):
+                if len(df) >= 1:
+                    country_name=df['country'].values[0]
+                    print(f"Country code {ticker} refers to {country_name}")
+                else:
+                    print(f"Country code {ticker} found, but its name not found")
+            else:
+                print(f"Found country code {ticker}, but its name not found")
+        else:
+            print(f"Country code {ticker} not found")
+    
+    return result
+
+
 #==============================================================================
 if __name__ =="__main__":
     ticker='CN'
@@ -318,7 +366,7 @@ def economy_indicator_wb(ticker='CN',indicator='NY.GDP.MKTP.KN', \
         titletxt=titletxt+', '+unit
     
     import datetime; todaydt = datetime.date.today()
-    sourcetxt=text_lang("数据来源：世界银行","Data source: World Bank")
+    sourcetxt=text_lang("数据来源：WB/IMF/FRED","Data source: World Bank")
     footnote=sourcetxt+', '+str(todaydt)
     collabel=indicator_name
     
@@ -486,7 +534,7 @@ def economy_mindicators_wb(ticker='CN',indicator=['NY.GDP.MKTP.CN','NY.GDP.MKTP.
 
     y_label=text_lang('经济指标',"Economic Indicator")
     import datetime; todaydt = datetime.date.today()
-    footnote2=text_lang("数据来源：世界银行","Data source: World Bank")+', '+str(todaydt)
+    footnote2=text_lang("数据来源：WB/IMF/FRED","Data source: World Bank")+', '+str(todaydt)
 
     one_unit=False
     if len(set(unit_list)) == 1: one_unit=True
@@ -696,7 +744,7 @@ def economy_mtickers_wb(ticker=['CN','US','JP'],indicator='NY.GDP.MKTP.PP.CD', \
     y_label=indicator_name
     
     import datetime; todaydt = datetime.date.today()
-    footnote2=text_lang("数据来源：世界银行","Data source: World Bank")+', '+str(todaydt)
+    footnote2=text_lang("数据来源：WB/IMF/FRED","Data source: World Bank")+', '+str(todaydt)
 
     one_unit=False
     if len(set(unit_list)) == 1: one_unit=True
@@ -789,7 +837,7 @@ def economy_trend2(ticker='CN',indicator='NY.GDP.MKTP.KN', \
     """
     功能：分析宏观经济指标，支持单国家单指标、多国家单指标、单国家多指标
     主要公共参数：
-    ticker：国家编码，两位,默认'CN'
+    ticker：国家编码，两位或三位ISO编码，默认'CN'
     indicator：宏观经济指标，默认GDP (constant LCU)，即本币不变价格GDP
     start：开始日期，默认近十年
     end：结束日期，默认当前日期
@@ -831,8 +879,22 @@ def economy_trend2(ticker='CN',indicator='NY.GDP.MKTP.KN', \
     """
     # 判断ticker个数
     ticker_num=0
-    if isinstance(ticker,str): ticker_num=1
+    if isinstance(ticker,str): 
+        ticker_num=1
+        ticker=ticker.upper()
+        if not check_country_code(ticker=ticker,show_name=False):
+            print(f"  #Warning(economy_trend2): country code {ticker} not found")
+            return None
+        
     if isinstance(ticker,list):
+        ticker=[x.upper() for x in ticker]
+        for t in ticker:
+            if not check_country_code(ticker=t,show_name=False):
+                ticker.remove(t)
+                print(f"  #Warning(economy_trend2): country code {t} not found")
+        if len(ticker)==0:
+            return None
+        
         if len(ticker)==1: 
             ticker_num=1
             ticker=ticker[0]

siat/sector_china.py

@@ -889,9 +889,17 @@ def display_industry_sw(sw_level='1',numberPerLine=4,colalign='left'):
     """
     按照类别打印申万行业列表，名称(代码)，每行5个, 套壳函数
     """
-    itype_list=['1','2','3','F','S','B','C']
-    sw_level_list=['1','2','3','F','S','B','C']
-    pos=sw_level_list.index(sw_level)
+    #itype_list=['1','2','3','F','S','B','C']
+    itype_list=['1','2','3','F','S','B']
+    #sw_level_list=['1','2','3','F','S','B','C']
+    sw_level_list=['1','2','3','F','S','B']
+    
+    try:
+        pos=sw_level_list.index(sw_level)
+    except:
+        print(f"  #Warning(display_industry_sw): no such level in Shenwan system {sw_level}")
+        print(f"  Supported Shenwan system: {sw_level_list}")
+        
     itype=itype_list[pos]
 
     print_industry_sw(itype=itype,numberPerLine=numberPerLine,colalign=colalign) 
@@ -2884,6 +2892,7 @@ if __name__=='__main__':
     
 def find_peers_china(industry='',top=20,rank=20,sw_level='2'):
     """
+    ===========================================================================
     功能：找出一个申万行业的上市公司排名
     主要参数：
     industry：申万行业名称。当industry = ''，显示的内容由sw_level控制。申万二级行业分类
@@ -3489,10 +3498,13 @@ if __name__=='__main__':
     
 def find_industry_sw(ticker,level='1',ticker_order=True,max_sleep=30):
     """
+    ===========================================================================
     功能：寻找一只或一组股票所属的申万行业，支持股票代码和股票名称。
-    level='1'：默认只查找申万1级行业,以便节省时间
-    ticker_order=True：默认输出结果按照ticker中的顺序,而非按照所属行业排序
-    max_sleep：为防止反爬虫，默认每次爬虫后睡眠最多几秒钟
+    level='1'：默认只查找申万1级行业；查找2/3级行业时间较久，可能触发反爬虫机制。
+    ticker_order=True：默认输出结果按照ticker中的顺序,而非按照所属行业排序。
+    max_sleep：为防止触发反爬虫机制，默认每次爬虫后睡眠最多30秒钟。
+    
+    返回值：查找结果df。
     """
     print("  Searching shenwan industries for securities ... ...")
     
@@ -3621,21 +3633,24 @@ if __name__=='__main__':
     
     peers=stock_industry_peer_em(ticker,indicator="市盈率",rank=10)
 
-def stock_peers_em(ticker='',indicator='',rank=10, \
+def stock_peers_em(ticker='',indicator='市盈率',rank=10, \
                    force_show_stock=True, \
                    font_size="16px",facecolor="papayawhip", \
                    numberPerLine=5):
     """
-    功能：基于东方财富行业分类，查找股票所属的行业板块以及主要同行排名
-    特点：行业分类较粗糙，略胜于无
+    ===========================================================================
+    功能：基于东方财富行业分类，查找股票所属的行业板块以及主要同行排名。
+    特点：行业分类较粗糙，略胜于无。
     主要参数：
-    ticker：股票代码，默认''显示所有板块名称
-    indicator：排名指标，默认''。例如："股价"、"流动性"、"市净率"、"市盈率"
-    rank：排名数量，默认10前十名
-    force_show_stock：是否显示股票信息，默认False
-    font_size：表格字体大小，默认"16px"
-    facecolor：输出表格的背景颜色，默认"papayawhip",
-    numberPerLine：输出表格中的每行显示个数，默认5
+    ticker：股票代码，默认''显示所有板块名称。
+    indicator：排名指标，默认''。例如："股价"、"流动性"、"市净率"、"市盈率"。
+    rank：排名数量，默认10前十名。
+    force_show_stock：是否显示股票信息，默认False。
+    font_size：表格字体大小，默认"16px"。
+    facecolor：输出表格的背景颜色，默认"papayawhip"。
+    numberPerLine：输出表格中的每行显示个数，默认5。
+    
+    注意：若结果异常，可尝试升级插件akshare。
     
     示例：
     industries=stock_peers_em()     # 显示东方财富所有行业板块分类
@@ -3663,6 +3678,7 @@ def stock_peers_em(ticker='',indicator='',rank=10, \
     except:
         if not ticker=='':
             print("  #Warning(stock_peer_em): stock info not found for",ticker)
+            print("  Solution: if stock code is correct, upgrade akshare and try again")
             
         df_em=ak.stock_board_industry_name_em()
         #df_em.sort_values(by="板块名称",ascending=True,inplace=True)
@@ -3725,7 +3741,8 @@ def stock_peers_em(ticker='',indicator='',rank=10, \
         df_disp=cfg[collist].tail(abs(rank))
     
     #强制显示所选股票
-    if force_show_stock and rank != 10:
+    #if force_show_stock and rank != 10:
+    if force_show_stock:
         #所选股票是否在其中？
         if not ticker[:6] in list(df_disp["代码"]):
             ticker_seq=cfg[cfg["代码"]==ticker[:6]]["序号"].values[0]
@@ -3780,8 +3797,22 @@ def concept_stocks_em(concept='',ticker='',indicator="市盈率",rank=10, \
                       force_show_stock=False, \
                       font_size="16px",facecolor="papayawhip",numberPerLine=5):
     """
-    功能：基于东方财富概念板块，查找关键字相关概念以及股票业绩
-    特点：概念板块划分细致，同一股票可能分属多个板块，与行业分类不同
+    ===========================================================================
+    功能：基于东方财富概念板块，查找关键字相关概念以及股票业绩。
+    特点：概念板块划分细致，同一股票可能分属多个板块，与行业分类不同。
+    参数：
+    concept：概念板块名称，或名称中的关键字，默认''输出所有概念板块名称；
+        若查找到多个名称，则优先输出查找到的名称；
+        若仅仅找到一个板块，则按indicator输出该板块的股票排行。
+    ticker：股票代码，默认''
+    indicator：指标名称，默认"市盈率"，还支持：股价，市净率，涨跌幅，流动性，换手率。
+    rank：排名，支持正负数，默认10。
+    force_show_stock：若ticker不为空，且不在rank范围内，是否强制显示该股票，默认False。
+    font_size：显示字体大小，默认"16px"。
+    facecolor：表格背景颜色，默认"papayawhip"。
+    numberPerLine：当显示板块名称时，每行显示个数，默认5。
+    
+    返回值：df
     """
     if indicator in ["股价","股票价格","价格","收盘价","价位"]:
         indicator="最新价"

{siat-3.8.1.dist-info → siat-3.8.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: siat
-Version: 3.8.1
+Version: 3.8.2
 Summary: Securities Investment Analysis Tools (siat)
 Home-page: https://pypi.org/project/siat/
 Author: Prof. WANG Dehong, International Business School, Beijing Foreign Studies University

{siat-3.8.1.dist-info → siat-3.8.2.dist-info}/RECORD

@@ -30,7 +30,7 @@ siat/cryptocurrency_test.py,sha256=3AikTNJ7j-HwLGLIYEfyXZ3bLVuLeru9mwiwHQi2SdA,2
 siat/derivative.py,sha256=qV8n09799eqLc26ojR6vN5n_X-xd7rGwdYjgq-wBih8,41483
 siat/economy-20230125.py,sha256=vxZZlPnLkh7SpGMVEPLwxjt0yYLSVmdZrO-s2NYLyoM,73848
 siat/economy.py,sha256=BFVQDxOTbuizyumpCgpZIauH6sqnwUXebpqRMmQCzys,84198
-siat/economy2.py,sha256=hZJUvavyX6RQxmmrsjMMKCSlnwdHLwNhLSUkBTLitko,37491
+siat/economy2.py,sha256=q9Sx_6l7jP0qkruIwdzF24O7ngAsP5EKDg928pkFGl4,39791
 siat/economy_test.py,sha256=6vjNlPz7W125pJb7simCddobSEp3jmLIMvVkLRZ7zW8,13339
 siat/esg.py,sha256=GMhaonIKtvOK83rhpQUH5aJt2OL3HQBSVfD__Yw-0oo,19040
 siat/esg_test.py,sha256=Z9m6GUt8O7oHZSEG9aDYpGdvvrv2AiRJdHTiU6jqmZ0,2944
@@ -98,7 +98,7 @@ siat/risk_evaluation.py,sha256=I6B3gty-t--AkDCO0tKF-291YfpnF-IkXcFjqNKCt9I,76286
 siat/risk_evaluation_test.py,sha256=YEXM96gKzTfwN4U61AS4Rr1tV7KgUvn4rRC6f3iMw9s,3731
 siat/risk_free_rate.py,sha256=IBuRqA2kppdZsW4D4fapW7vnM5HMEXOn95A5r9Pkwlo,12384
 siat/risk_free_rate_test.py,sha256=CpmhUf8aEAEZeNu4gvWP2Mz2dLoIgBX5bI41vfUBEr8,4285
-siat/sector_china.py,sha256=cWa5EiSDX7E7w_XCDNOBgxtim474ayYMDfPkVex8fFA,155257
+siat/sector_china.py,sha256=jTzHeeMjQNN-VyKZy1Sk6FwMSERKpScbgySHi5PY0QY,157031
 siat/sector_china_test.py,sha256=1wq7ef8Bb_L8F0h0W6FvyBrIcBTEbrTV7hljtpj49U4,5843
 siat/security_price.py,sha256=2oHskgiw41KMGfqtnA0i2YjNNV6cYgtlUK0j3YeuXWs,29185
 siat/security_price2.py,sha256=dYwvz9H-uWp-Gyc1g_MId9k8cITS6ZHmjW-Fc2ypp-0,26587
@@ -144,8 +144,8 @@ siat/valuation_china.py,sha256=CVp1IwIsF3Om0J29RGkyxZLt4n9Ug-ua_RKhLwL9fUQ,69624
 siat/valuation_market_china_test.py,sha256=gbJ0ioauuo4koTPH6WKUkqcXiQPafnbhU5eKJ6lpdLA,1571
 siat/var_model_validation.py,sha256=R0caWnuZarrRg9939hxh3vJIIpIyPfvelYmzFNZtPbo,14910
 siat/yf_name.py,sha256=laNKMTZ9hdenGX3IZ7G0a2RLBKEWtUQJFY9CWuk_fp8,24058
-siat-3.8.1.dist-info/LICENSE,sha256=NTEMMROY9_4U1szoKC3N2BLHcDd_o5uTgqdVH8tbApw,1071
-siat-3.8.1.dist-info/METADATA,sha256=ofydAqYklStHxKyPE6-Eo-P5QjT4IAvsHBt5E9Ogupg,8320
-siat-3.8.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-siat-3.8.1.dist-info/top_level.txt,sha256=r1cVyL7AIKqeAmEJjNR8FMT20OmEzufDstC2gv3NvEY,5
-siat-3.8.1.dist-info/RECORD,,
+siat-3.8.2.dist-info/LICENSE,sha256=NTEMMROY9_4U1szoKC3N2BLHcDd_o5uTgqdVH8tbApw,1071
+siat-3.8.2.dist-info/METADATA,sha256=GIZ_8Xueq4_MIj358C_7qVYbopqyujXD5ru_AFO63ik,8320
+siat-3.8.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+siat-3.8.2.dist-info/top_level.txt,sha256=r1cVyL7AIKqeAmEJjNR8FMT20OmEzufDstC2gv3NvEY,5
+siat-3.8.2.dist-info/RECORD,,