add algo-method support (get-problem only)

Author: takeaship

README.md

@@ -23,6 +23,7 @@ $ pip3 install online-judge-api-client
 
 | website                                                                        | get sample cases   | get system cases   | get metadata       | get contest data   | login service      | submit code        |
 |--------------------------------------------------------------------------------|--------------------|--------------------|--------------------|--------------------|--------------------|--------------------|
+| [Algo-Method](https://algo-method.com)                     | :heavy_check_mark: |                    |                    |                    |                    |                    |
 | [Aizu Online Judge](https://onlinejudge.u-aizu.ac.jp/home)                     | :heavy_check_mark: | :heavy_check_mark: |                    |                    |                    |                    |
 | [Anarchy Golf](http://golf.shinh.org/)                                         | :heavy_check_mark: | :grey_question: (same to samples) |                    |                    |                    |                    |
 | [AtCoder](https://atcoder.jp/)                                                 | :heavy_check_mark: | :heavy_check_mark: | :heavy_check_mark: | :heavy_check_mark: | :heavy_check_mark: | :heavy_check_mark: |

onlinejudge/service/__init__.py

@@ -1,4 +1,5 @@
 # Python Version: 3.x
+import onlinejudge.service.algo_method
 import onlinejudge.service.anarchygolf
 import onlinejudge.service.aoj
 import onlinejudge.service.atcoder

onlinejudge/service/algo_method.py

@@ -0,0 +1,81 @@
+# Python Version: 3.x
+# -*- coding: utf-8 -*-
+"""
+the module for Algo-Method (https://algo-method.com/)
+"""
+
+import urllib.parse
+from typing import *
+import json
+import re
+
+import bs4
+import requests
+
+import onlinejudge._implementation.testcase_zipper
+import onlinejudge._implementation.utils as utils
+import onlinejudge.dispatch
+import onlinejudge.type
+
+import posixpath
+
+class AlgoMethodService(onlinejudge.type.Service):
+    def get_url(self) -> str:
+        return 'https://algo-method.com/'
+
+    def get_name(self) -> str:
+        return 'algo-method'
+
+    @classmethod
+    def from_url(cls, url: str) -> Optional['AlgoMethodService']:
+        # example: https://algo-method.com/
+        result = urllib.parse.urlparse(url)
+        if result.scheme in ('', 'http', 'https') \
+                and result.netloc == 'algo-method.com':
+            return cls()
+        return None
+
+
+class AlgoMethodProblem(onlinejudge.type.Problem):
+    def __init__(self, *, problem_id: str):
+        self.problem_id = problem_id
+
+    def download_sample_cases(self, *, session: Optional[requests.Session] = None) -> List[onlinejudge.type.TestCase]:
+        session = session or utils.get_default_session()
+        # get
+        resp = utils.request('GET', self.get_url(), session=session)
+        # parse
+        soup = bs4.BeautifulSoup(resp.content.decode(resp.encoding), utils.HTML_PARSER)
+        samples = onlinejudge._implementation.testcase_zipper.SampleZipper()
+        for case, name in self._parse_sample_cases(soup):
+            samples.add(case.encode(), name)
+        return samples.get()
+
+    def _parse_sample_cases(self, soup: bs4.BeautifulSoup) -> List[Tuple[str, str]]:
+        bodyMD = json.loads(soup.find(id='__NEXT_DATA__').get_text())['props']['pageProps']['tasks']['body']
+        pattern = r'#### ([入出]力例 \d ?)\r\n```IOExample\r\n([\s\S]+?)```'
+        cases = re.findall(pattern, bodyMD)
+        for name, case in cases:
+            yield (case, name)
+
+    def get_url(self) -> str:
+        return 'https://algo-method.com/tasks/{}'.format(self.problem_id)
+
+    def get_service(self) -> AlgoMethodService:
+        return AlgoMethodService()
+
+    @classmethod
+    def from_url(cls, url: str) -> Optional['AlgoMethodProblem']:
+        # example: https://algo-method.com/tasks/15
+        result = urllib.parse.urlparse(url)
+        dirname, basename = posixpath.split(utils.normpath(result.path))
+        if result.scheme in ('', 'http', 'https') \
+                and result.netloc == 'algo-method.com' \
+                and dirname == '/tasks' \
+                and basename:
+            return cls(problem_id=basename)
+        return None
+
+
+onlinejudge.dispatch.services += [AlgoMethodService]
+onlinejudge.dispatch.problems += [AlgoMethodProblem]

onlinejudge_api/main.py

@@ -58,6 +58,7 @@ def get_parser() -> argparse.ArgumentParser:
     # get-problem
     epilog = textwrap.dedent("""\
         supported services:
+          Algo-Method
           Aizu Online Judge
           Anarchy Golf
           AtCoder

tests/get_problem_algo_method.py

@@ -0,0 +1,71 @@
+import unittest
+
+from onlinejudge_api.main import main
+
+
+class GetProblemAlgoMethodTask(unittest.TestCase):
+
+    def test_tasks_22(self):
+        url = 'https://algo-method.com/tasks/22'
+        expected = {
+            "status": "ok",
+            "messages": [],
+            "result": {
+                "url": "https://algo-method.com/tasks/22",
+                "tests": [
+                    {
+                        "input": "power\r\n",
+                        "output": "w\r\n"
+                    },
+                    {
+                        "input": "otter\r\n",
+                        "output": "t\r\n"
+                    },
+                ],
+                "context": {}
+            },
+        }
+        actual = main(['get-problem', url], debug=True)
+        self.assertEqual(expected, actual)
+
+    def test_tasks_24(self):
+        url = 'https://algo-method.com/tasks/24'
+        expected = {
+            "status": "ok",
+            "messages": [],
+            "result": {
+                "url": "https://algo-method.com/tasks/24",
+                "tests": [
+                    {
+                        "input": "1 2\r\n",
+                        "output": "3\r\n"
+                    }
+                ],
+                "context": {}
+            },
+        }
+        actual = main(['get-problem', url], debug=True)
+        self.assertEqual(expected, actual)
+
+    def test_tasks_529(self):
+        url = 'https://algo-method.com/tasks/529'
+        expected = {
+            "status": "ok",
+            "messages": [],
+            "result": {
+                "url": "https://algo-method.com/tasks/529",
+                "tests": [
+                    {
+                        "input": "7\r\n0 1 0 0 1 5\r\n",
+                        "output": "0\r\n1\r\n2\r\n1\r\n1\r\n2\r\n3\r\n"
+                    },
+                    {
+                        "input": "7\r\n0 0 0 0 0 0\r\n",
+                        "output": "0\r\n1\r\n1\r\n1\r\n1\r\n1\r\n1\r\n"
+                    }
+                ],
+                "context": {}
+            },
+        }
+        actual = main(['get-problem', url], debug=True)
+        self.assertEqual(expected, actual)