当前位置:首页> 正文

Python+Tkinter制作股票数据抓取小程序

目录

程序布局

抓取与保存功能

添加功能

个股查询按钮

批量查询开关

在前面的文章中,我们一起学习了如何通过 Python 抓取东方财富网的实时股票数据,链接如下

用 Python 爬取股票实时数据

今天我们就在这个基础上,实现一个 Tkinter GUI 程序,完成无代码股票抓取!

首先对于 Tkinter 相信大家都是比较了解的,如果有小伙伴对于 Tkinter 的相关用法不是特别熟悉的话,可以看如下文章

Tkinter 入门之旅

首先我们先看一下 GUI 程序的最终效果

该程序共分三个区域:

个股查询:用于查询某只股票的数据,可以查询1天或者多天数据

批量查询:查询某个交易所所有股票的数据

日志区域:打印抓取信息

下面我们就来看看该如何从头完成这个 GUI 程序

程序布局

首先就是程序布局,这里我们使用了 ttkbootstrap 库来美化整体程序

程序初始化部分

import ttkbootstrap as ttk from ttkbootstrap.constants import * import tkinter.messagebox as messbox class MainCreator(ttk.Window):     def __init__(self):         super().__init__("股票抓取工具", themename="solar", resizable=(False, False))         self.configure_frame = ttk.Frame(self, padding=(10, 10, 5, 10))         self.configure_frame.pack(side=LEFT, fill=BOTH, expand=YES)         self.demo_frame = ttk.Frame(self, padding=(5, 10, 10, 10))         self.demo_frame.pack(side=LEFT, fill=BOTH, expand=YES)

程序布局部分

    def create_frame(self):         """Create all the frame widgets"""         container = ttk.Frame(self)         container.pack(side=LEFT, fill=BOTH, expand=YES, padx=5)         color_group = ttk.Labelframe(             master=container, text="个股查询", padding=10         )         color_group.pack(fill=X, side=TOP)         en_command = super().register(self.en_validate)         self.en0 = ttk.Entry(color_group, width=5, text='', bootstyle='warning',                              validate='key', validatecommand=(en_command, '%P'))         self.en0.insert('0', 1)         self.en0.config(state=DISABLED)         ...

总体上来说,我们所有的组件都是从 ttk 当中实例化的,也就是直接复用了库 ttkbootstrap 的相关美化功能,使得我们的程序看起来更加高级美观

抓取与保存功能

下面我们编写股票抓取代码和对应的保存代码

股票抓取

def get_A_mins(code):     if code.startswith("3") or code.startswith("0"):         url = shang_A_url.replace("%s", code)     elif code.startswith("6"):         url = shen_A_url.replace("%s", code)     else:         return False     res = requests.get(url)     result = res.text.split("cb_1659146437934_51841953")[1].split("(")[1].split(");")[0]     result_json = json.loads(result)     stock_data = result_json['data']     return stock_data def get_A_days(code):     if code.startswith("3") or code.startswith("0"):         url = shang_A_days.replace("%s", code)     elif code.startswith("6"):         url = shen_A_days.replace("%s", code)     else:         return False     res = requests.get(url)     result = res.text.split("cb_1659171393020_15037673")[1].split("(")[1].split(");")[0]     result_json = json.loads(result)     stock_data = result_json['data']     return stock_data def get_hsj(date):     total_data = []     try:         for i in range(1, 5):             res = requests.get(hsj_url.replace("%s", str(i)))             result = res.text.split("jQuery112402508937289440778_1658838703304")[1].split("(")[1].split(");")[0]             result_json = json.loads(result)             stock_data = result_json['data']             if stock_data:                 total_data.append(stock_data)                 saveFunc.save_data_hsj(stock_data['diff'], date)             else:                 return total_data         return total_data     except Exception as e:         return False def get_Center():     total_data = []     for i in range(1, 20):         res = requests.get(center_url.replace("%s", str(i)))         result = res.text.split("jQuery112404177389105264733_1659176039486")[1].split("(")[1].split(");")[0]         result_json = json.loads(result)         center_data = result_json['data']         if center_data:             total_data.append(center_data)         else:             return total_data def get_shang_A():     pass def get_shen_A():     pass def get_bei_A():     pass

股票代码分为上交所,深交所和北交所以及大盘行情数据,所以我们分别编写函数进行处理

数据保存

import os def save_data_mins(data, date):     Code = data['code']     Name = data['name']     if not os.path.exists(r"stock_data_%s_%s_%s_mins.csv" % (Code, Name, date)):         with open("stock_data_%s_%s_%s_mins.csv" % (Code, Name, date), "a+", encoding='utf-8') as f:             f.write("时间,最新价,成交量(手),成交额\n")             for i in data['trends']:                 i_list = i.split(",")                 time = i_list[0]                 price = i_list[2]                 mount = i_list[5]                 count = i_list[6]                 row = '{},{},{},{}'.format(                     time,price,mount,count)                 f.write(row)                 f.write('\n')     else:         ... def save_data_days(data, days):     print(days)     Code = data['code']     Name = data['name']     if not os.path.exists(r"stock_data_%s_%s_days.csv" % (Code, Name)):         with open("stock_data_%s_%s_days.csv" % (Code, Name), "a+", encoding='utf-8') as f:             f.write("时间,开盘价,收盘价,最高价,最低价,成交量(手),成交额,振幅,涨跌幅,涨跌额,换手率\n")             for i in data["klines"][::-1][:int(days)]:                 i_list = i.split(",")                 time = i_list[0]                 Open = i_list[1]                 close = i_list[2]                 heigh = i_list[3]                 low = i_list[4]                 mount = i_list[5]                 count = i_list[6]                 amplitude = i_list[7]                 changePercent = i_list[8]                 change = i_list[9]                 turnoverRate = i_list[10]                 row = '{},{},{},{},{},{},{},{},{},{},{}'.format(                     time,Open,close,heigh,low,mount,count,amplitude,changePercent,change,turnoverRate)                 f.write(row)                 f.write('\n')     else:         ... def save_data_hsj(data, date):     if not os.path.exists(r'stock_data_%s.csv' % date):         with open("stock_data_%s.csv" % date, "a+", encoding='utf-8') as f:             f.write("股票代码,股票名称,最新价,涨跌幅,涨跌额,成交量(手),成交额,振幅,换手率,市盈率,量比,最高,最低,今开,昨收,市净率\n")             for i in data:                 Code = i["f12"]                 Name = i["f14"]                 Close = i['f2']                 ChangePercent = i["f3"]                 Change = i['f4']                 Volume = i['f5']                 Amount = i['f6']                 Amplitude = i['f7']                 TurnoverRate = i['f8']                 PERation = i['f9']                 VolumeRate = i['f10']                 Hign = i['f15']                 Low = i['f16']                 Open = i['f17']                 PreviousClose = i['f18']                 PB = i['f22']                 row = '{},{},{},{},{},{},{},{},{},{},{},{},{},{},{},{}'.format(                     Code,Name,Close,ChangePercent,Change,Volume,Amount,Amplitude,                     TurnoverRate,PERation,VolumeRate,Hign,Low,Open,PreviousClose,PB)                 f.write(row)                 f.write('\n')     else:         ... def save_data_center():     pass def save_data_shang_A():     pass 添加功能

接下来就是为布局好的 GUI 程序添加各种响应功能

个股查询按钮

为个股查询的抓取按钮绑定方法catch

self.bt = ttk.Button(color_group, text='抓取', bootstyle='success', command=self.catch)

方法catch的定义如下

    def catch(self):         self.txt.yview_moveto(1)         now = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')         day = datetime.datetime.now().strftime('%Y-%m-%d')         if self.cbvar.get() == 1:             catch_day = self.en0.get()             if self.en.get():                 code = self.en.get()                 result = catchFunc.get_A_days(code)                 if result:                     saveFunc.save_data_days(result, catch_day)                     self.txt.insert(ttk.INSERT, "抓取股票%s成功 %s" % (code, now))                     self.txt.insert(ttk.INSERT, "\n")                     self.txt.update()                     self.txt.yview_moveto(1)                 else:                     print("股票代码错误")                     messbox.showerror("股票代码错误", "请输入正确的股票代码!")             else:                 print("请输入股票代码")                 messbox.showerror("股票代码为空", "请输入股票代码!")         else:         ... 批量查询开关

对于批量查询,我们是通过一个多选框开关控制的

self.cb_batch = ttk.Checkbutton(cr_group, text="开启批量", variable=self.cbvar1, command=self.cb_button)

绑定的方法如下

    def cb_button(self):         if self.cbvar1.get() == 1:             self.bt_batch.config(state=NORMAL)         else:             self.bt_batch.config(state=DISABLED)

以上就是Python+Tkinter制作股票数据抓取小程序的详细内容,更多关于Python Tkinter股票数据抓取的资料请关注易知道(ezd.cc)其它相关文章!

展开全文阅读

相关内容