VaporwaveGenerator/main.py

186 lines
5.4 KiB
Python

from src.VaporSong import VaporSong
from src.VHSImage import generateVHSStyle
from src.VHSVideo import VHS_Vid
import os
import youtube_dl
from logzero import logger
import re
import urllib.request
import urllib.parse
import argparse
version = 2.1
style = False
text = "| V A P O R W A V E || G E N E R A T O R |"
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("-P", "--picture", help="generate VHS Style image", action="store_true")
parser.add_argument("-V", "--video", help="VHS Style Video", action="store_true")
parser.add_argument("-v", "--version", help="show program version", action="store_true")
parser.add_argument("-i", "--input")
parser.add_argument("-o", "--output", help="Output for specifying output video")
parser.add_argument("-d","--date",help="Custom Date in yyyy/mm/dd format. e.g 2020/5/14")
args = parser.parse_args()
music = False
picture = False
video = False
date = None
if args.version:
print("vaporwave generator 旺育栄", version)
exit
if args.music:
music = True
elif args.picture:
picture = True
elif args.video:
video = True
if args.input:
query = args.input
if args.output:
outfile = args.output
if args.date:
date = args.date
else:
parser.print_help()
exit
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)
elif picture:
generateVHSStyle(query, "out.jpg",date=date)
elif video:
VHS_Vid(query, outfile)