From 1a0b15057b88bba4af0d0743ed17c24e738275f5 Mon Sep 17 00:00:00 2001 From: msglm Date: Wed, 11 Jan 2023 20:51:07 -0600 Subject: fixed bug where JS mode overrode itself. Changed output format to be less insane --- comedyGenerator | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/comedyGenerator b/comedyGenerator index 0b21710..01eb216 100755 --- a/comedyGenerator +++ b/comedyGenerator @@ -43,7 +43,7 @@ if not args.usingJavascript: videos = video[:arg.amount] for video in videos: epoch = epoch + 1 - print(video['data-src']) + print("saving " + video['data-src'] + " as " + str(epoch) + '.mp4') urllib.request.urlretrieve(video['data-src'], '/home/' + os.environ['USER'] + '/Videos/unsorted/' + str(epoch) + '.mp4') elif args.usingJavascript: from selenium import webdriver @@ -92,7 +92,8 @@ elif args.usingJavascript: videos = videos[:args.amount] epoch = len([item for item in os.listdir('/home/'+ os.environ['USER'] + '/Videos/unsorted/')]) for video in videos: - print("saving " + video.get_attribute("data-src") + " as " + '/home/' + os.environ['USER'] + '/Videos/unsorted/' + str(epoch) + '.mp4') + epoch = epoch + 1 + print("saving " + video.get_attribute("data-src") + " as " + str(epoch) + '.mp4') urllib.request.urlretrieve(video.get_attribute("data-src"), '/home/' + os.environ['USER'] + '/Videos/unsorted/' + str(epoch) + '.mp4') driver.quit() -- cgit v1.2.3