1#!/usr/bin/env python3
2# -*- coding: utf-8 -*-
3"""
4股票数据获取示例脚本
5"""
6 
7import tushare as ts
8import pandas as pd
9import os
10 
11# 读取环境变量中的token, 或者读取本地记录的token
12token = os.getenv('TUSHARE_TOKEN') or ts.get_token()
13 
14# 初始化pro接口
15pro = ts.pro_api(token)
16 
17 
18def get_stock_list():
19    """
20    获取股票列表
21    """
22    try:
23        data = pro.stock_basic(exchange='', list_status='L', fields='ts_code,symbol,name,area,industry,list_date')
24        print("股票列表获取成功：")
25        print(data.head())
26        return data
27    except Exception as e:
28        print(f"获取股票列表失败：{e}")
29        return None
30 
31 
32def get_daily_data(ts_code, start_date, end_date):
33    """
34    获取股票日线数据
35    """
36    try:
37        data = pro.daily(ts_code=ts_code, start_date=start_date, end_date=end_date)
38        print(f"{ts_code}日线数据获取成功：")
39        print(data.head())
40        return data
41    except Exception as e:
42        print(f"获取日线数据失败：{e}")
43        return None
44 
45 
46def get_financial_data(ts_code, year, quarter):
47    """
48    获取财务指标数据
49    """
50    try:
51        data = pro.fina_indicator(ts_code=ts_code, year=year, quarter=quarter)
52        print(f"{ts_code}财务指标数据获取成功：")
53        print(data.head())
54        return data
55    except Exception as e:
56        print(f"获取财务指标数据失败：{e}")
57        return None
58 
59 
60def main():
61    """
62    主函数
63    """
64    print("===== tushare 股票数据获取示例 =====")
65    
66    # 获取股票列表
67    stock_list = get_stock_list()
68    
69    if stock_list is not None:
70        # 获取第一只股票的代码
71        ts_code = stock_list['ts_code'].iloc[0]
72        print(f"\n使用股票代码：{ts_code}")
73        
74        # 获取日线数据（最近30天）
75        import datetime
76        end_date = datetime.datetime.now().strftime('%Y%m%d')
77        start_date = (datetime.datetime.now() - datetime.timedelta(days=30)).strftime('%Y%m%d')
78        print(f"\n获取日线数据：{start_date} 至 {end_date}")
79        get_daily_data(ts_code, start_date, end_date)
80        
81        # 获取财务数据（最近一年）
82        current_year = datetime.datetime.now().year
83        print(f"\n获取财务数据：{current_year-1}年 第4季度")
84        get_financial_data(ts_code, current_year-1, 4)
85 
86 
87if __name__ == "__main__":
88    main()
89