import wx
import wx.html2
import os
import time
import logging
from pyecharts import options as opts
from pyecharts.charts import Map
from pyecharts.globals import ThemeType
import random
 
# 設(shè)置日志記錄以調(diào)試問題
logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
class ChinaMapFrame(wx.Frame):
    def __init__(self, parent=None, title="中國(guó)地圖展示"):
        # 設(shè)置框架為全屏大小
        screen_width, screen_height = wx.GetDisplaySize()
        super().__init__(parent, title=title, size=(screen_width, screen_height))
 
        # 創(chuàng)建一個(gè)垂直的sizer來管理布局
        sizer = wx.BoxSizer(wx.VERTICAL)
 
        # 創(chuàng)建WebView并將其添加到sizer，使其完全展開
        self.browser = wx.html2.WebView.New(self)
        sizer.Add(self.browser, proportion=1, flag=wx.EXPAND | wx.ALL, border=0)
 
        # 設(shè)置sizer和布局
        self.SetSizer(sizer)
        self.Layout()
        self.Centre()
        self.Maximize(True)  # 最大化以確保全屏
        self.Show()
 
        # 創(chuàng)建并加載中國(guó)地圖
        self.create_china_map()
 
    def create_china_map(self):
        """創(chuàng)建上海地圖并在WebView中顯示"""
        logging.debug("開始創(chuàng)建上海地圖")
 
        # 上海示例數(shù)據(jù)（替換為您自己的實(shí)際數(shù)據(jù)）
        shanghai_districts = [
            "黃浦區(qū)", "徐匯區(qū)", "長(zhǎng)寧區(qū)", "靜安區(qū)", "普陀區(qū)", "虹口區(qū)", "楊浦區(qū)", "閔行區(qū)",
            "寶山區(qū)", "嘉定區(qū)", "浦東新區(qū)", "金山區(qū)", "松江區(qū)", "青浦區(qū)", "奉賢區(qū)", "崇明區(qū)"
        ]
        shanghai_values = [random.randint(30, 120) for _ in range(len(shanghai_districts))]  # 生成隨機(jī)數(shù)據(jù)
        shanghai_data = list(zip(shanghai_districts, shanghai_values))
 
        map_chart = (
            Map(init_opts=opts.InitOpts(
                width="100vw",
                height="100vh",
                theme=ThemeType.LIGHT,
                page_title="上海地圖展示"
            ))
            .add("上海各區(qū)", shanghai_data, "上海", is_map_symbol_show=False, label_opts=opts.LabelOpts(is_show=True))  # 顯示標(biāo)簽
            .set_global_opts(
                title_opts=opts.TitleOpts(title="上海市行政區(qū)域圖"),
                visualmap_opts=opts.VisualMapOpts(
                    min_=0,
                    max_=120,
                    is_piecewise=True,
                    range_color=["#50a3ba", "#eac763", "#d94e5d"],  # 使用 range_color
                    orient="vertical",
                    pos_left="5%",
                    pos_bottom="5%"
                ),
                toolbox_opts=opts.ToolboxOpts(),
                tooltip_opts=opts.TooltipOpts(trigger="item", formatter=": {c}"),
            )
        )
 
        # 生成HTML文件
        html_file = "shanghai_map.html"
        try:
            logging.debug(f"正在渲染上海地圖到 {html_file}")
            map_chart.render(path=html_file)  # 不要在此處傳遞模板或環(huán)境。Pyecharts將生成完整的HTML。
            logging.debug(f"上海地圖渲染成功到 {html_file}")
        except Exception as e:
            logging.error(f"渲染上海地圖時(shí)出錯(cuò)：{e}", exc_info=True)
            return
 
        # 在WebView中加載HTML文件
        file_path = os.path.join(os.getcwd(), html_file)
        normalized_path = file_path.replace('\\', '/')
        file_url = f"file:///{normalized_path}"
        logging.debug(f"正在加載上海URL：{file_url}")
        self.browser.LoadURL(file_url)
 
 
def main():
    app = wx.App(False)
    frame = ChinaMapFrame()
    app.MainLoop()
 
 
if __name__ == "__main__":
    main()