from time import time

from mypy_boto3_dynamodb.service_resource import DynamoDBServiceResource


class DynamoDBClient:
    def __init__(self, db: DynamoDBServiceResource, tablename: str) -> None:
        self.table = db.Table(tablename)

    def get_fan_state(self, fan_key: str) -> str:
        response = self.table.get_item(Key={"key": fan_key})
        item = response.get("Item")
        if item:
            return item.get("state")  # type: ignore
        return "greetings"

    def set_fan_state(self, fan_key: str, state: str) -> None:
        updated_at = int(time())
        self.table.put_item(
            Item={
                "key": fan_key,
                "state": state,
                "updated_at": updated_at,
            },
        )