def current_user(self):
"""Function to get the current user"""
# TODO: See if it can be scraped earlier without an extra request
if self.current_user_id is None:
req = session.base_get("index.php")
self.current_user_id = re.search(r"user.php\?id=(\d+)", req.text).group(1)
return CurrentUser(self.current_user_id)
def collage(self, coll_id, search_terms={}):
"""Simplistic representation of a collage, might be split out later"""
search_terms["id"] = coll_id
req = session.base_get("collages.php", params=search_terms)
movies = []
for movie in ptpapi.util.snarf_cover_view_data(req.content):
movie["Torrents"] = []
for group in movie["GroupingQualities"]:
movie["Torrents"].extend(group["Torrents"])
movies.append(ptpapi.Movie(data=movie))
return movies
def need_for_seed(self, filters={}):
"""List torrents that need seeding"""
data = ptpapi.util.snarf_cover_view_data(
session.base_get("needforseed.php", params=filters).content)
torrents = []
for m in data:
torrent = m['GroupingQualities'][0]['Torrents'][0]
torrent['Link'] = config.get('Main', 'baseURL') + bs4(
torrent['Title'], 'lxml').find('a')['href']
torrents.append(torrent)
return torrents
def contest_leaders(self):
"""Get data on who's winning"""
LOGGER.debug("Fetching contest leaderboard")
soup = bs4(session.base_get("contestleaders.php").content, "html.parser")
ret_array = []
for cell in (
soup.find("table", class_="table--panel-like").find("tbody").find_all("tr")
):
ret_array.append(
(cell.find_all("td")[1].get_text(), cell.find_all("td")[2].get_text())
)
return ret_array
def log(self):
"""Gets the PTP log"""
soup = bs4(session.base_get('log.php').content, "html.parser")
ret_array = []
for message in soup.find('table').find('tbody').find_all('tr'):
ret_array.append(
(message.find('span', class_='time')['title'],
message.find(
'span',
class_='log__message').get_text().lstrip().encode('UTF-8')
))
return ret_array
def artist(self, art_id, search_terms={}):
"""Simplistic representation of an artist page, might be split out later"""
search_terms["id"] = art_id
req = session.base_get("artist.php", params=search_terms)
movies = []
for movie in ptpapi.util.snarf_cover_view_data(
req.content, key=b"ungroupedCoverViewJsonData"
):
movie["Torrents"] = []
for group in movie["GroupingQualities"]:
movie["Torrents"].extend(group["Torrents"])
movies.append(ptpapi.Movie(data=movie))
return movies
def search_single(self, filters):
"""If you know ahead of time that a filter will redirect to a single movie,
you can use this method to avoid an exception until that behaivor is
fixed upstream."""
if "name" in filters:
filters["searchstr"] = filters["name"]
filters["json"] = "noredirect"
resp = session.base_get("torrents.php", params=filters)
id = re.search(r"id=([0-9]+)", resp.url)
if id is not None:
return Movie(ID=id.group(1))
else:
return None
def search_single(self, filters):
"""If you know ahead of time that a filter will redirect to a single movie,
you can use this method to avoid an exception until that behaivor is
fixed upstream."""
if 'name' in filters:
filters['searchstr'] = filters['name']
filters['json'] = 'noredirect'
resp = session.base_get('torrents.php', params=filters)
id = re.search(r'id=([0-9]+)', resp.url)
if id is not None:
return Movie(ID=id.group(1))
else:
return None
def need_for_seed(self, filters={}):
"""List torrents that need seeding"""
data = ptpapi.util.snarf_cover_view_data(
session.base_get("needforseed.php", params=filters).content
)
torrents = []
for m in data:
torrent = m["GroupingQualities"][0]["Torrents"][0]
torrent["Link"] = (
config.get("Main", "baseURL")
+ bs4(torrent["Title"], "lxml").find("a")["href"]
)
torrents.append(torrent)
return torrents
def search(self, filters):
"""Perform a movie search"""
if "name" in filters:
filters["searchstr"] = filters["name"]
filters["json"] = "noredirect"
ret_array = []
for movie in session.base_get("torrents.php", params=filters).json()["Movies"]:
if "Directors" not in movie:
movie["Directors"] = []
if "ImdbId" not in movie:
movie["ImdbId"] = "0"
movie["Title"] = html_parser.HTMLParser().unescape(movie["Title"])
ret_array.append(ptpapi.Movie(data=movie))
return ret_array
def log(self):
"""Gets the PTP log"""
soup = bs4(session.base_get("log.php").content, "html.parser")
ret_array = []
for message in soup.find("table").find("tbody").find_all("tr"):
ret_array.append(
(
message.find("span", class_="time")["title"],
message.find("span", class_="log__message")
.get_text()
.lstrip()
.encode("UTF-8"),
)
)
return ret_array
def search(self, filters):
"""Perform a movie search"""
if 'name' in filters:
filters['searchstr'] = filters['name']
filters['json'] = 'noredirect'
ret_array = []
for movie in session.base_get('torrents.php',
params=filters).json()['Movies']:
if 'Directors' not in movie:
movie['Directors'] = []
if 'ImdbId' not in movie:
movie['ImdbId'] = '0'
movie['Title'] = html_parser.HTMLParser().unescape(movie['Title'])
ret_array.append(ptpapi.Movie(data=movie))
return ret_array
def upload_info(self):
"""Scrape as much info as possible from upload.php"""
data = {}
soup = bs4(session.base_get("upload.php").content, "html.parser")
data["announce"] = soup.find_all(
"input",
type="text",
value=re.compile("http://please.passthepopcorn.me:2710/.*/announce"),
)[0]["value"]
data["subtitles"] = {}
label_list = [
u.find_all("label") for u in soup.find_all(class_="languageselector")
]
labels = [item for sublist in label_list for item in sublist]
for l in labels:
data["subtitles"][l["for"].lstrip("subtitle_")] = l.get_text().strip()
data["remaster_title"] = [
a.get_text() for a in soup.find(id="remaster_tags").find_all("a")
]
data["resolutions"] = [
o.get_text() for o in soup.find(id="resolution").find_all("option")
]
data["containers"] = [
o.get_text() for o in soup.find(id="container").find_all("option")
]
data["sources"] = [
o.get_text() for o in soup.find(id="source").find_all("option")
]
data["codecs"] = [
o.get_text() for o in soup.find(id="codec").find_all("option")
]
data["tags"] = [
o.get_text() for o in soup.find(id="genre_tags").find_all("option")
]
data["categories"] = [
o.get_text() for o in soup.find(id="categories").find_all("option")
]
return data
def __init__(self, username=None, password=None, passkey=None):
self.current_user_id = None
j = None
self.cookies_file = os.path.expanduser(config.get("Main", "cookiesFile"))
logger = logging.getLogger(__name__)
LOGGER.info("Initiating login sequence.")
if config.has_option("PTP", "ApiUser") and config.has_option("PTP", "ApiKey"):
pass
elif (
config.has_option("PTP", "password")
and config.has_option("PTP", "username")
and config.has_option("PTP", "passkey")
):
logger.warn(
"Using your password/passkey to access the site is deprecated, see README.md for instructions on using the new ApiUser/ApiKey."
)
else:
logger.critical("No credentials found! Exiting...")
sys.exit(78) # EX_CONFIG
req = None
if config.has_option("PTP", "ApiUser"):
session.headers.update(
{
"ApiUser": config.get("PTP", "ApiUser"),
"ApiKey": config.get("PTP", "ApiKey"),
}
)
elif os.path.isfile(self.cookies_file):
self.__load_cookies()
# A really crude test to see if we're logged in
session.max_redirects = 1
try:
req = session.base_get("torrents.php")
ptpapi.util.raise_for_cloudflare(req.text)
except requests.exceptions.TooManyRedirects:
if os.path.isfile(self.cookies_file):
os.remove(self.cookies_file)
session.cookies = requests.cookies.RequestsCookieJar()
session.max_redirects = 3
# If we're not using the new method and we don't have a cookie, get one
if not config.has_option("PTP", "ApiUser") and not os.path.isfile(
self.cookies_file
):
password = password or config.get("PTP", "password")
username = username or config.get("PTP", "username")
passkey = passkey or config.get("PTP", "passkey")
if not password or not passkey or not username:
raise PTPAPIException("Not enough info provided to log in.")
try:
req = session.base_post(
"ajax.php?action=login",
data={
"username": username,
"password": password,
"passkey": passkey,
},
)
j = req.json()
except ValueError:
if req.status_code == 200:
raise PTPAPIException("Could not parse returned json data.")
else:
if req.status_code == 429:
LOGGER.critical(req.text.strip())
req.raise_for_status()
if j["Result"] != "Ok":
raise PTPAPIException(
"Failed to log in. Please check the username, password and passkey. Response: %s"
% j
)
self.__save_cookie()
# Get some information that will be useful for later
req = session.base_get("index.php")
ptpapi.util.raise_for_cloudflare(req.text)
LOGGER.info("Login successful.")
def logout(self):
"""Forces a logout. In ApiUser mode, essentially a waste of two request tokens."""
req = session.base_get("index.php")
self.auth_key = re.search(r"auth=([0-9a-f]{32})", req.text).group(1)
os.remove(self.cookies_file)
return session.base_get("logout.php", params={"auth": self.auth_key})
