from datetime import datetime, time, timedelta, timezone
import socket

import aw_client

if __name__ == "__main__":
    bucket_id = f"aw-watcher-afk_{socket.gethostname()}"

    daystart = datetime.combine(datetime.now().date(), time()).astimezone(timezone.utc)
    dayend = daystart + timedelta(days=1)

    awc = aw_client.ActivityWatchClient("testclient")
    events = awc.get_events(bucket_id, start=daystart, end=dayend)
    events = [e for e in events if e.data["status"] == "not-afk"]
    total_duration = sum((e.duration for e in events), timedelta())
    print(f"Total time spent on computer today: {total_duration}")