LYVP/utils.py at main · Nyrok/LYVP · 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
from pytube import YouTube, extract
from PIL import Image
from moviepy.editor import VideoFileClip
from os import mkdir, path
from time import time, sleep
from rgbmatrix import RGBMatrix


async def delayed_function(t, func, asc=False, *args):
    expected = time() + t
    while True:
        if time() < expected:
            continue
        func(*args) if not asc else await func(*args)
        break


def create_folder(folder_path):
    not path.exists(folder_path) and mkdir(folder_path)


def download(link, width, height):
    video_id = extract.video_id(link)
    video_obj = YouTube(link, on_progress_callback=__on_progress)
    video_obj.register_on_progress_callback(__on_progress)
    create_folder(f"./cache/youtube/{video_id}")
    filename = f"{video_id}.mp4"
    try:
        path = video_obj.streams.filter(res="144p").first().download(f'cache/youtube/{video_id}/video', filename)
    except Exception as e:
        print("Je n'ai pas pu télécharger la vidéo YouTube.\nErreur: " + str(e))
        return None, str(e)
    print(f"Le téléchargement a été fait avec succès vers {path}")
    full_video = VideoFileClip(f"./cache/youtube/{video_id}/video/{video_id}.mp4")
    full_video = full_video.resize((width, height))
    create_folder(f"./cache/youtube/{video_id}/gif")
    new_path = f"./cache/youtube/{video_id}/gif/{video_id}.gif"
    full_video.write_gif(new_path, fps=30, program='ffmpeg')
    return new_path

def start(options, p, isImage=False, loop=True):
    matrix = RGBMatrix(options=options)
    if not isImage:
        print(f"Lancement du gif à: {p}")
        gif = Image.open(path.realpath(p))
        num_frames = gif.n_frames
        for frame_index in range(0, num_frames):
            gif.seek(frame_index)
            duration = gif.info.get("duration")
            framerate = 0.001 + (duration / 1000 / 2)
            frame = gif.copy()
            frame.thumbnail((matrix.width, matrix.height), Image.ANTIALIAS)
            canvas = matrix.CreateFrameCanvas()
            canvas.SetImage(frame.convert("RGB"))
            matrix.SwapOnVSync(canvas, framerate_fraction=framerate)
            sleep(framerate)
        if loop:
            start(options, p, isImage, loop)
    else:
        print(f"Lancement de l'image à: {p}")
        image = Image.open(path.realpath(p))
        image.thumbnail((matrix.width, matrix.height), Image.ANTIALIAS)
        matrix.SetImage(image.convert('RGB'))
        while loop:
            sleep(100)


def convert(path, width, height):
    video = VideoFileClip(path)
    video = video.resize((width, height))
    new_path = path[:-3] + 'gif'
    video.write_gif(new_path, fps=30, program='ffmpeg')
    return new_path

def __on_progress(vid, chunk, bytes_remaining):
    total_size = vid.filesize
    bytes_downloaded = total_size - bytes_remaining
    percentage_of_completion = bytes_downloaded / total_size * 100
    totalsz = (total_size / 1024) / 1024
    totalsz = round(totalsz, 1)
    remain = (bytes_remaining / 1024) / 1024
    remain = round(remain, 1)
    dwnd = (bytes_downloaded / 1024) / 1024
    dwnd = round(dwnd, 1)
    percentage_of_completion = round(percentage_of_completion, 2)
    print(
        f'Download Progress: {percentage_of_completion}%, Total Size:{totalsz} MB, Downloaded: {dwnd} MB, Remaining:{remain} MB')


def __percent(t, total):
    return (float(t) / float(total)) * float(100)