Attribute types

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

import asyncio

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


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

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


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

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

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


asyncio.run(main())

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

import asyncio
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(partition_key=True)
    sk = StringAttribute(sort_key=True)
    status = EnumAttribute(Status, default=Status.PENDING)


async def main():
    # Create with enum value
    user = User(pk="USER#ENUM", sk="PROFILE", status=Status.ACTIVE)
    await user.save()
    # Stored as "active" in DynamoDB

    # Load it back - returns the enum
    loaded = await User.get(pk="USER#ENUM", sk="PROFILE")
    print(loaded.status)  # Status.ACTIVE
    print(loaded.status == Status.ACTIVE)  # True

    # Default value works
    user2 = User(pk="USER#ENUM2", sk="PROFILE")
    print(user2.status)  # Status.PENDING


asyncio.run(main())

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

import asyncio
from datetime import datetime, timezone

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


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

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


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

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


asyncio.run(main())

"""StringSetAttribute and NumberSetAttribute examples."""

import asyncio

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


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

    pk = StringAttribute(partition_key=True)
    sk = StringAttribute(sort_key=True)
    tags = StringSetAttribute()
    scores = NumberSetAttribute()


async def main():
    # Create with sets
    user = User(
        pk="USER#SET",
        sk="PROFILE",
        tags={"admin", "verified", "premium"},
        scores={100, 95, 88},
    )
    await user.save()

    # Load it back - returns Python sets
    loaded = await User.get(pk="USER#SET", sk="PROFILE")
    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#SET2", sk="PROFILE", tags={"a", "a", "b"})
    print(user2.tags)  # {'a', 'b'}


asyncio.run(main())

"""Basic S3 upload example."""

import asyncio

from pydynox import Model, ModelConfig
from pydynox.attributes import S3Attribute, S3File, StringAttribute


# Define model with S3Attribute
class Document(Model):
    model_config = ModelConfig(table="documents")

    pk = StringAttribute(partition_key=True)
    name = StringAttribute()
    content = S3Attribute(bucket="my-bucket", prefix="docs/")


async def main():
    # Upload from bytes
    doc = Document(pk="DOC#S3", name="report.pdf")
    doc.content = S3File(b"PDF content here", name="report.pdf", content_type="application/pdf")
    await doc.save()

    print(f"Uploaded to: s3://{doc.content.bucket}/{doc.content.key}")
    print(f"Size: {doc.content.size} bytes")
    print(f"ETag: {doc.content.etag}")


asyncio.run(main())