Compare commits

...

2 Commits

Author SHA1 Message Date
Jef Roosens dafc04a109
Small changes 2021-08-11 17:43:49 +02:00
Jef Roosens 56ff5d4cc3
Added better cli output 2021-08-11 17:27:11 +02:00
3 changed files with 16 additions and 9 deletions

View File

@ -42,7 +42,17 @@ async def main():
args.club_id, config["DEFAULT"]["weather_api_key"], args.days
)
print(res)
for field_name, start_time, duration, weather_names in res:
# Very satisfying, create a set & sort it using the original list's index
pruned_weather_names = sorted(set(weather_names),
key=lambda x: weather_names.index(x))
print("{} ({}, {}m): {}".format(
field_name,
start_time.strftime("%Y-%m-%d %H:%M"),
int(duration.total_seconds() // 60),
" -> ".join(pruned_weather_names)
))
if __name__ == "__main__":

View File

@ -1,6 +1,6 @@
from tennis import get_time_slots, get_club_address
from weather import WeatherAPI
from datetime import timedelta
from datetime import timedelta, datetime
import asyncio as aio
@ -29,11 +29,10 @@ async def get_decent_timeslots(club_id, weather_api_key, days=1):
weather_forecasts = {
date_obj: weather_str for date_obj, weather_str, _ in weather_forecasts
}
print(weather_forecasts)
# Filter out non-free timeslots
timeslots = list(filter(lambda x: x[1] == 0, timeslots))
print(timeslots)
# Filter out non-free timeslots or timeslots in the past
now = datetime.now()
timeslots = list(filter(lambda x: x[1] == 0 and x[2] > now, timeslots))
output = []

View File

@ -13,9 +13,7 @@ def extract_timeslots(tbody, column_headers):
for tr in tbody.findAll("tr"):
# Determine time for row
time_str = tr.find("th").text
hour, minute = map(int, time_str.split(":"))
start_time = time(hour=hour, minute=minute)
start_time = time.fromisoformat(tr.find("th").text)
# Iterate over each column
for td in tr.findAll("td"):