mirror of
https://github.com/leon-ai/leon.git
synced 2024-12-24 01:04:28 +03:00
98 lines
2.8 KiB
Python
98 lines
2.8 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding:utf-8 -*-
|
|
|
|
import requests
|
|
import os
|
|
import utils
|
|
from time import time
|
|
from pytube import YouTube
|
|
|
|
def run(string, entities):
|
|
"""Download new videos from a YouTube playlist"""
|
|
|
|
db = utils.db()['db']
|
|
query = utils.db()['query']
|
|
operations = utils.db()['operations']
|
|
apikey = utils.config('api_key')
|
|
playlistid = utils.config('playlist_id')
|
|
# https://developers.google.com/youtube/v3/docs/playlistItems/list
|
|
url = 'https://www.googleapis.com/youtube/v3/playlistItems?part=snippet&maxResults=50&playlistId=' + playlistid + '&key=' + apikey
|
|
|
|
utils.output('inter', 'reaching_playlist', utils.translate('reaching_playlist'))
|
|
# Get videos from the playlist
|
|
try:
|
|
r = utils.http('GET', url)
|
|
|
|
# In case there is a problem like wrong settings
|
|
if 'error' in r.json():
|
|
error = r.json()['error']['errors'][0]
|
|
return utils.output('settings_error', 'settings_error', utils.translate('settings_errors', {
|
|
'reason': error['reason'],
|
|
'message': error['message']
|
|
}))
|
|
|
|
|
|
items = r.json()['items']
|
|
videoids = []
|
|
videos = []
|
|
|
|
for item in items:
|
|
resource = item['snippet']['resourceId']
|
|
|
|
if resource['kind'] == 'youtube#video':
|
|
videoids.append(resource['videoId'])
|
|
videos.append({
|
|
'id': resource['videoId'],
|
|
'title': item['snippet']['title']
|
|
})
|
|
except requests.exceptions.RequestException as e:
|
|
return utils.output('request_error', 'request_error', utils.translate('request_errors'))
|
|
|
|
Entry = query()
|
|
|
|
# First initialization
|
|
if db.count(Entry.platform == 'youtube') == 0:
|
|
db.insert({
|
|
'platform': 'youtube',
|
|
'checked_at': int(time()),
|
|
'downloaded_videos': []
|
|
})
|
|
else:
|
|
db.update({ 'checked_at': int(time()) }, Entry.platform == 'youtube')
|
|
|
|
# Get videos already downloaded
|
|
downloadedvideos = db.get(Entry.platform == 'youtube')['downloaded_videos']
|
|
|
|
todownload = []
|
|
for video in videos:
|
|
if video['id'] not in downloadedvideos:
|
|
todownload.append(video)
|
|
|
|
nbrtodownload = len(todownload)
|
|
|
|
if nbrtodownload == 0:
|
|
return utils.output('nothing_to_download', 'nothing_to_download', utils.translate('nothing_to_download'))
|
|
|
|
utils.output('inter', 'nb_to_download', utils.translate('nb_to_download', {
|
|
'nb': nbrtodownload
|
|
}))
|
|
|
|
# Create the module downloads directory
|
|
moduledldir = utils.createdldir()
|
|
|
|
for i, video in enumerate(todownload):
|
|
utils.output('inter', 'downloading', utils.translate('downloading', {
|
|
'video_title': video['title']
|
|
}))
|
|
|
|
# Download the video
|
|
yt = YouTube('https://youtube.com/watch?v=' + video['id'])
|
|
yt.streams.first().download(moduledldir)
|
|
|
|
# Add the new downloaded video to the DB
|
|
downloadedvideos.append(video['id'])
|
|
db.update({ 'downloaded_videos': downloadedvideos }, Entry.platform == 'youtube')
|
|
|
|
# Will synchronize the content (because "end" type) if synchronization enabled
|
|
return utils.output('end', 'success', utils.translate('success'))
|