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

[reddit] allow users to override the API User-Agent

Browse Source 
Only overriding the Client-ID is not enough if you want to follow
Reddit's API access rules [1].

[1] https://github.com/reddit/reddit/wiki/API#rules
This commit is contained in:
Mike Fährmann
2017-10-10 17:29:46 +02:00
parent 31ea6001e8
commit f3fbaa5c3e
2 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
gallery_dl/extractor/reddit.py
View File

@@ -110,11 +110,21 @@ class RedditAPI():
self.extractor = extractor
self.comments = extractor.config("comments", 500)
self.morecomments = extractor.config("morecomments", False)
self.client_id = extractor.config("client-id", self.CLIENT_ID)
self.refresh_token = extractor.config("refresh-token")
self.log = extractor.log
self.session = extractor.session
self.session.headers["User-Agent"] = self.USER_AGENT
client_id = extractor.config("client-id", self.CLIENT_ID)
user_agent = extractor.config("user-agent", self.USER_AGENT)
if (client_id == self.CLIENT_ID) ^ (user_agent == self.USER_AGENT):
self.client_id = None
self.log.warning(
"Conflicting values for 'client-id' and 'user-agent': "
"override either both or non of them.")
else:
self.client_id = client_id
self.session.headers["User-Agent"] = user_agent
def submission(self, submission_id):
"""Fetch the (submission, comments)=-tuple for a submission id"""