Added handling incorrect values in get_solution

VadVergasov · VadVergasov · commit 5c1963bed123 · 2020-03-08T23:12:46.000+03:00
diff --git a/.gitignore b/.gitignore
@@ -2,5 +2,6 @@ dist/
 build/
 CodeforcesApiPy.egg-info
 tests/conf.py
+tests/__pycache__/
 .pylintrc
 codeforces_api/__pycache__/
diff --git a/codeforces_api/parse_methods.py b/codeforces_api/parse_methods.py
@@ -18,4 +18,6 @@ def get_solution(self, contest_id, submit_id):
         )
         tree2 = html.fromstring(solutionPage.text)
         code = tree2.xpath('//*[@id="pageContent"]/div[3]/pre/text()')
+        if len(code) == 0:
+            raise ValueError("Incorrect contest_id or sumbit_id")
         return code[0].replace("\r", "")

Original file line number	Diff line number	Diff line change
`@@ -18,4 +18,6 @@ def get_solution(self, contest_id, submit_id):`
`18`	`18`	`)`
`19`	`19`	`tree2 = html.fromstring(solutionPage.text)`
`20`	`20`	`code = tree2.xpath('//*[@id="pageContent"]/div[3]/pre/text()')`
	`21`	`+ if len(code) == 0:`
	`22`	`+ raise ValueError("Incorrect contest_id or sumbit_id")`
`21`	`23`	`return code[0].replace("\r", "")`