Started main.py file

main
Jef Roosens 2021-08-11 13:55:39 +02:00
parent 4e4f78334a
commit 699097345c
Signed by: Jef Roosens
GPG Key ID: 955C0660072F691F
3 changed files with 98 additions and 5 deletions

View File

@ -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()

View File

@ -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]
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}
if date:
params["planningDay"] = date.strftime("%d-%m-%Y")
output = []
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")
return extract_calendar(soup, date)
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")
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

26
padel/weather.py 100644
View File

@ -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"]], [])