time_handling.py

import sys
from datetime import datetime
from zoneinfo import ZoneInfo  # Built-in since Python 3.9
import logging

from settings import LOG_LEVEL, LOG_DATE_FORMAT, LOG_FORMAT, DATE_FORMAT, TIMESTAMP_FILE

log = logging.getLogger(__name__)
logging.basicConfig(stream=sys.stdout, level=LOG_LEVEL, format=LOG_FORMAT, datefmt=LOG_DATE_FORMAT)

# Define Stockholm timezone
STOCKHOLM_TZ = ZoneInfo("Europe/Stockholm")

def elapsed_time(start):
    return datetime.now(STOCKHOLM_TZ) - start

def timestamp_now():
    return datetime.now(STOCKHOLM_TZ).strftime(DATE_FORMAT)

def write_timestamp(timestamp=None):
    if not timestamp:
        timestamp = timestamp_now()
    with open(file=TIMESTAMP_FILE, mode='w') as f:
        f.write(timestamp)
    log.info(f"New timestamp written: {timestamp}")
    return timestamp

def read_timestamp():
    with open(file=TIMESTAMP_FILE, mode='r') as f:
        timestamp = f.read()
    return timestamp.strip('\n')