JDM170/pornhub_dl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix single video download mode

Browse Source
This commit is contained in:
tnt2402
2022-12-29 20:48:17 +07:00
parent 91765c9ded
commit a7d0d70b08
3 changed files with 27 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
__pycache__/lib_pornhub.cpython-311.pyc Normal file
View File

Binary file not shown.

88
lib_pornhub.py
View File

@@ -1,15 +1,12 @@
#!/usr/bin/env python #!/usr/bin/env python
import youtube_dl
import requests as req
import sys import sys
import os import os
from bs4 import BeautifulSoup
import ast import ast
import string import string
import subprocess import subprocess
import pyfiglet import pyfiglet
import signal import signal
import mechanicalsoup import json
############# #############
def ascii_banner(text): def ascii_banner(text):
@@ -17,17 +14,8 @@ def ascii_banner(text):
ascii_banner = pyfiglet.figlet_format(text) ascii_banner = pyfiglet.figlet_format(text)
print(ascii_banner) print(ascii_banner)
################################ ################################
### Configuration
ydl_opts_start = {
'nooverwrites': True,
'no_warnings': True,
'ignoreerrors': True,
'format': "bestvideo",
}
ydl = youtube_dl.YoutubeDL(ydl_opts_start)
download_dir = os.getcwd() download_dir = os.getcwd()
def ph_config_dl_dir(dir): def ph_config_dl_dir(dir):
@@ -53,7 +41,12 @@ def ph_check_valid_pornhub_url(url):
def download_video(url, filename): def download_video(url, filename):
try: try:
p = subprocess.run(['.\youtube-dl', '--hls-prefer-ffmpeg', '--ffmpeg-location', os.getcwd(), '-o', filename, url], shell = True) p = subprocess.Popen(['.\youtube-dl', '--no-warnings', '--hls-prefer-ffmpeg', '--ffmpeg-location', os.getcwd(), '-o', filename, url], stdout=subprocess.PIPE, stderr=subprocess.STDOUT, shell = False)
for line in iter(p.stdout.readline, b''):
line = line.decode('utf-8').strip()
if ("[ffmpeg] " in line or "[download] " in line):
print(line.strip())
print("[$] Video download successfully!")
except KeyboardInterrupt: except KeyboardInterrupt:
os.kill(p.pid, signal.CTRL_C_EVENT) os.kill(p.pid, signal.CTRL_C_EVENT)
sys.exit() sys.exit()
@@ -74,7 +67,6 @@ def check_output_dir(model_name):
def ph_download_video(url, model_name, filename): def ph_download_video(url, model_name, filename):
try: try:
check_output_dir(model_name) check_output_dir(model_name)
# video = ydl.extract_info(url, download=False)
# filename = os.path.join(download_dir, model_name, fix_title(str(video["title"])) + '.' + str(video['ext'])) # filename = os.path.join(download_dir, model_name, fix_title(str(video["title"])) + '.' + str(video['ext']))
download_video(url, filename) download_video(url, filename)
except: except:
@@ -107,57 +99,31 @@ def ph_download_playlist(url, model_name, limit):
except: except:
print("Cannot download video") print("Cannot download video")
def get_model_name(url): # def fix_url(url, type):
# url = ph_check_valid_pornhub_url(url)
# html = req.get(url).text # # model_name = get_model_name(url)
# soup = BeautifulSoup(html, 'lxml') # url = 'www.pornhub.com/model/' + model_name + '/videos'
# if (type == 'most-viewed'):
browser = mechanicalsoup.Browser() # url = url + '?o=mv'
# elif (type == 'top-rated'):
page = browser.get(url) # url = url + '?o=tr'
# elif (type == 'longest'):
soup = BeautifulSoup(page.text, 'lxml') # url = url + '?o=lg'
# print("[+] Model: " + model_name)
if ("Page Not Found" in soup.title): # return (url, model_name)
print("Page not found!")
sys.exit()
user = soup.find(class_='userInfo')
if (user != None):
finder = user.find(class_='usernameBadgesWrapper')
name = finder.find(class_='bolded').text.replace('\n', '').strip()
else:
user = soup.find(class_='topProfileHeader')
finder = user.find(class_='nameSubscribe')
name = finder.find(class_='name').text.replace('\n', '').strip()
return name
def fix_url(url, type):
url = ph_check_valid_pornhub_url(url)
model_name = get_model_name(url)
url = 'www.pornhub.com/model/' + model_name + '/videos'
if (type == 'most-viewed'):
url = url + '?o=mv'
elif (type == 'top-rated'):
url = url + '?o=tr'
elif (type == 'longest'):
url = url + '?o=lg'
print("[+] Model: " + model_name)
return (url, model_name)
def ph_get_video(url): def ph_get_video(url):
url = ph_check_valid_pornhub_url(url) url = ph_check_valid_pornhub_url(url)
info = ydl.extract_info(url, download=False)
retry = 0 p = subprocess.Popen(['.\youtube-dl', '--no-warnings', '--dump-json', '--skip-download', url], stdout=subprocess.PIPE, stderr=None, shell = True)
while (info == None and retry < 5): output = p.communicate()[0]
info = ydl.extract_info(url, download=False) info = json.loads(output.decode('utf-8'))
retry += 1
model_name = info['uploader'] model_name = info['uploader']
filename = os.path.join(download_dir, model_name, fix_title(str(info["title"])) + '.' + str(info['ext'])) filename = os.path.join(download_dir, model_name, fix_title(str(info["title"])) + '.' + str(info['ext']))
print("[+] Model: " + model_name) print("[+] Model: " + model_name)
print("[+] Filename: " + fix_title(str(info["title"])) + '.' + str(info['ext']))
ph_download_video(url, model_name, filename) ph_download_video(url, model_name, filename)
def ph_get_playlist(url, type, limit): # def ph_get_playlist(url, type, limit):
url, model_name = fix_url(url, type) # url, model_name = fix_url(url, type)
ph_download_playlist(url, model_name, limit) # ph_download_playlist(url, model_name, limit)

0
regex.txt Normal file
View File