Mirrors/gallery-dl
1
0
Fork 0
mirror of https://github.com/mikf/gallery-dl.git synced 2024-11-22 18:53:21 +01:00
Code Issues Releases Wiki Activity

[gofile] add basic password support (#4056)

Browse Source
This commit is contained in:
Mike Fährmann 2023-05-13 15:46:29 +02:00
parent 790dd365e1
commit e41e45ff6b
No known key found for this signature in database
GPG Key ID: 5680CA389D365A88
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
gallery_dl/extractor/gofile.py
View File

@ -7,6 +7,7 @@
from .common import Extractor, Message
from .. import text, exception
from ..cache import memcache
import hashlib
class GofileFolderExtractor(Extractor):
@ -66,6 +67,7 @@ class GofileFolderExtractor(Extractor):
def items(self):
recursive = self.config("recursive")
password = self.config("password")
token = self.config("api-token")
if not token:
@ -78,7 +80,7 @@ class GofileFolderExtractor(Extractor):
token = self._get_website_token()
self.website_token = token
folder = self._get_content(self.content_id)
folder = self._get_content(self.content_id, password)
yield Message.Directory, folder
num = 0
@ -115,11 +117,14 @@ class GofileFolderExtractor(Extractor):
page = self.request(self.root + "/contents/files.html").text
return text.extract(page, "websiteToken:", ",")[0].strip("\" ")
def _get_content(self, content_id):
def _get_content(self, content_id, password=None):
if password is not None:
password = hashlib.sha256(password.encode()).hexdigest()
return self._api_request("getContent", {
"contentId" : content_id,
"token" : self.api_token,
"websiteToken": self.website_token,
"password" : password,
})
def _api_request(self, endpoint, params=None):