152 lines
4.7 KiB
Python
152 lines
4.7 KiB
Python
from VaporSong import VaporSong
|
|
import os
|
|
import sys
|
|
import youtube_dl
|
|
import logzero
|
|
from logzero import logger
|
|
from logzero import setup_logger
|
|
import re
|
|
import urllib.request
|
|
import urllib.parse
|
|
import argparse
|
|
import time
|
|
|
|
version = 1.0
|
|
style = False
|
|
|
|
text = 'This program takes YouTube URL or title of a song and converts it into vaporwave'
|
|
|
|
parser = argparse.ArgumentParser(description = text)
|
|
parser.add_argument("-M", "--music", help="generate v a p o r w a v e music", action="store_true")
|
|
parser.add_argument("-V", "--version", help="show program version", action="store_true")
|
|
parser.add_argument("-i", "--input", help="input url")
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.version:
|
|
print("ｖａｐｏｒｗａｖｅ ｇｅｎｅｒａｔｏｒ 旺育栄", version)
|
|
exit
|
|
if args.music:
|
|
music = True
|
|
if args.input:
|
|
query = args.url
|
|
else:
|
|
query = input("Enter target song's name or YouTube URL: ")
|
|
|
|
MAX_DURATION = 600 # In-case the program finds a compilation
|
|
youtube_urls = ('youtube.com', 'https://www.youtube.com/', 'http://www.youtube.com/', 'http://youtu.be/', 'https://youtu.be/', 'youtu.be')
|
|
|
|
def download_file(query,request_id=1):
|
|
"""Returns audio to the vapor command handler
|
|
|
|
Searches YouTube for 'query', finds first match that has
|
|
duration under the limit, download video with youtube_dl
|
|
and extract .wav audio with ffmpeg.
|
|
|
|
Query can be YouTube link.
|
|
"""
|
|
ydl_opts = {
|
|
'quiet': 'True',
|
|
'format': 'bestaudio/best',
|
|
'outtmpl': str(request_id) +'.%(ext)s',
|
|
'prefer_ffmpeg': 'True',
|
|
'noplaylist': 'True',
|
|
'postprocessors': [{
|
|
'key': 'FFmpegExtractAudio',
|
|
'preferredcodec': 'wav',
|
|
'preferredquality': '192',
|
|
}],
|
|
}
|
|
|
|
original_path = str(request_id) + ".wav"
|
|
file_title = ""
|
|
|
|
# check if query is youtube url
|
|
if not query.lower().startswith((youtube_urls)):
|
|
# search for youtube videos matching query
|
|
query_string = urllib.parse.urlencode({"search_query" : query})
|
|
html_content = urllib.request.urlopen("http://www.youtube.com/results?" + query_string)
|
|
search_results = re.findall(r'href=\"\/watch\?v=(.{11})', html_content.read().decode())
|
|
info = False
|
|
|
|
# find video that fits max duration
|
|
logger.info("Get video information...")
|
|
for url in search_results:
|
|
# check for video duration
|
|
try:
|
|
info = youtube_dl.YoutubeDL(ydl_opts).extract_info(url,download = False)
|
|
except Exception as e:
|
|
logger.error(e)
|
|
raise ValueError('Could not get information about video.')
|
|
full_title = info['title']
|
|
if (info['duration'] < MAX_DURATION and info['duration'] >= 5):
|
|
# get first video that fits the limit duration
|
|
logger.info("Got video: " + str(full_title))
|
|
file_title = info['title']
|
|
break
|
|
|
|
# if we ran out of urls, return error
|
|
if (not info):
|
|
raise ValueError('Could not find a video.')
|
|
|
|
# query was a youtube link
|
|
else:
|
|
logger.info("Query was a YouTube URL.")
|
|
url = query
|
|
info = youtube_dl.YoutubeDL(ydl_opts).extract_info(url,download = False)
|
|
file_title = info['title']
|
|
# check if video fits limit duration
|
|
if (info['duration'] < 5 or info['duration'] > MAX_DURATION):
|
|
raise ValueError('Video is too short. Need 5 seconds or more.')
|
|
|
|
# download video and extract audio
|
|
logger.info("Downloading video...")
|
|
with youtube_dl.YoutubeDL(ydl_opts) as ydl:
|
|
try:
|
|
ydl.download([url])
|
|
except Exception as e:
|
|
logger.error(e)
|
|
raise ValueError('Could not download ' + str(full_title) + '.')
|
|
|
|
return original_path, file_title
|
|
|
|
|
|
def gen_vapor(filePath, title):
|
|
# Delete stuff if there is anything left over.
|
|
os.system("rm -r download/")
|
|
os.system("rm -r beats/")
|
|
|
|
# Make the proper folders for intermediate steps
|
|
os.system("mkdir download/")
|
|
os.system("mkdir beats/")
|
|
|
|
|
|
# Download the youtube query's first result. Might be wrong but YOLO
|
|
#YTDownloader.download_wav_to_samp2(query)
|
|
|
|
# For every song in download folder(just one for now)
|
|
"""
|
|
for fs in os.listdir("download/"):
|
|
# Slow down the song.
|
|
VaporSong.vaporize_song(query,"download/"+fs)
|
|
pass
|
|
# When we are finished, delete the old folders.
|
|
"""
|
|
VaporSong.vaporize_song(filePath, title)
|
|
|
|
os.system("rm -r download/")
|
|
os.system("rm -r beats/")
|
|
|
|
|
|
"""
|
|
## Makes this a command line tool: disable when we get the webserver going
|
|
sys.argv.pop(0)
|
|
query = ""
|
|
for s in sys.argv:
|
|
query = query + s
|
|
"""
|
|
|
|
if music:
|
|
name, title = download_file(query)
|
|
gen_vapor(name, title)
|