qqbot-ClockInAddon/imageHandler.py at master · ISUNCY/qqbot-ClockInAddon · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
from PIL import Image
from PIL import ImageDraw
from PIL import ImageFont
from io import BytesIO
import base64
from dataHandler import DataHandler

class ImageHandler:
    monthDayCount = [-1, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    IMGWIDTH = 1200
    IMGHEIGHT = 1400
    LUIMGWIDTH = 200
    LUIMGHEIGHT = 200
    TITLESIZE = 100
    NUMSIZE = 80


    def __init__(self, image_path: str):
        self.image_path = image_path
        self.image = None
        self.userName = ""
        self.month = None
        self.record = None
        self.userId = None

    @staticmethod
    def resizeOkImage(okImage):
        width = ImageHandler.LUIMGWIDTH
        height = ImageHandler.LUIMGHEIGHT
        okImage = okImage.resize((width, height))
        newImage = Image.new('RGBA', (int(width*1.5), int(height*1.5)), (255, 255, 255, 0))
        newImage.paste(okImage, (int(width/2), int(width*0.6)))
        newImage = newImage.resize((width, height))
        return newImage

    @staticmethod
    def resizeImage(image):
        width = ImageHandler.LUIMGWIDTH
        height = ImageHandler.LUIMGHEIGHT
        image = image.resize((width, height))
        image = image.convert("RGBA")
        newImage = Image.new('RGBA', (int(width*1.05), int(height*1.05)), (255, 255, 255, 0))
        newImage.paste(image, (int(width*0.025), int(width*0.025)))
        newImage = newImage.resize((width, height))
        return newImage

    # @staticmethod
    # def getLuOkImage():
    #     okImage = Image.open("img/yes.png")
    #     luImage = Image.open("img/lu.png")
    #     luImage = luImage.resize((ImageHandler.LUIMGWIDTH, ImageHandler.LUIMGHEIGHT))
    #     okImage = ImageHandler.resizeOkImage(okImage)
    #     okLuImage = Image.alpha_composite(luImage, okImage)
    #     return okLuImage


    # @staticmethod
    # def getLuImage():
    #     luImage = Image.open("img/lu.png")
    #     luImage = luImage.resize((ImageHandler.LUIMGWIDTH, ImageHandler.LUIMGHEIGHT))
    #     return luImage

    @staticmethod
    def getLuOkImage():
        okImage = Image.open("img/yes.png")
        luImage = Image.open("img/lu2.jpg")
        luImage = luImage.resize((ImageHandler.LUIMGWIDTH, ImageHandler.LUIMGHEIGHT))
        luImage = luImage.convert("RGBA")
        okImage = ImageHandler.resizeOkImage(okImage)
        luImage = Image.alpha_composite(luImage, okImage)
        luImage = ImageHandler.resizeImage(luImage)
        return luImage

    @staticmethod
    def getLuImage():
        commonImage = Image.open("img/lu1.jpg")
        commonImage = commonImage.resize((ImageHandler.LUIMGWIDTH, ImageHandler.LUIMGHEIGHT))
        commonImage = commonImage.convert("RGBA")
        commonImage = ImageHandler.resizeImage(commonImage)
        return commonImage


    def getImage(self):
        self.image = Image.new('RGBA', (self.IMGWIDTH, self.IMGHEIGHT), (255, 255, 255, 0))
        draw = ImageDraw.Draw(self.image)
        formalFont = ImageFont.truetype("font/STXINGKA.TTF", self.TITLESIZE)
        titleText = str(self.month)+"月份 " + self.userName + " 打卡记录"
        draw.text((self.TITLESIZE/2, self.TITLESIZE/2), titleText, font=formalFont, fill=(255, 255, 255, 255))
        okLuImage = ImageHandler.getLuOkImage()

        luImage = ImageHandler.getLuImage()
        for i in range(self.monthDayCount[self.month]):
            day = i+1
            count = DataHandler.getCount(self.userId, self.month, day)
            if count == 0:
                self.image.paste(luImage, ((day%6)*self.LUIMGWIDTH, (day//6)*self.LUIMGHEIGHT + self.TITLESIZE*2), luImage)
            else :
                self.image.paste(okLuImage, ((day%6)*self.LUIMGWIDTH, (day//6)*self.LUIMGHEIGHT + self.TITLESIZE*2), okLuImage)
                if count[0] != 1 :
                    formalFont = ImageFont.truetype("font/STHUPO.TTF", int(self.NUMSIZE*0.8))
                    draw.text(((day%6)*self.LUIMGWIDTH+self.TITLESIZE+self.NUMSIZE//3, (day//6)*self.LUIMGHEIGHT + self.TITLESIZE*2+self.TITLESIZE+self.NUMSIZE//2), str(count[0]), font=formalFont, fill=(255, 0, 0, 255))
            formalFont = ImageFont.truetype("font/STCAIYUN.TTF", self.NUMSIZE)
            draw.text(((day%6)*self.LUIMGWIDTH, (day//6)*self.LUIMGHEIGHT + self.TITLESIZE*2), str(day), font=formalFont, fill=(255, 255, 255, 255))

        # background = Image.new('RGBA', (self.IMGWIDTH, self.IMGHEIGHT), (249,245,215, 255))
        # background = Image.new('RGBA', (self.IMGWIDTH, self.IMGHEIGHT), (70,170,255, 255))
        # background = Image.new('RGBA', (self.IMGWIDTH, self.IMGHEIGHT), (60,140,210, 255))
        # background = Image.new('RGBA', (self.IMGWIDTH, self.IMGHEIGHT), (50,120,180, 255))
        background = Image.new('RGBA', (self.IMGWIDTH, self.IMGHEIGHT), (145,135,175, 255))
        self.image = Image.alpha_composite(background, self.image)
        buffer = BytesIO()
        self.image.save(buffer, format="PNG")
        imageData = buffer.getvalue()
        base64_image = base64.b64encode(imageData).decode('utf-8')
        data_url = f"data:image/png;base64,{base64_image}"
        return data_url