Attribute types

"""JSONAttribute example - store dict/list as JSON string."""

from pydynox import Model, ModelConfig
from pydynox.attributes import JSONAttribute, StringAttribute


class Config(Model):
    model_config = ModelConfig(table="configs")

    pk = StringAttribute(hash_key=True)
    settings = JSONAttribute()


# Save a dict
config = Config(
    pk="CFG#1",
    settings={"theme": "dark", "notifications": True, "max_items": 50},
)
config.save()
# Stored as '{"theme": "dark", "notifications": true, "max_items": 50}'

# Load it back
loaded = Config.get(pk="CFG#1")
print(loaded.settings["theme"])  # "dark"

# Works with lists too
config2 = Config(pk="CFG#2", settings=["item1", "item2", "item3"])
config2.save()

"""EnumAttribute example - store Python enum as string."""

from enum import Enum

from pydynox import Model, ModelConfig
from pydynox.attributes import EnumAttribute, StringAttribute


class Status(Enum):
    PENDING = "pending"
    ACTIVE = "active"
    INACTIVE = "inactive"


class User(Model):
    model_config = ModelConfig(table="users")

    pk = StringAttribute(hash_key=True)
    status = EnumAttribute(Status, default=Status.PENDING)


# Create with enum value
user = User(pk="USER#1", status=Status.ACTIVE)
user.save()
# Stored as "active" in DynamoDB

# Load it back - returns the enum
loaded = User.get(pk="USER#1")
print(loaded.status)  # Status.ACTIVE
print(loaded.status == Status.ACTIVE)  # True

# Default value works
user2 = User(pk="USER#2")
print(user2.status)  # Status.PENDING

"""DatetimeAttribute example - store datetime as ISO string."""

from datetime import datetime, timezone

from pydynox import Model, ModelConfig
from pydynox.attributes import DatetimeAttribute, StringAttribute
from pydynox.hooks import before_save


class Event(Model):
    model_config = ModelConfig(table="events")

    pk = StringAttribute(hash_key=True)
    created_at = DatetimeAttribute()


# Save with datetime
event = Event(pk="EVT#1", created_at=datetime.now(timezone.utc))
event.save()
# Stored as "2024-01-15T10:30:00+00:00"

# Load it back - returns datetime object
loaded = Event.get(pk="EVT#1")
print(loaded.created_at)  # datetime object
print(loaded.created_at.year)  # 2024


# Auto-set timestamps with hooks
class Article(Model):
    model_config = ModelConfig(table="articles")

    pk = StringAttribute(hash_key=True)
    created_at = DatetimeAttribute(null=True)
    updated_at = DatetimeAttribute(null=True)

    @before_save
    def set_timestamps(self):
        now = datetime.now(timezone.utc)
        if self.created_at is None:
            self.created_at = now
        self.updated_at = now


article = Article(pk="ART#1")
article.save()
print(article.created_at)  # Auto-set on first save
print(article.updated_at)  # Updated on every save

"""StringSetAttribute and NumberSetAttribute examples."""

from pydynox import Model, ModelConfig
from pydynox.attributes import (
    NumberSetAttribute,
    StringAttribute,
    StringSetAttribute,
)


class User(Model):
    model_config = ModelConfig(table="users")

    pk = StringAttribute(hash_key=True)
    tags = StringSetAttribute()
    scores = NumberSetAttribute()


# Create with sets
user = User(
    pk="USER#1",
    tags={"admin", "verified", "premium"},
    scores={100, 95, 88},
)
user.save()

# Load it back - returns Python sets
loaded = User.get(pk="USER#1")
print(loaded.tags)  # {'admin', 'verified', 'premium'}
print(loaded.scores)  # {100, 95, 88}

# Check membership
print("admin" in loaded.tags)  # True
print(100 in loaded.scores)  # True

# Sets don't allow duplicates
user2 = User(pk="USER#2", tags={"a", "a", "b"})
print(user2.tags)  # {'a', 'b'}