47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
import isodate
|
|
import youtube_dl
|
|
from googleapiclient.discovery import build
|
|
from googleapiclient.errors import HttpError
|
|
from .import baseFile
|
|
from update.utils import seconds_to_string
|
|
|
|
DEVELOPER_KEY = "AIzaSyCU_hvZJEjLlAGAnlscquKEkE8l0lVOfn0"
|
|
YOUTUBE_API_SERVICE_NAME = "youtube"
|
|
YOUTUBE_API_VERSION = "v3"
|
|
|
|
class interface(object):
|
|
|
|
def __init__(self):
|
|
self.results = []
|
|
self.name = "youtube"
|
|
self.needs_transcode = True
|
|
|
|
def search(self, text, page=1):
|
|
type = "video"
|
|
max_results = 20
|
|
youtube = build(YOUTUBE_API_SERVICE_NAME, YOUTUBE_API_VERSION, developerKey=DEVELOPER_KEY)
|
|
search_response = youtube.search().list(q=text, part="id,snippet", maxResults=max_results, type=type).execute()
|
|
self.results = []
|
|
ids = []
|
|
for search_result in search_response.get("items", []):
|
|
if search_result["id"]["kind"] == "youtube#video":
|
|
s = baseFile.song(self)
|
|
s.title = search_result["snippet"]["title"]
|
|
ids.append(search_result["id"]["videoId"])
|
|
s.url = "https://www.youtube.com/watch?v="+search_result["id"]["videoId"]
|
|
self.results.append(s)
|
|
ssr = youtube.videos().list(id=",".join(ids), part="contentDetails", maxResults=1).execute()
|
|
for i in range(len(self.results)):
|
|
self.results[i].duration = seconds_to_string(isodate.parse_duration(ssr["items"][i]["contentDetails"]["duration"]).total_seconds())
|
|
|
|
def get_download_url(self, url):
|
|
ydl = youtube_dl.YoutubeDL({'quiet': True, 'format': 'bestaudio/best', 'outtmpl': u'%(id)s%(ext)s'})
|
|
with ydl:
|
|
result = ydl.extract_info(url, download=False)
|
|
if 'entries' in result:
|
|
video = result['entries'][0]
|
|
else:
|
|
video = result
|
|
return video["url"]
|