qnloft-stock/fp_main.py

import logging
import sys

from jinja2 import Environment, FileSystemLoader
from loguru import logger

from fp.新闻资讯 import News
from fp.日线数据入库 import StockDailyMain
from fp.板块数据入库 import SectorOpt
from fp.涨跌停数据 import LimitList
from fp.自选股数据 import OptionalStock
from fp.行情指数数据 import AllIndex
from fp.资金出入概览 import MoneyFlow
from utils.comm import *


class AppMain:
	def __init__(self, trade_date=datetime.now()):
		# 配置日志输出到文件和控制台
		logger.add("log/FPAppMain.log", rotation="500 MB", level="INFO")
		logger.add(sys.stderr, level="INFO")
		self.trade_date = trade_date.strftime('%Y%m%d')
		if if_run(trade_date):
			# 创建一个文件夹
			self.file_name = f'html/{self.trade_date}'
			create_file(self.file_name)
			# 各大指数数据
			self.index = AllIndex(trade_date=trade_date)
			# 板块数据
			self.sector_opt = SectorOpt(trade_date=trade_date)
			# 涨跌停 炸板
			self.limit_list = LimitList(trade_date=trade_date)
			# 资金流入流出
			self.money_flow = MoneyFlow(trade_date=trade_date)
			# 日线数据
			self.stock_daily = StockDailyMain()
			# 新闻咨询
			self.news = News(trade_date=trade_date)
			# 我的自选
			self.optional_stock = OptionalStock(trade_date=trade_date)

	def calendar_page_data(self):
		logger.info(f"{self.trade_date} start--> 开始 JSON 文件生成！")
		res_df = self.index.calendar_page_json()
		# 将 DataFrame 转换为 JSON，并写入文件
		res_df.to_json('html/json/events.json', orient='records')
		logger.info(f"{self.trade_date} --> JSON 文件已更新")

	def init_data(self):
		"""
		每日数据初始化
		:return:
		"""
		logger.info(f"{self.trade_date} start--> 开始数据初始化！")
		# 日线数据入库
		self.stock_daily.task_data(self.trade_date)
		# 行业板块 入库
		self.sector_opt.save_hy_sector()
		# 概念板块 入库
		self.sector_opt.save_gn_sector()
		# 行业板块个股
		self.sector_opt.stock_by_hy_sector()
		# 概念板块个股
		self.sector_opt.stock_by_gn_sector()
		logger.info(f"{self.trade_date} end--> 数据初始化完毕！")

	def create_chart(self):
		logger.info(f"{self.trade_date} start--> 开始图表页面生成！")
		# 板块图表
		self.sector_opt.industry_generate_chart()
		self.sector_opt.concept_generate_chart()
		# 涨跌停 炸板 图表
		self.limit_list.limit_list_chart()
		# 资金出入图表
		self.money_flow.money_flow_chart()
		logger.info(f"{self.trade_date} end--> 图表页面生成完毕！")

	def create_page(self):
		logger.info(f"{self.trade_date} start--> 开始数据静态页面生成！")
		badge, card_deck = self.index.html_page_data()
		limit_list_table_content = self.limit_list.html_page_data()
		# 新闻数据
		cctv_content, cls_content = self.news.html_page_data()
		# 首页数据 页面
		self.__write_html(template_name="index_template.html", write_html_name="index.html",
						  badge=badge, news_cctv=cctv_content, news_cls=cls_content)
		# 指数行情 页面
		self.__write_html(template_name="zs_data_template.html", write_html_name="zs_data.html", card_deck=card_deck)
		# 涨跌停 炸板 页面
		self.__write_html(template_name="limit_list_template.html", write_html_name="limit_list.html",
						  limit_table=limit_list_table_content)
		# 我的自选 页面
		# optional_stock_pills_tab_content = self.optional_stock.html_page_data()
		# self.__write_html(template_name="zx_data_template.html", write_html_name="zx_data.html",
		# 				  pills_tabContent=optional_stock_pills_tab_content)
		logger.info(f"{self.trade_date} --> 数据静态页面生成完毕！")

	def __write_html(self, template_name, write_html_name, *args, **kwargs):
		print(template_name, "---", write_html_name)
		print(dict(*args, **kwargs))
		# 设置 Jinja2 环境
		env = Environment(loader=FileSystemLoader('template/'))
		template = env.get_template(template_name)
		# 将 DataFrame 数据传递给模板并渲染
		rendered_output = template.render(dict(*args, **kwargs))
		# 将渲染后的内容写入 HTML 文件
		with open(self.file_name + f'/{write_html_name}', 'w', encoding='utf-8') as f:
			f.write(rendered_output)
		print("HTML rendered and saved as 'output.html'")

	def upload_ftp(self):
		logger.info(f"{self.trade_date} start--> 开始FTP上传！")
		# ftp上传
		# 上传json文件
		upload_files_to_ftp(
			'html/json', '/htdocs/stock/json', "events.json")
		# 上传html文件
		upload_files_to_ftp(
			f'html/{self.trade_date}', f'/htdocs/stock/{self.trade_date}')
		logger.info(f"{self.trade_date} end--> FTP上传完毕！")


if __name__ == '__main__':
	# date_obj = datetime.strptime('20231110', "%Y%m%d")
	app = AppMain()
	app.calendar_page_data()
	app.init_data()
	app.create_chart()
	app.create_page()
	app.upload_ftp()