Skip to content

Commit 10c964c

Browse files
feat:Seperate sample input and output
1 parent aafd742 commit 10c964c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

scraper.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
from bs4 import BeautifulSoup
22
import requests
3-
import json
43

54
def scrape(q):
65
web_page = requests.get('https://codeforces.com/problemset/problem/'+q.split('-')[0]+'/'+q.split('-')[1]).text.encode("utf-8")
@@ -19,6 +18,7 @@ def scrape(q):
1918
"content": content,
2019
"input_specs": soup.find('div', class_ = "input-specification").text.replace('Input','').replace("$$$","$$"),
2120
"output_specs": soup.find('div', class_ = "output-specification").text.replace('Output','').replace("$$$","$$"),
22-
"sample_tests": soup.find('div', class_ = "sample-tests").text.replace('Examples', '').replace('Example',''),
21+
"sample_input": soup.find('div', class_ = "sample-tests").text.replace('Examples', '').replace('Example','').split('Output\n')[0].replace('Input\n',''),
22+
"sample_output": soup.find('div', class_ = "sample-tests").text.replace('Examples', '').replace('Example','').split('Output\n')[1],
2323
}
2424
return data

0 commit comments

Comments
 (0)