TV_program/main.py at master · sychsergiy/TV_program · 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
import json
import datetime
import requests
from io import BytesIO
from PIL import Image, ImageDraw, ImageFont


def parse_json():
    url = "https://api.ovva.tv/v2/ua/tvguide/1plus1"
    response = requests.get(url)
    json_data = response.text
    data = json.loads(json_data)
    return data["data"]["programs"]


def get_image(url):
    response = requests.get(url)
    img = Image.open(BytesIO(response.content))
    return img


def get_time_string(time):
    seconds = ('/Date({}000)/'.format(time)).split('(')[1][:-5]
    return datetime.datetime.fromtimestamp(float(seconds)).strftime('%Y-%m-%d %H:%M:%S')[-8:-3]


def draw_text(canvas, title, subtitle, begin_time, end_time):
    draw = ImageDraw.Draw(canvas)
    font_text1 = ImageFont.truetype('arial.ttf', 16, encoding="unic")
    font_text2 = ImageFont.truetype('arial.ttf', 14, encoding="unic")
    font_time = ImageFont.truetype('arial.ttf', 14, encoding="unic")
    draw.text((25, 30), begin_time, font=font_time, fill="black")
    draw.text((25, 50), end_time, font=font_time, fill="black")
    draw.text((220, 15), title, font=font_text1, fill="white")
    draw.text((230, 50), subtitle, font=font_text2, fill="black")


def optimize_image(image, k):
    width = image.size[0] * k
    height = image.size[1] * k
    size = (width, height)
    image.thumbnail(size)


def draw_images(canvas, image, tag_height):
    optimize_image(image, 0.15)
    canvas.paste(image, (100, 0))

    background = Image.new("RGB", (270, tag_height - 1), "#f33")
    canvas.paste(background, (215, 0))


def create_tag(time_begin, time_end, image_link, title, subtitle, tag_height):
    begin_time = get_time_string(time_begin)
    end_time = get_time_string(time_end)

    canvas = Image.new("RGB", (500, 200), "#fff")

    image = get_image(image_link)

    draw_images(canvas, image, tag_height)
    draw_text(canvas, title, subtitle, begin_time, end_time)

    return canvas


def compose_tags(width, height, tags, tag_height):
    canvas = Image.new("RGB", (width, height), "#fff")
    offest_y = 0
    for tag in tags:
        canvas.paste(tag, (0, offest_y))
        offest_y += tag_height
    canvas.save("tv_program.jpeg")


def create_tags(programs):
    tag_height = 87
    tags = []
    for program in programs:
        tag = create_tag(program["realtime_begin"], program["realtime_end"],
                         program["image"]["preview"], program["title"], program["subtitle"], tag_height)
        tags.append(tag)
    return tags


if __name__ == "__main__":
    tag_height = 87
    programs = parse_json()
    tags = create_tags(programs)
    compose_tags(485, len(programs) * tag_height, tags, tag_height)