ci.py

import rumps
from AppKit import NSScreen
from PIL import Image, ImageFont, ImageDraw
from time import time, sleep
import json
import random
import os
import subprocess


class CiApp(object):
    def __init__(self):
        self.config = {
            "app_name": "Ci",
            "generate_poem": "Generate Ci",
            "interval": 600,
        }
        self.app = rumps.App(self.config["app_name"])
        self.set_up_menu()
        self.generate_poem_button = rumps.MenuItem(
            title=self.config["generate_poem"], callback=self.generate_poem)
        self.app.menu = [self.generate_poem_button]
        self.interval = self.config["interval"]
        self.timer = rumps.Timer(self.on_tick, 1)
        self.start_timer()

    def set_up_menu(self):
        self.app.title = "词"

    def on_tick(self, sender):
        time_left = sender.end - sender.count
        mins = time_left // 60 if time_left >= 0 else time_left // 60 + 1
        secs = time_left % 60 if time_left >= 0 else (-1 * time_left) % 60
        if mins == 0 and time_left < 0:
            self.generate_poem()
            self.reset_timer()
        sender.count += 1

    def start_timer(self):
        self.timer.count = 0
        self.timer.end = self.interval
        self.timer.start()

    def reset_timer(self):
        self.timer.stop()
        self.timer.count = 0
        self.start_timer()

    def generate_poem(self, sender=None):
        __location__ = os.path.realpath(os.path.join(
            os.getcwd(), os.path.dirname(__file__)))

        self.set_up_menu()

        n = random.randint(1, 320)  # generate random poem id
        print(n)

        # with open('poems.json') as f:
        # data = json.load(f)
        with open(os.path.join(os.path.dirname(__file__), 'poems.json'), 'r') as f:
            data = json.load(f)

        def find_poem(n):  # extract poem from json
            for keyval in data:
                if n == keyval['id']:
                    return keyval['contents']

        if (find_poem(n) != None):
            title_text = find_poem(n)

        def find_title(n):  # extract poem title from json
            for keyval in data:
                if n == keyval['id']:
                    return keyval['title']
        if (find_title(n) != None):
            title = find_title(n)

        # base = Image.open(os.path.join(
        #     __location__, 'base.jpeg'))  # open base image
        # base = Image.open(
        #     "/System/Library/Desktop Pictures/Solid Colors/Ocher.png")
        screenHeight = int(NSScreen.mainScreen().frame().size.height)
        screenWidth = int(NSScreen.mainScreen().frame().size.width)
        base = Image.new('RGB', (screenWidth,  screenHeight), (random.randint(0, 255)  # generate random poem id
                                                               , random.randint(0, 255), random.randint(0, 255)))
        image_editable = ImageDraw.Draw(base)  # make it editable

        ttf = ImageFont.truetype(
            'qiji-combo.ttf', 64)  # load font

        # get the size of the poem with font
        w, h = image_editable.textsize(title_text, font=ttf)

        image_editable.text(((screenWidth-w)/2, (screenHeight-h)/2), title_text,
                            (49, 27, 8, 64), font=ttf)  # center the poem itself

        # get the size of the title with font
        w2, h2 = image_editable.textsize(title, font=ttf)

        # center the title horizontally, keep it above the poem vertically
        image_editable.text(((screenWidth-w2)/2, ((screenHeight-h)/2)-h),
                            title, (49, 27, 8, 64), font=ttf)
        outputPath = os.path.expanduser("~/Pictures/Ci-Wallpaper.png")
        # base.save('output.png')  # save output
        base.save(outputPath)
        print(title_text)
        print("Current screen resolution: %dx%d" %
              (screenWidth, screenHeight))
        # change macos wallpaper
        output_location = os.path.join(
            os.path.expanduser("~/Pictures/"), 'Ci-Wallpaper.png')
        command = "osascript -e 'tell application \"System Events\" to tell every desktop to set picture to " + \
            output_location + " as POSIX file'"
        print(command)
        # refresh dock to show new wallpaper
        # TODO: instead of killing dock after setting new wallpaper, add transitionary empty wallpaper to show for a split second before setting new wallpaper
        subprocess.call(['/usr/bin/killall', 'Dock'])
        os.system(command)
        self.reset_timer()

    def run(self):
        self.app.run()


if __name__ == '__main__':
    app = CiApp()
    app.run()