main.py

import tkinter as tk
import akshare as ak
import pandas as pd
import matplotlib
import matplotlib.pyplot as plt

from tkinter import ttk
from matplotlib.figure import Figure
from matplotlib.backends.backend_tkagg import (FigureCanvasTkAgg, NavigationToolbar2Tk)

def plot_eva(stock_symbol): 
    indicator_dfs = []
    for ind in ['总市值', '市盈率(TTM)', '市净率']:
        df = ak.stock_zh_valuation_baidu(symbol=stock_symbol, indicator=ind, period='全部')
        df = df.set_index('date').rename(columns={'value':ind})
        indicator_dfs.append(df)
    df = pd.concat(indicator_dfs, axis=1)
    print(df.sample(5))

    plt.rcParams['font.family']=['STFangsong']
    figure = Figure(figsize=(8, 4), dpi=100)
    ax1 = figure.add_subplot()
    ax1.plot(df.index, df['总市值'], label='总市值')
    ax1.legend(loc=3)

    ax2 = ax1.twinx()
    ax2.plot(df.index, df['市盈率(TTM)'], 'b-', label='市盈率(TTM)')
    ax2.plot(df.index, df['市净率'], 'g-', label='市净率')
    ax2.legend(loc=1)

    canvas = FigureCanvasTkAgg(figure, window)
    canvas.get_tk_widget().grid(row=0, column=1, rowspan=2)

    toolbar = NavigationToolbar2Tk(canvas, window, pack_toolbar=False)
    toolbar.update()
    toolbar.grid(row=2, column=1)

def display_info(symbol):
    stock_individual_info_em_df = ak.stock_individual_info_em(symbol=symbol)
    print(stock_individual_info_em_df)
    label_info['text'] = stock_individual_info_em_df

    for i in range(len(stock_individual_info_em_df)):
        for j in range(2):
            e = tk.Entry(master=label_info, relief=tk.GROOVE)
            e.grid(row=i, column=j, sticky="ew")
            e.insert(tk.END, f"{stock_individual_info_em_df.iloc[i, j]}")

def query():
    stock = stock_entry.get()
    if stock.isnumeric():        
        display_info(stock)
        plot_eva(stock)
    else:
        df = ak.stock_zh_a_spot_em()
        match = df[df['名称'].str.contains(stock)].index.values
        if match.size > 0:
            symbol = df['代码'].iloc[match[0]]
            display_info(symbol)
            plot_eva(symbol)
        else:
            print(f"Can't find {stock}")

window = tk.Tk()
window.title("Stock Financial Sheet Analyzer")
window.rowconfigure([0], minsize=100, weight=1)
window.rowconfigure([1], minsize=300, weight=1)
window.columnconfigure(1, minsize=800, weight=1)

# 左侧列：按钮区
option_frame = tk.Frame(master=window)
option_frame.grid(row=0, column=0, sticky="n")

stock_entry = tk.Entry(master=option_frame, width=15)
stock_entry.grid(row=0, column=0, sticky="e")
info_button = tk.Button(master=option_frame, text="估值信息", width=15, command=query)
info_button.grid(row=0, column=2, sticky="w")
zcfzb_button = tk.Button(master=option_frame, text="资产负债表分析", width=15)
zcfzb_button.grid(row=1, column=0, sticky="w")
lrb_button = tk.Button(master=option_frame, text="利润表分析", width=15)
lrb_button.grid(row=1, column=1, sticky="w")
xjllb_button = tk.Button(master=option_frame, text="现金流量表分析", width=15)
xjllb_button.grid(row=1, column=2, sticky="w")

# 左侧列：信息区
info_frame = tk.Frame(master=window)
info_frame.grid(row=1, column=0, sticky="n")

label = tk.Label(master=info_frame, text="股票信息")
label.grid(row=0, column=0, sticky="n")
label_info = tk.Label(master=info_frame, text="")
label_info.grid(row=1, column=0)

# 分割线
separator = ttk.Separator(master=window, orient='vertical')
separator.grid(row=0, column=1, rowspan=2, sticky="nswe")

# 右侧列：估值区

window.mainloop()