seed.py

from lib.db import DB
from lib import utils
from collections import deque
import random
from datetime import timedelta, datetime
from lib.camera import Conditions
from lib.ml import WeatherConditions, SurfRating


def daterange(date1, date2):
    for n in range(int((date2 - date1).days) + 1):
        yield date1 + timedelta(n)


def seed(spot_id, start: datetime, end: datetime):
    db = DB.get_db()
    db.setup()
    surf_ratings = [c.name for c in SurfRating]
    # for range over the 3rd to 10th oct.
    # which is Monday-Sunday
    conditions = deque(surf_ratings)

    weather_conditions = [c.name for c in WeatherConditions]
    # 1 = FLAT: Unsurfable or flat conditions. No surf.
    # 2 = VERY POOR: Due to lack of surf, very poor wave shape for surfing, bad surf due to other conditions like wind, tides, or very stormy surf.
    # 3 = POOR: Poor surf with some (30%) FAIR waves to ride.
    # 4 = POOR to FAIR: Generally poor surf many (50%) FAIR waves to ride.
    # 5 = FAIR: Very average surf with most (70%) waves rideable.
    # 6 = FAIR to GOOD: Fair surf with some (30%) GOOD waves.
    # 7 = GOOD: Generally fair surf with many (50%) GOOD waves.
    # 8 = VERY GOOD: Generally good surf with most (70%) GOOD waves.
    # 9 = GOOD to EPIC: Very good surf with many (50%) EPIC waves.
    # 10 = EPIC: Incredible surf with most (70%) waves being EPIC to ride and generally some of the best surf all year.

    conditions_max = {
        "FLAT": 1,
        "VERY_POOR": 3,
        "POOR": 5,
        "POOR_TO_FAIR": 7,
        "FAIR": 10,
        "FAIR_TO_GOOD": 15,
        "GOOD": 15,
        "VERY_GOOD": 20,
        "GOOD_TO_EPIC": 25,
        "EPIC": 30,
    }
    rows = []

    for dt in daterange(start, end):
        r = random.randint(0, len(conditions))
        conditions.rotate(r)
        random.seed(dt.timestamp())
        r = range(0, 24)
        total_hours = len(r)
        factor = total_hours / len(conditions)
        for h in r:
            if h % factor == 0:
                conditions.rotate(1)

            # for each hour in the day.
            if 6 > h < 18:
                weather_condition = random.choice(
                    [w for w in weather_conditions if "NIGHT" not in w]
                )
            else:
                weather_condition = random.choice(
                    [w for w in weather_conditions if "NIGHT" in w]
                )

            crowd_count = random.randint(0, conditions_max[conditions[0]])
            temp = random.randint(0, 30)
            wind_speed = random.randint(0, 40)
            wave_height_min = random.randint(0, 3)
            wind_gust = random.randint(0, 10)

            c = Conditions(
                wind_direction=149.1,
                wind_speed=wind_speed,
                wind_gust=wind_gust,
                water_temp_max=temp - 2,
                water_temp_min=temp - 6,
                weather_temp=temp,
                weather_condition=weather_condition,
                wave_height_max=wave_height_min + 2,
                wave_height_min=wave_height_min,
                surf_rating=conditions[0],
                tide_height=0.4,
            )

            row = db.insert(
                crowd_count,
                c,
                spot_id,
                dt.replace(hour=h).strftime(utils.DATETIME_FORMAT),
                2,
            )

            rows.append(row)

    return rows