Started main.py file
parent
4e4f78334a
commit
699097345c
|
@ -0,0 +1,46 @@
|
||||||
|
from tennis import get_time_slots, get_club_address
|
||||||
|
import argparse
|
||||||
|
from configparser import ConfigParser
|
||||||
|
from pathlib import Path
|
||||||
|
from weather import WeatherAPI
|
||||||
|
|
||||||
|
|
||||||
|
def existing_path(path_str):
|
||||||
|
path = Path(path_str)
|
||||||
|
|
||||||
|
if not path.exists() or not path.is_file():
|
||||||
|
raise argparse.ArgumentTypeError("Config file doesn't exist or isn't a file.")
|
||||||
|
|
||||||
|
return path
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
parser = argparse.ArgumentParser()
|
||||||
|
parser.add_argument("-c", "--config-file", help="Path to config file.",
|
||||||
|
default="padel.ini", type=existing_path)
|
||||||
|
parser.add_argument("-d", "--days",
|
||||||
|
help="How many days in advance to look.", default=1,
|
||||||
|
type=int)
|
||||||
|
parser.add_argument("club_id", help="ID of the club to check.", type=int)
|
||||||
|
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
# Read config file
|
||||||
|
config = ConfigParser()
|
||||||
|
# TODO check if config file can be read
|
||||||
|
config.read(args.config_file)
|
||||||
|
|
||||||
|
club_address = get_club_address(args.club_id)
|
||||||
|
|
||||||
|
if club_address is None:
|
||||||
|
print("Couldn't get club address.")
|
||||||
|
return
|
||||||
|
|
||||||
|
weather_api = WeatherAPI(config["DEFAULT"]["weather_api_key"])
|
||||||
|
weather_forecasts = weather_api.get_hourly_conditions(club_address,
|
||||||
|
days=args.days)
|
||||||
|
timeslots = get_time_slots(args.club_id, days=args.days)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
|
@ -61,13 +61,34 @@ def extract_calendar(soup: BeautifulSoup, reservation_date):
|
||||||
return [(col, status, datetime.combine(reservation_date, start), duration) for col, status, start, duration in timeslots]
|
return [(col, status, datetime.combine(reservation_date, start), duration) for col, status, start, duration in timeslots]
|
||||||
|
|
||||||
|
|
||||||
def get_time_slots(club_id: int, date: date = None):
|
def get_time_slots(club_id: int, days=1):
|
||||||
|
dates = [date.today() + timedelta(days=i) for i in range(days)]
|
||||||
params = {"clubId": club_id}
|
params = {"clubId": club_id}
|
||||||
|
|
||||||
if date:
|
output = []
|
||||||
params["planningDay"] = date.strftime("%d-%m-%Y")
|
|
||||||
|
|
||||||
r = requests.get(BASE_URL, params=params)
|
for planning_date in dates:
|
||||||
|
params["planningDay"] = planning_date.strftime("%d-%m-%Y")
|
||||||
|
|
||||||
|
r = requests.get(BASE_URL, params=params)
|
||||||
|
soup = BeautifulSoup(r.content, "html.parser")
|
||||||
|
|
||||||
|
output.extend(extract_calendar(soup, planning_date))
|
||||||
|
|
||||||
|
return output
|
||||||
|
|
||||||
|
def get_club_address(club_id: int):
|
||||||
|
r = requests.get(BASE_URL, params={
|
||||||
|
"clubId": club_id,
|
||||||
|
"tab": "club",
|
||||||
|
})
|
||||||
soup = BeautifulSoup(r.content, "html.parser")
|
soup = BeautifulSoup(r.content, "html.parser")
|
||||||
|
|
||||||
return extract_calendar(soup, date)
|
tab_div = soup.find("div", id="club")
|
||||||
|
info_ul = tab_div.find("ul")
|
||||||
|
|
||||||
|
for li in info_ul.findAll("li"):
|
||||||
|
if li.find("span", text="Adres (hoofdlocatie)"):
|
||||||
|
return li.find("span", class_="list-value").text.strip()
|
||||||
|
|
||||||
|
return None
|
||||||
|
|
|
@ -0,0 +1,26 @@
|
||||||
|
import requests
|
||||||
|
|
||||||
|
|
||||||
|
class WeatherAPI:
|
||||||
|
BASE_URL = "https://api.weatherapi.com/v1"
|
||||||
|
|
||||||
|
def __init__(self, api_key):
|
||||||
|
self._api_key = api_key
|
||||||
|
|
||||||
|
def _get(self, endpoint, params):
|
||||||
|
params["key"] = self._api_key
|
||||||
|
return requests.get(f"{self.BASE_URL}{endpoint}", params=params)
|
||||||
|
|
||||||
|
def get_hourly_conditions(self, query, days=1):
|
||||||
|
r = self._get("/forecast.json", {
|
||||||
|
"q": query,
|
||||||
|
"days": days,
|
||||||
|
"aqi": "no",
|
||||||
|
"alert": "no",
|
||||||
|
})
|
||||||
|
|
||||||
|
if r.status_code != 200:
|
||||||
|
return None
|
||||||
|
|
||||||
|
data = r.json()
|
||||||
|
return sum([obj["hour"] for obj in data["forecast"]["forecastday"]], [])
|
Loading…
Reference in New Issue