Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
ferru97 committed Mar 23, 2021
2 parents 4c7bfd6 + 4a1ebaf commit 074dd14
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 15 deletions.
29 changes: 15 additions & 14 deletions PyPaperBot/Downloader.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@
import time
from .HTMLparsers import getSchiHubPDF, SciHubUrls
import random
from urllib.parse import urljoin
from .NetInfo import NetInfo

def setSciHubUrl():
Expand All @@ -12,17 +11,19 @@ def setSciHubUrl():
found = False

for l in links:
r = requests.get(l, headers=NetInfo.HEADERS)
if r.status_code == 200:
found = True
NetInfo.SciHub_URL = l
break

try:
r = requests.get(l, headers=NetInfo.HEADERS)
if r.status_code == 200:
found = True
NetInfo.SciHub_URL = l
break
except:
pass
if found:
print("\nUsing {} as Sci-Hub instance".format(NetInfo.SciHub_URL))
else:
print("\nNo working Sci-Hub instance found!\nIf in your country Sci-Hub is not available consider using a VPN")
NetInfo.SciHub_URL = "https://sci-hub.st/"
NetInfo.SciHub_URL = "https://sci-hub.st"


def getSaveDir(folder, fname):
Expand All @@ -44,9 +45,10 @@ def saveFile(file_name,content, paper,dwn_source):


def downloadPapers(papers, dwnl_dir, num_limit, SciHub_URL=None):
if SciHub_URL:
NetInfo.SciHub_URL = SciHub_URL

def URLjoin(*args):
return "/".join(map(lambda x: str(x).rstrip('/'), args))

NetInfo.SciHub_URL = SciHub_URL
if NetInfo.SciHub_URL==None:
setSciHubUrl()

Expand All @@ -66,10 +68,9 @@ def downloadPapers(papers, dwnl_dir, num_limit, SciHub_URL=None):

dwn_source = 1 #1 scihub 2 scholar
if faild==0 and p.DOI!=None:
url = urljoin(NetInfo.SciHub_URL, p.DOI)
url = URLjoin(NetInfo.SciHub_URL, p.DOI)
if faild==1 and p.scholar_link!=None:
url = urljoin(NetInfo.SciHub_URL, p.scholar_link)

url = URLjoin(NetInfo.SciHub_URL, p.scholar_link)
if faild==2 and p.scholar_link!=None and p.scholar_link[-3:]=="pdf":
url = p.scholar_link
dwn_source = 2
Expand Down
2 changes: 1 addition & 1 deletion PyPaperBot/__init__.py
Original file line number Diff line number Diff line change
@@ -1 +1 @@
__version__= "1.0.2"
__version__= "1.0.2"

0 comments on commit 074dd14

Please sign in to comment.