github/gallery-dl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[oauth] import extractor modules on demand

Browse Source
This commit is contained in:
Mike Fährmann
2023-03-02 18:39:54 +01:00
parent dd88740ec7
commit b8a702929d
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
gallery_dl/extractor/oauth.py
View File

@@ -9,7 +9,6 @@
"""Utility classes to setup OAuth and link accounts to gallery-dl""" """Utility classes to setup OAuth and link accounts to gallery-dl"""
from .common import Extractor, Message from .common import Extractor, Message
from . import deviantart, flickr, mastodon, pixiv, reddit, smugmug, tumblr
from .. import text, oauth, util, config, exception from .. import text, oauth, util, config, exception
from ..output import stdout_write from ..output import stdout_write
from ..cache import cache from ..cache import cache
@@ -243,6 +242,7 @@ class OAuthFlickr(OAuthBase):
def items(self): def items(self):
yield Message.Version, 1 yield Message.Version, 1
from . import flickr
self._oauth1_authorization_flow( self._oauth1_authorization_flow(
flickr.FlickrAPI.API_KEY, flickr.FlickrAPI.API_KEY,
@@ -259,6 +259,7 @@ class OAuthSmugmug(OAuthBase):
def items(self): def items(self):
yield Message.Version, 1 yield Message.Version, 1
from . import smugmug
self._oauth1_authorization_flow( self._oauth1_authorization_flow(
smugmug.SmugmugAPI.API_KEY, smugmug.SmugmugAPI.API_KEY,
@@ -275,6 +276,7 @@ class OAuthTumblr(OAuthBase):
def items(self): def items(self):
yield Message.Version, 1 yield Message.Version, 1
from . import tumblr
self._oauth1_authorization_flow( self._oauth1_authorization_flow(
tumblr.TumblrAPI.API_KEY, tumblr.TumblrAPI.API_KEY,
@@ -295,6 +297,7 @@ class OAuthDeviantart(OAuthBase):
def items(self): def items(self):
yield Message.Version, 1 yield Message.Version, 1
from . import deviantart
self._oauth2_authorization_code_grant( self._oauth2_authorization_code_grant(
self.oauth_config("client-id"), self.oauth_config("client-id"),
@@ -314,6 +317,7 @@ class OAuthReddit(OAuthBase):
def items(self): def items(self):
yield Message.Version, 1 yield Message.Version, 1
from . import reddit
self.session.headers["User-Agent"] = reddit.RedditAPI.USER_AGENT self.session.headers["User-Agent"] = reddit.RedditAPI.USER_AGENT
self._oauth2_authorization_code_grant( self._oauth2_authorization_code_grant(
@@ -338,6 +342,7 @@ class OAuthMastodon(OAuthBase):
def items(self): def items(self):
yield Message.Version, 1 yield Message.Version, 1
from . import mastodon
for application in mastodon.INSTANCES.values(): for application in mastodon.INSTANCES.values():
if self.instance == application["root"].partition("://")[2]: if self.instance == application["root"].partition("://")[2]:
@@ -390,6 +395,7 @@ class OAuthPixiv(OAuthBase):
def items(self): def items(self):
yield Message.Version, 1 yield Message.Version, 1
from . import pixiv
code_verifier = util.generate_token(32) code_verifier = util.generate_token(32)
digest = hashlib.sha256(code_verifier.encode()).digest() digest = hashlib.sha256(code_verifier.encode()).digest()