main.pyw

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from threading import Thread
from time import sleep

from PIL import Image, ImageDraw
from pystray import Icon, Menu, MenuItem

import modules.bing
import modules.worldmap
import modules.earthmap
import modules.netbian
import modules.toopic


def create_icon(width, height, colors):
    image = Image.new("RGB", (width, height), "white")
    dc = ImageDraw.Draw(image)
    dc.rectangle((1, 1, width // 2 - 1, height // 2 - 1), fill=colors[0])
    dc.rectangle((width // 2 + 1, 1, width - 1, height // 2 - 1), fill=colors[1])
    dc.rectangle((1, height // 2 + 1, width // 2 - 1, height - 1), fill=colors[3])
    dc.rectangle((width // 2 + 1, height // 2 + 1, width - 1, height - 1), fill=colors[2])
    return image


def on_exit():
    global exited_flag
    exited_flag = True
    tray.stop()


colors = ["orange", "green", "yellow", "blue"]
icon = create_icon(64, 64, colors)
submenu = Menu(
    MenuItem("地球气象", lambda: modules.earthmap.Wallpaper().crawl().zoom().setup()),
    MenuItem("全球气象", lambda: modules.worldmap.Wallpaper().crawl().zoom().setup()),
    MenuItem("必应风景", lambda: modules.bing.Wallpaper().crawl().setup()),
    MenuItem("彼岸网", lambda: modules.netbian.Wallpaper().crawl().setup()),
    MenuItem("壁纸社", lambda: modules.toopic.Wallpaper().crawl().setup()),
)
menu = Menu(
    MenuItem("换壁纸", submenu),
    Menu.SEPARATOR,
    MenuItem("退出", on_exit),
)

tray = Icon("Wallpaper", icon=icon, menu=menu)
Thread(target=lambda: tray.run()).start()
exited_flag = False

while True:
    if exited_flag:
        break
    colors = [colors[-1]] + colors[:-1]
    icon = create_icon(64, 64, colors)
    tray.icon = icon
    sleep(1)