bamboo-timesheet.py

#!/usr/bin/env python3

import argparse
import requests
from datetime import datetime
from calendar import monthrange

BAMBOOHR_URL = "https://worldsensing.bamboohr.com/timesheet/hour/entries"
EMPLOYEE_ID = '<ID>'
PROJECT_ID = 6
TASK_ID = 9
DAILY_HOURS = 8
COUNTRY_ISO = 'ES'


parser = argparse.ArgumentParser()
parser.add_argument("--csrf", required=True)
parser.add_argument("--cookie", required=True)
args = parser.parse_args()

now = datetime.now()
days = monthrange(now.year, now.month)[1]
headers = {
    "Accept": "application/json, text/plain, */*",
    "Accept-Language": "en-US,en;q=0.7,es-ES;q=0.3",
    "Content-Type": "application/json;charset=utf-8",
    "X-CSRF-TOKEN": args.csrf,
    "Origin": "https://worldsensing.bamboohr.com",
    "Referer": f"https://worldsensing.bamboohr.com/employees/timesheet/?id={EMPLOYEE_ID}",
    "Cookie": args.cookie,
    "Sec-Fetch-Dest": "empty",
    "Sec-Fetch-Mode": "cors",
    "Sec-Fetch-Site": "same-origin",
}


def get_holidays(year):
    url = f"https://date.nager.at/api/v3/PublicHolidays/{year}/{COUNTRY_ISO}"
    try:
        r = requests.get(url, timeout=10)
        r.raise_for_status()
        data = r.json()

        return {item["date"] for item in data}
    except requests.RequestException:
        return set()


holidays = get_holidays(now.year)


for day in range(1, days + 1):
    current_date = datetime(now.year, now.month, day)
    date_str = current_date.strftime("%Y-%m-%d")

    if current_date.weekday() in (5, 6):
        print(f"{date_str} -> Weekend (skipped)")
        continue

    if date_str in holidays:
        print(f"{date_str} -> Public holiday in {COUNTRY_ISO} (skipped)")
        continue

    date_str = f"{now.year}-{now.month:02d}-{day:02d}"
    data = {
        "hours": [
            {
                "id": None,
                "dailyEntryId": 1,
                "employeeId": EMPLOYEE_ID,
                "date": date_str,
                "hours": DAILY_HOURS,
                "note": "",
                "projectId": PROJECT_ID,
                "taskId": TASK_ID
            }
        ]
    }
    try:
        r = requests.post(BAMBOOHR_URL, headers=headers, json=data)
        r.raise_for_status()

        print(f"{date_str} -> OK")
    except requests.RequestException as e:
        print(f"{date_str} -> ERROR: {e}")