131 lines
4.7 KiB
Python
131 lines
4.7 KiB
Python
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()
|