CodeMirror/bulk-downloader-for-reddit
1
0
Fork 0
mirror of synced 2024-06-27 02:20:45 +12:00
Code Issues Releases Wiki Activity

Fix OAuth2

Browse source
This commit is contained in:
Serene-Arc 2021-03-08 12:46:32 +10:00 committed by Ali Parlakci
parent 36f516e3f0
commit 862121cac6
2 changed files with 13 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
bulkredditdownloader/downloader.py
View file

@ -72,7 +72,10 @@ class RedditDownloader:
if not self.cfg_parser.has_option('DEFAULT', 'user_token'):
scopes = self.cfg_parser.get('DEFAULT', 'scopes')
scopes = OAuth2Authenticator.split_scopes(scopes)
oauth2_authenticator = OAuth2Authenticator(scopes)
oauth2_authenticator = OAuth2Authenticator(
scopes,
self.cfg_parser.get('DEFAULT', 'client_id'),
self.cfg_parser.get('DEFAULT', 'client_secret'))
token = oauth2_authenticator.retrieve_new_token()
self.cfg_parser['DEFAULT']['user_token'] = token
token_manager = OAuth2TokenManager(self.cfg_parser)

12
bulkredditdownloader/oauth2.py
View file

@ -17,9 +17,11 @@ logger = logging.getLogger(__name__)
class OAuth2Authenticator:
def __init__(self, wanted_scopes: set[str]):
def __init__(self, wanted_scopes: set[str], client_id: str, client_secret: str):
self._check_scopes(wanted_scopes)
self.scopes = wanted_scopes
self.client_id = client_id
self.client_secret = client_secret
@staticmethod
def _check_scopes(wanted_scopes: set[str]):
@ -37,7 +39,11 @@ class OAuth2Authenticator:
return set(scopes)
def retrieve_new_token(self) -> str:
reddit = praw.Reddit(redirect_uri='http://localhost:8080', user_agent='obtain_refresh_token for BDFR')
reddit = praw.Reddit(
redirect_uri='http://localhost:7634',
user_agent='obtain_refresh_token for BDFR',
client_id=self.client_id,
client_secret=self.client_secret)
state = str(random.randint(0, 65000))
url = reddit.auth.url(self.scopes, state, 'permanent')
logger.warning('Authentication action required before the program can proceed')
@ -62,7 +68,7 @@ class OAuth2Authenticator:
def receive_connection() -> socket.socket:
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
server.bind(('localhost', 8080))
server.bind(('localhost', 7634))
logger.debug('Server listening on localhost:8080')
server.listen(1)