38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
|
from cliff.command import Command
|
||
|
|
import dcc.config
|
||
|
|
import io
|
||
|
|
import json
|
||
|
|
import pathlib
|
||
|
|
import urllib.request
|
||
|
|
import zipfile
|
||
|
|
|
||
|
|
class Fetch(Command):
|
||
|
|
def get_parser(self, prog_name):
|
||
|
|
parser = super().get_parser(prog_name)
|
||
|
|
parser.add_argument("id_or_name")
|
||
|
|
return parser
|
||
|
|
|
||
|
|
def take_action(self, parsed_args):
|
||
|
|
idgames_id = parsed_args.id_or_name
|
||
|
|
if not parsed_args.id_or_name.isdigit():
|
||
|
|
idgames_id = self.search_idgames(parsed_args.id_or_name)
|
||
|
|
|
||
|
|
with urllib.request.urlopen("https://www.doomworld.com/idgames/api/api.php?action=get&id={}&out=json".format(idgames_id)) as response:
|
||
|
|
reply = json.loads(response.read())
|
||
|
|
rpath = "/".join([dcc.config.MIRROR, reply["content"]["dir"], reply["content"]["filename"]])
|
||
|
|
wad = reply["content"]["filename"][0:-4]
|
||
|
|
|
||
|
|
with urllib.request.urlopen(rpath) as response:
|
||
|
|
z = zipfile.ZipFile(io.BytesIO(response.read()))
|
||
|
|
z.extractall(path=dcc.config.PwadPath(wad))
|
||
|
|
|
||
|
|
# TODO: explicit error handling. Let users choose when >1 result.
|
||
|
|
def search_idgames(self, wad):
|
||
|
|
with urllib.request.urlopen("https://www.doomworld.com/idgames/api/api.php?action=search&query={}&out=json".format(wad)) as response:
|
||
|
|
reply = json.loads(response.read())
|
||
|
|
files = reply["content"]["file"]
|
||
|
|
if type(files) is dict: # One result.
|
||
|
|
return files["id"]
|
||
|
|
else: # More than one. Zero will raise an error.
|
||
|
|
return files[0]["id"]
|