87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
from bs4 import BeautifulSoup
|
|
import requests, time
|
|
|
|
class LiveStats:
|
|
def __init__(self):
|
|
self.refresh_live()
|
|
self.get_races()
|
|
|
|
def refresh_live(self):
|
|
req = requests.get("https://www.procyclingstats.com/")
|
|
html = req.text
|
|
soup = BeautifulSoup(html, "html.parser")
|
|
|
|
all = soup.find_all(attrs={"class":"hp3-livestats"})
|
|
live = all[0]
|
|
for tag in live.find_all(attrs={"class":"inverse"}):
|
|
tag.decompose()
|
|
|
|
self.live = live
|
|
|
|
def get_races(self):
|
|
races_raw = self.live.find_all("li")
|
|
self.races = []
|
|
for race in races_raw:
|
|
this_race = Race(race)
|
|
self.races.append(this_race)
|
|
|
|
def print_races(self):
|
|
for race in self.races:
|
|
race.print_stats()
|
|
|
|
class Race:
|
|
def __init__(self, raw):
|
|
self.raw = raw
|
|
self.refresh_info()
|
|
|
|
def refresh_info(self):
|
|
title_r = self.raw.find(attrs={"class":"title"})
|
|
status_r = self.raw.find(attrs={"class":"status"})
|
|
togo_r = self.raw.find(attrs={"class":"togo"})
|
|
situation_r = self.raw.find(attrs={"class":"situ_txt"})
|
|
url_r = self.raw.find("a", href=True)
|
|
|
|
self.title = self.remove_tags(title_r)
|
|
self.status = self.remove_tags(status_r)
|
|
self.togo = self.remove_tags(togo_r)
|
|
self.situation = self.remove_tags(situation_r)
|
|
self.url = self.remove_tags(url_r["href"])
|
|
|
|
def print_stats(self):
|
|
print(self.title)
|
|
print(self.status)
|
|
if self.togo != "None":
|
|
print(self.togo, "to go")
|
|
print(self.situation)
|
|
print("===============")
|
|
|
|
def print_raw(self):
|
|
print(self.raw)
|
|
print("")
|
|
print(self.url)
|
|
|
|
def get_timeline(self):
|
|
if self.url != "None":
|
|
full_url = "https://www.procyclingstats.com/" + self.url
|
|
|
|
req = requests.get(full_url)
|
|
html = req.text
|
|
soup = BeautifulSoup(html, "html.parser")
|
|
|
|
all = soup.find_all(attrs={"class":"timeline3cont"})
|
|
live = all[0]
|
|
timeline = live.find_all("li")
|
|
self.timeline = []
|
|
for item in timeline:
|
|
stat = item.find(attrs={"class":"stat"})
|
|
if stat:
|
|
stat_content = stat.find(attrs={"class":"textCont"})
|
|
self.timeline.append(self.remove_tags(stat_content))
|
|
|
|
|
|
def remove_tags(self, text):
|
|
text = str(text)
|
|
text_soup = BeautifulSoup(text, "html.parser")
|
|
for data in text_soup(["style", "script"]):
|
|
data.decompose()
|
|
return " ".join(text_soup.stripped_strings) |