trackonio.py

#!/usr/bin/env python3

import json
import os
import re
import sys
import uuid
import functools
from datetime import datetime, timedelta
from random import randint
import time

import requests

MAIN_URL = "https://"+str(os.environ.get('PERSONIO_HOST',"personio.personio.de"))
EMAIL = str(os.environ.get('PERSONIO_USERNAME'))
PASSWORD = str(os.environ.get('PERSONIO_PASSWORD'))
COOKIE = str(os.environ.get('PERSONIO_COOKIE'))
CALENDAR_ID = str(os.environ.get('CALENDAR_ID', '59097'))
STARTING_HOUR = str(os.environ.get('WORK_START_TIME', '09'))
BREAK_HOUR = str(os.environ.get('BREAK_START_TIME', '13'))
WORKING_HOURS = int(os.environ.get('WORK_DURATION', 8))
WORK_SATURDAY = bool(os.environ.get('WORK_SATURDAY', False))
WORK_SUNDAY = bool(os.environ.get('WORK_SUNDAY', False))
BREAK_TIME_MINUTES = int(os.environ.get('BREAK_DURATION', 60))
RANDOM_TIMES_DELTA = int(os.environ.get('RANDOM_TIMES_DELTA', 0))
RANDOM_DURATIONS_DELTA = int(os.environ.get('RANDOM_TIMES_DELTA', 0))

LOGIN_URL = f'{MAIN_URL}/login/index'
ATTENDANCE_URL = f'{MAIN_URL}/api/v1/attendances/periods'
HOLIDAYS_URL = f'{MAIN_URL}/api/v1/holidays?holiday_calendar_ids[]=' 
ABSENCES_URL = f'{MAIN_URL}/api/v1/employees'

def check_date(dateInput):
    return re.fullmatch(r"\A([\d]{4})-([\d]{2})-([\d]{2})", dateInput)


def generate_attendance(
    date, startingHour, breakHour, workingHours, breakDuration, employeeId, randomTimes, randomDurations):
    # First Record
    startTime = datetime.strptime(f"{date} {startingHour}", "%Y-%m-%d %H")
    breakTime = datetime.strptime(f"{date} {breakHour}", "%Y-%m-%d %H")
    workingDuration = (workingHours) * 60
    
    # Randomize input times and durations
    if randomTimes:
        startTime += timedelta(minutes=randint(-randomTimes, randomTimes))
        breakTime += timedelta(minutes=randint(-randomTimes, randomTimes))
    if randomDurations:
        workingDuration = randint(workingDuration - randomDurations, workingDuration + randomDurations)
        breakDuration = randint(breakDuration - randomDurations, breakDuration + randomDurations)
        
    if breakDuration == 0:
        end_time = startTime + timedelta(minutes=workingDuration)
        return [
            {
                "id": str(uuid.uuid1()),
                "start": startTime.strftime("%Y-%m-%dT%H:%M:%SZ"),
                "end": end_time.strftime("%Y-%m-%dT%H:%M:%SZ"),
                "comment": "",
                "project_id": None,
                "employee_id": employeeId,
                "activity_id": None,
            },
        ]
    else:
        # Second Record
        startTime2 = breakTime + timedelta(minutes=breakDuration)

        workingDuration2 = int(
            workingDuration - (breakTime - startTime).total_seconds() / 60
        )
        end_time = startTime2 + timedelta(minutes=workingDuration2)

        return [
            {
                "id": str(uuid.uuid1()),
                "start": startTime.strftime("%Y-%m-%dT%H:%M:%SZ"),
                "end": breakTime.strftime("%Y-%m-%dT%H:%M:%SZ"),
                "comment": "",
                "project_id": None,
                "employee_id": employeeId,
                "activity_id": None,
            },
            {
                "id": str(uuid.uuid1()),
                "start": startTime2.strftime("%Y-%m-%dT%H:%M:%SZ"),
                "end": end_time.strftime("%Y-%m-%dT%H:%M:%SZ"),
                "comment": "",
                "project_id": None,
                "employee_id": employeeId,
                "activity_id": None,
            },
        ]


if __name__ == "__main__":
    if len(sys.argv) == 1 or sys.argv[1] == "--help" or not check_date(sys.argv[1]):
        print(
            f"""
            Error. No argument or wrong date format\n\n
            Usage: {__file__} <date> [-f/--fast]\n
            Note: Date format yyyy-mm-dd \n
            """
        )
        exit()

    attendanceDate = sys.argv[1]
    fastExec = "--slow"
    if len(sys.argv) == 3:
        fastExec = sys.argv[2]
    
    # Create request session
    session = requests.Session()

    # Login into Personio
    if COOKIE != 'None':
        if os.path.exists('.session'):
            with open('.session', 'r') as file:
                session.cookies.set('personio_session', file.read().strip())
        else:
            session.cookies.set('personio_session', COOKIE)
        print("Authenticating with following Cookie: " + str(session.cookies.get('personio_session')))
        response = session.get(MAIN_URL)
        # Waiting 1 minute to make sure the returned cookie will be valid in next iteration
        if fastExec != "-f" and fastExec != "--fast":
            time.sleep(60)
    elif EMAIL != 'None' and PASSWORD != 'None':
        print("Authenticating with Username/Password: " + EMAIL)
        response = session.post(
            LOGIN_URL,
            headers={"Content-Type": "application/x-www-form-urlencoded"},
            data={"email": EMAIL, "password": PASSWORD},
        )
    if 'response' in locals():
        if 'XSRF-TOKEN' in response.cookies:
            XSRF_TOKEN=response.cookies['XSRF-TOKEN']
            PROFILE_ID=response.text.split("EMPLOYEE={id:")[1].split("}")[0]
        else:
            print("Failed to login with provided credentials to "+MAIN_URL)
            exit()
    else:
        print("No auth method provided. Please RTFM!")
        exit()
    
    # Check User Holiday
    response = session.get(
        f'{HOLIDAYS_URL}{CALENDAR_ID}&start_date={attendanceDate}&end_date={attendanceDate}'
    )
    isHoliday = len(json.loads(response.text)['data'])

    # Check User Abscense
    response = session.get(
        f'{ABSENCES_URL}/{PROFILE_ID}/absences/types'
    )
    absenceTypes = ','.join([str(a['id']) for a in json.loads(response.text)['data'] if str(a['id']) != "2352878"]) # Removing Personio's On-Call absence, if existing
    response = session.get(
        f'{ABSENCES_URL}/{PROFILE_ID}/absences/periods?filter[startDate]={attendanceDate}&filter[endDate]={attendanceDate}&filter[absenceTypes]={absenceTypes}'
    )
    isAbsence = len(json.loads(response.text)['data'])
    
    # Check Working Day or Weekend
    weekday = datetime.strptime(attendanceDate, "%Y-%m-%d").weekday()
    isWeekendLeave = (not WORK_SATURDAY and weekday == 5) or (not WORK_SUNDAY and weekday == 6)

    if isHoliday or isAbsence or isWeekendLeave:
        if isHoliday: excuse = " You're on Leave"
        if isAbsence: excuse = " It's Holiday"
        if isWeekendLeave: excuse = "It's weekend"
        message = 'Not working day: ' + excuse
        print(message)
        exit()

    # Log the attendance
    response = session.post(
        ATTENDANCE_URL,
        headers={'x-csrf-token':XSRF_TOKEN},
        json=generate_attendance(
            attendanceDate,
            STARTING_HOUR,
            BREAK_HOUR,
            WORKING_HOURS,
            BREAK_TIME_MINUTES,
            PROFILE_ID,
            RANDOM_TIMES_DELTA,
            RANDOM_DURATIONS_DELTA
        ),
    )

    data = json.loads(response.text)
    personio_cookie = response.cookies.get('personio_session')
    if personio_cookie:
        with open('.session', 'w') as file:
            file.write(personio_cookie)
            print("Refreshed session cookie stored successfully")
    try:
        message = f"Error: {attendanceDate} - {data['error']['message']}"
    except KeyError:
        message = f"Success: attendance on {attendanceDate} registered!"

    print(message)