# 参考文档　
# 　https://www.jb51.net/article/235161.htm

# pip install pyecharts
# pip install snapshot_selenium

from unittest import mock


def get_chrome_driver():
    from selenium import webdriver
    options = webdriver.ChromeOptions()
    options.add_argument("headless")
    options.add_argument('--no-sandbox')
    options.add_argument('--disable-gpu')
    options.add_argument('--disable-dev-shm-usage')
    return webdriver.Chrome(options=options)


from snapshot_selenium import snapshot as driver
from pyecharts import options as opts
from pyecharts.charts import Bar
from pyecharts.render import make_snapshot
from snapshot_selenium import snapshot  # 使用 snapshot-selenium 渲染图片
from selenium.webdriver.chrome.options import Options


c = (
    Bar()
    # 添加X轴数据
    .add_xaxis(["衬衫", "羊毛衫", "雪纺衫", "裤子", "高跟鞋", "袜子"])
    # 添加Y轴数据,系列的名称
    .add_yaxis("商家A", [5, 20, 36, 10, 75, 90])  # 此处数据，应该从PG中通过SQL进行获取，然后再转换为数组形式,封装一个通用的PgUtil.py
    .add_yaxis("商家B", [8, 15, 60, 20, 25, 30])
    # 添加标题
    .set_global_opts(title_opts=opts.TitleOpts(title="主标题: 双十一销量", subtitle="副标题:服饰类"))
)

# 导出图片
with mock.patch('snapshot_selenium.snapshot.get_chrome_driver', get_chrome_driver):
    # 需要安装 snapshot-selenium 或者 snapshot-phantomjs
    make_snapshot(driver, c.render(), "bar.png")