Oracle/LibRay
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Release 0.0.5

Browse source
This commit is contained in:
Nichlas Severinsen 2020-08-03 19:52:46 +02:00
parent df4c1a6f36
commit c3b90319eb
4 changed files with 33 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

14
libray/core.py
View file

@ -95,11 +95,16 @@ def error(msg):
def warning(msg):
"""Print a warning message"""
print('WARNING: %s. Continuing regardless' % msg)
print('WARNING: %s. Continuing regardless.' % msg)
def download_ird(ird_name):
"""Download an .ird from GET_IRD_NET_LOC"""
# Check if file already exists and skip if it does
if os.path.exists(ird_name):
return
ird_link = GET_IRD_NET_LOC + ird_name
r = requests.get(ird_link, stream=True)
@ -111,7 +116,10 @@ def download_ird(ird_name):
def ird_by_game_id(game_id):
"""Using a game_id, download the responding .ird from ALL_IRD_NET_LOC"""
gameid = game_id.replace('-','')
r = requests.get(ALL_IRD_NET_LOC, headers = {'User-Agent': 'Anonymous (You)' }, timeout=5)
try:
r = requests.get(ALL_IRD_NET_LOC, headers = {'User-Agent': 'Anonymous (You)' }, timeout=5)
except requests.exceptions.ReadTimeout:
core.error('Server timed out, fix your connection or manually specify a key/ird.')
soup = BeautifulSoup(r.text, "html.parser")
ird_name = False
@ -121,7 +129,7 @@ def ird_by_game_id(game_id):
ird_name = url
if not ird_name:
error("Unable to download IRD, couldn't find link")
error("Unable to download IRD, couldn't find link. You could specify the decryption key with -d if you have it.")
download_ird(ird_name)