scripts/utils.py

import pandas as pd
from pathlib import Path
from datetime import datetime, timedelta

SCRIPTS_DIR = Path(__file__).parent
ROOT_DIR = SCRIPTS_DIR.parent
DATA_DIR = ROOT_DIR / "data"
TMP_DIR = ROOT_DIR / "tmp"


def get_traders_family(row: pd.DataFrame) -> str:
    if row.staking == "non_agent":
        return "non_agent"
    elif row.market_creator == "pearl":
        return "pearl_agent"
    # quickstart
    return "quickstart_agent"


def get_current_week():
    current_date = datetime.now()

    # Get the start and end dates of the current week (starting on Sunday)
    current_week_start = current_date - timedelta(days=current_date.weekday() + 1)
    return current_week_start.strftime("%b-%d")


def get_next_week():
    current_date = datetime.now()
    next_week_start = current_date + timedelta(days=7 - (current_date.weekday() + 1))
    return next_week_start.strftime("%b-%d")