add ctrl-c to exit process

This commit is contained in:
tnt2402
2022-09-12 21:30:12 +07:00
parent cc28568a80
commit a64c7bbf4f
2 changed files with 10 additions and 3 deletions

View File

@@ -8,6 +8,7 @@ import ast
import string
import subprocess
import pyfiglet
import signal
#############
def ascii_banner(text):
@@ -49,12 +50,18 @@ def ph_check_valid_pornhub_url(url):
url = 'https://www.' + url
return url
def download_video(url, filename):
try:
p = subprocess.run(["downloadm3u8", "-o", filename, url])
except KeyboardInterrupt:
os.kill(p.pid, signal.CTRL_C_EVENT)
sys.exit()
def fix_title(s):
decoded_unicode = ''.join([i if i in string.printable else ' ' for i in s])
deny_char = ['\\', '/', '.', '?', '*', ':']
for i in deny_char:
decoded_unicode.replace[i, ' ']
decoded_unicode = decoded_unicode.replace(i, '')
return decoded_unicode
def check_output_dir(model_name):
@@ -71,7 +78,7 @@ def ph_download_video(url, model_name):
print(' [-] {}'.format(filename))
filename = download_dir + '\\' + model_name + '\\' + filename
url_video = video['url']
res = subprocess.run(["downloadm3u8", "-o", filename, url_video])
download_video(url_video, filename)
except:
print('Cannot download video')
@@ -96,7 +103,7 @@ def ph_download_playlist(url, model_name, limit):
print('\n\n\n\n#######################\n\n[-] Video #{}: {}\n\n'.format(count, filename))
filename = download_dir + '\\' + model_name + '\\' + filename
url_video = video['url']
result = subprocess.run(["downloadm3u8", "-o", filename, url_video])
download_video(url_video, filename)
count = count + 1
except:
print("Cannot download video")