| from bs4 import BeautifulSoup |
| import requests |
| import numpy as np |
| import pdfkit |
| from datetime import datetime |
|
|
| def gen_link(): |
| if(np.random.choice([True, False])): |
| |
| np.random.seed() |
| year=np.random.randint(2015,2023) |
| AB=np.random.choice(['A', 'B']) |
| |
| mu,sigma=18, 5 |
| s=np.random.normal(mu,sigma,1000) |
| s = np.round(s) |
| s=s[s>=10] |
| s=s[s<=25] |
| q=int(np.random.choice(s)) |
| link='https://artofproblemsolving.com/wiki/index.php/{}_AMC_12{}_Problems/Problem_{}'.format(year, AB, q) |
| else: |
| |
| np.random.seed() |
| year=np.random.randint(2005,2023) |
| I=np.random.choice(['I', 'II']) |
| mu,sigma=6, 4 |
| s=np.random.normal(mu,sigma,1000) |
| s = np.round(s) |
| s=s[s>=1] |
| s=s[s<=15] |
| q=int(np.random.choice(s)) |
| link='https://artofproblemsolving.com/wiki/index.php/{}_AIME_{}_Problems/Problem_{}'.format(year,I, q) |
| return link |
|
|
| def convert_to_renderable_html(text): |
| text = text.replace('//latex.artofproblemsolving.com', 'https://latex.artofproblemsolving.com') |
| return text |
|
|
| def get_problem(url): |
| headers = { |
| "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3" |
| } |
| response = requests.get(url, headers=headers) |
| soup = BeautifulSoup(response.text, 'html.parser') |
| problem_headline = soup.find('span', {'class': 'mw-headline', 'id': 'Problem'}) |
| |
| if problem_headline: |
| problem_content = [] |
| for sibling in problem_headline.parent.find_next_siblings(): |
| |
| if sibling.name == 'h2': |
| break |
| |
| elif sibling.name == 'p': |
| problem_content.append(convert_to_renderable_html(str(sibling))) |
|
|
| |
| problem_html = " ".join(problem_content) |
| return problem_html |
| else: |
| print("No problem found") |
|
|
| def gen_html(num): |
| all_q=str() |
| num_tried=0 |
| num_succ=0 |
| while(True): |
| try: |
| link=gen_link() |
| print(link) |
| hype = '<a href="{}" target="_blank">to link</a>'.format(link) |
| qhtml=get_problem(link) |
| all_q+=(hype+qhtml) |
| num_succ+=1 |
| except: |
| pass |
| num_tried+=1 |
| if num_succ>=num or num_tried>20: |
| break |
|
|
| all_q=f''' |
| <html> |
| <head> |
| </head> |
| <body> |
| {all_q} |
| </body> |
| </html> |
| ''' |
| |
| return all_q |
