def main():
aapi = AppPixivAPI()
# aapi.set_additional_headers({'Accept-Language':'en-US'})
aapi.set_accept_language("en-us") # zh-cn
aapi.auth(refresh_token=_REFRESH_TOKEN)
json_result = aapi.illust_ranking(
"day", date=(datetime.now() - timedelta(days=5)).strftime("%Y-%m-%d"))
print(
"Printing image titles and tags with English tag translations present when available"
)
for illust in json_result.illusts[:3]:
print('Illustration: "' + str(illust.title) + '"\nTags: ' +
str(illust.tags) + "\n")
class Pixiv:
def __init__(self):
self.enabled = config.getboolean('Pixiv', 'enabled', fallback=False)
self._refresh_token = config.get('Pixiv', 'refresh_token', fallback=None)
self._log = logging.getLogger(__name__)
self._pixiv = AppPixivAPI()
self._pixiv.set_accept_language(config.get('Pixiv', 'language', fallback='en-US'))
self._re_twitter = re.compile(r'^https?://(www.)?twitter.com/(?P<username>.+)$')
if self.enabled:
self._login()
def _login(self) -> None:
"""
Authenticate to Pixiv
Returns:
None
"""
self._log.debug(f'[PIXIV] Authenticating to Pixiv with the token {self._refresh_token}')
try:
self._pixiv.auth(refresh_token=self._refresh_token)
except Exception as error:
self._log.exception("[PIXIV] Failed to authenticate to Pixiv", exc_info=error)
def get_illust(self, illust_id: int) -> typing.Optional[dict]:
"""
Look up the provided illustration ID from SauceNao
Args:
illust_id (int):
Returns:
typing.Optional[dict]
"""
if not self.enabled:
return None
illust = self._pixiv.illust_detail(illust_id)
if 'error' in illust and 'invalid_grant' in illust['error']['message']:
self._log.warning(f'Pixiv login session is no longer valid; re-authenticating with saved token')
self._login()
illust = self._pixiv.illust_detail(illust_id)
return illust['illust'] if illust and 'illust' in illust else None
def get_author(self, author_id: int) -> typing.Optional[dict]:
"""
Get the author for the specified illustration
Args:
author_id (int):
Returns:
typing.Optional[dict]
"""
if not self.enabled:
return None
user = self._pixiv.user_detail(author_id)
if 'error' in user and 'invalid_grant' in user['error']['message']:
self._log.info(f'Re-Authenticating to Pixiv with the saved refresh token')
self._login()
user = self._pixiv.user_detail(author_id)
return user
def get_author_twitter(self, author_id: int) -> typing.Optional[str]:
"""
Get the Pixiv artists Twitter page, if available
Args:
author_id (int):
Returns:
typing.Optional[str]
"""
if not self.enabled:
return None
user = self.get_author(author_id)
twitter_url = user['profile']['twitter_url'] if (user and 'profile' in user) else None
if twitter_url:
match = self._re_twitter.match(twitter_url)
if match and match.group('username'):
return f"@{match.group('username')}"
