mirror of
https://github.com/mikf/gallery-dl.git
synced 2024-11-22 18:53:21 +01:00
156a70bec0
- add test case - add docs/supportedsites entry - add custom directory_fmt and archive_fmt - remove constructor - appease flake8
249 lines
8.2 KiB
Python
249 lines
8.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
# Copyright 2020-2023 Mike Fährmann
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License version 2 as
|
|
# published by the Free Software Foundation.
|
|
|
|
"""Extractors for https://aryion.com/"""
|
|
|
|
from .common import Extractor, Message
|
|
from .. import text, util, exception
|
|
from ..cache import cache
|
|
from email.utils import parsedate_tz
|
|
from datetime import datetime
|
|
|
|
BASE_PATTERN = r"(?:https?://)?(?:www\.)?aryion\.com/g4"
|
|
|
|
|
|
class AryionExtractor(Extractor):
|
|
"""Base class for aryion extractors"""
|
|
category = "aryion"
|
|
directory_fmt = ("{category}", "{user!l}", "{path:J - }")
|
|
filename_fmt = "{id} {title}.{extension}"
|
|
archive_fmt = "{id}"
|
|
cookies_domain = ".aryion.com"
|
|
cookies_names = ("phpbb3_rl7a3_sid",)
|
|
root = "https://aryion.com"
|
|
|
|
def __init__(self, match):
|
|
Extractor.__init__(self, match)
|
|
self.user = match.group(1)
|
|
self.recursive = True
|
|
|
|
def login(self):
|
|
if self.cookies_check(self.cookies_names):
|
|
return
|
|
|
|
username, password = self._get_auth_info()
|
|
if username:
|
|
self.cookies_update(self._login_impl(username, password))
|
|
|
|
@cache(maxage=14*86400, keyarg=1)
|
|
def _login_impl(self, username, password):
|
|
self.log.info("Logging in as %s", username)
|
|
|
|
url = self.root + "/forum/ucp.php?mode=login"
|
|
data = {
|
|
"username": username,
|
|
"password": password,
|
|
"login": "Login",
|
|
}
|
|
|
|
response = self.request(url, method="POST", data=data)
|
|
if b"You have been successfully logged in." not in response.content:
|
|
raise exception.AuthenticationError()
|
|
return {c: response.cookies[c] for c in self.cookies_names}
|
|
|
|
def items(self):
|
|
self.login()
|
|
data = self.metadata()
|
|
|
|
for post_id in self.posts():
|
|
post = self._parse_post(post_id)
|
|
if post:
|
|
if data:
|
|
post.update(data)
|
|
yield Message.Directory, post
|
|
yield Message.Url, post["url"], post
|
|
elif post is False and self.recursive:
|
|
base = self.root + "/g4/view/"
|
|
data = {"_extractor": AryionPostExtractor}
|
|
for post_id in self._pagination_params(base + post_id):
|
|
yield Message.Queue, base + post_id, data
|
|
|
|
def posts(self):
|
|
"""Yield relevant post IDs"""
|
|
|
|
def metadata(self):
|
|
"""Return general metadata"""
|
|
|
|
def _pagination_params(self, url, params=None, needle=None):
|
|
if params is None:
|
|
params = {"p": 1}
|
|
else:
|
|
params["p"] = text.parse_int(params.get("p"), 1)
|
|
|
|
if needle is None:
|
|
needle = "class='gallery-item' id='"
|
|
|
|
while True:
|
|
page = self.request(url, params=params).text
|
|
|
|
cnt = 0
|
|
for post_id in text.extract_iter(page, needle, "'"):
|
|
cnt += 1
|
|
yield post_id
|
|
|
|
if cnt < 40:
|
|
return
|
|
params["p"] += 1
|
|
|
|
def _pagination_next(self, url):
|
|
while True:
|
|
page = self.request(url).text
|
|
yield from text.extract_iter(page, "thumb' href='/g4/view/", "'")
|
|
|
|
pos = page.find("Next >>")
|
|
if pos < 0:
|
|
return
|
|
url = self.root + text.rextract(page, "href='", "'", pos)[0]
|
|
|
|
def _parse_post(self, post_id):
|
|
url = "{}/g4/data.php?id={}".format(self.root, post_id)
|
|
with self.request(url, method="HEAD", fatal=False) as response:
|
|
|
|
if response.status_code >= 400:
|
|
self.log.warning(
|
|
"Unable to fetch post %s ('%s %s')",
|
|
post_id, response.status_code, response.reason)
|
|
return None
|
|
headers = response.headers
|
|
|
|
# folder
|
|
if headers["content-type"] in (
|
|
"application/x-folder",
|
|
"application/x-comic-folder",
|
|
"application/x-comic-folder-nomerge",
|
|
):
|
|
return False
|
|
|
|
# get filename from 'Content-Disposition' header
|
|
cdis = headers["content-disposition"]
|
|
fname, _, ext = text.extr(cdis, 'filename="', '"').rpartition(".")
|
|
if not fname:
|
|
fname, ext = ext, fname
|
|
|
|
# get file size from 'Content-Length' header
|
|
clen = headers.get("content-length")
|
|
|
|
# fix 'Last-Modified' header
|
|
lmod = headers["last-modified"]
|
|
if lmod[22] != ":":
|
|
lmod = "{}:{} GMT".format(lmod[:22], lmod[22:24])
|
|
|
|
post_url = "{}/g4/view/{}".format(self.root, post_id)
|
|
extr = text.extract_from(self.request(post_url).text)
|
|
|
|
title, _, artist = text.unescape(extr(
|
|
"<title>g4 :: ", "<")).rpartition(" by ")
|
|
|
|
return {
|
|
"id" : text.parse_int(post_id),
|
|
"url" : url,
|
|
"user" : self.user or artist,
|
|
"title" : title,
|
|
"artist": artist,
|
|
"path" : text.split_html(extr(
|
|
"cookiecrumb'>", '</span'))[4:-1:2],
|
|
"date" : datetime(*parsedate_tz(lmod)[:6]),
|
|
"size" : text.parse_int(clen),
|
|
"views" : text.parse_int(extr("Views</b>:", "<").replace(",", "")),
|
|
"width" : text.parse_int(extr("Resolution</b>:", "x")),
|
|
"height": text.parse_int(extr("", "<")),
|
|
"comments" : text.parse_int(extr("Comments</b>:", "<")),
|
|
"favorites": text.parse_int(extr("Favorites</b>:", "<")),
|
|
"tags" : text.split_html(extr("class='taglist'>", "</span>")),
|
|
"description": text.unescape(text.remove_html(extr(
|
|
"<p>", "</p>"), "", "")),
|
|
"filename" : fname,
|
|
"extension": ext,
|
|
"_mtime" : lmod,
|
|
}
|
|
|
|
|
|
class AryionGalleryExtractor(AryionExtractor):
|
|
"""Extractor for a user's gallery on eka's portal"""
|
|
subcategory = "gallery"
|
|
categorytransfer = True
|
|
pattern = BASE_PATTERN + r"/(?:gallery/|user/|latest.php\?name=)([^/?#]+)"
|
|
example = "https://aryion.com/g4/gallery/USER"
|
|
|
|
def __init__(self, match):
|
|
AryionExtractor.__init__(self, match)
|
|
self.offset = 0
|
|
|
|
def _init(self):
|
|
self.recursive = self.config("recursive", True)
|
|
|
|
def skip(self, num):
|
|
if self.recursive:
|
|
return 0
|
|
self.offset += num
|
|
return num
|
|
|
|
def posts(self):
|
|
if self.recursive:
|
|
url = "{}/g4/gallery/{}".format(self.root, self.user)
|
|
return self._pagination_params(url)
|
|
else:
|
|
url = "{}/g4/latest.php?name={}".format(self.root, self.user)
|
|
return util.advance(self._pagination_next(url), self.offset)
|
|
|
|
|
|
class AryionFavoriteExtractor(AryionExtractor):
|
|
"""Extractor for a user's favorites gallery"""
|
|
subcategory = "favorite"
|
|
directory_fmt = ("{category}", "{user!l}", "favorites")
|
|
archive_fmt = "f_{user}_{id}"
|
|
categorytransfer = True
|
|
pattern = BASE_PATTERN + r"/favorites/([^/?#]+)"
|
|
example = "https://aryion.com/g4/favorites/USER"
|
|
|
|
def posts(self):
|
|
url = "{}/g4/favorites/{}".format(self.root, self.user)
|
|
return self._pagination_params(
|
|
url, None, "class='gallery-item favorite' id='")
|
|
|
|
|
|
class AryionTagExtractor(AryionExtractor):
|
|
"""Extractor for tag searches on eka's portal"""
|
|
subcategory = "tag"
|
|
directory_fmt = ("{category}", "tags", "{search_tags}")
|
|
archive_fmt = "t_{search_tags}_{id}"
|
|
pattern = BASE_PATTERN + r"/tags\.php\?([^#]+)"
|
|
example = "https://aryion.com/g4/tags.php?tag=TAG"
|
|
|
|
def _init(self):
|
|
self.params = text.parse_query(self.user)
|
|
self.user = None
|
|
|
|
def metadata(self):
|
|
return {"search_tags": self.params.get("tag")}
|
|
|
|
def posts(self):
|
|
url = self.root + "/g4/tags.php"
|
|
return self._pagination_params(url, self.params)
|
|
|
|
|
|
class AryionPostExtractor(AryionExtractor):
|
|
"""Extractor for individual posts on eka's portal"""
|
|
subcategory = "post"
|
|
pattern = BASE_PATTERN + r"/view/(\d+)"
|
|
example = "https://aryion.com/g4/view/12345"
|
|
|
|
def posts(self):
|
|
post_id, self.user = self.user, None
|
|
return (post_id,)
|