oaas-sdk2-py 0.2.2

No project description provided

OaaS-SDK2

This library helps you develop a runtime that can be run in a Object as a Service (OaaS) serverless. For more information on the OaaS model, visit https://github.com/hpcclab/OaaS.

Table of Contents

• Setup
• Installation
• Features
• Examples
  • Basic Usage
  • Interacting with Objects
  • Using the Mock Framework for Tests
• Reference
  • Oparaca
  • ClsMetadata
  • BaseObject
  • MockOaas
• Run on OaaS
  • Prerequisites
• Build the project
  • Prerequisites
  • Build

Installation

To install oaas-sdk2-py, you can use pip:

pip install oaas-sdk2-py

Or, if you are using uv:

# For adding/installing packages with uv, the command is 'uv pip install'
uv add oaas-sdk2-py

Features

• Define Classes and Objects: Easily define classes and create persistent objects.
• Remote Procedure Calls (RPC): Invoke methods on objects remotely.
• Data Persistence: Object data is persisted and can be retrieved.
• Asynchronous Support: Built with async/await for non-blocking operations.
• Mocking Framework: Includes a mocking utility for testing your OaaS applications without needing a live environment.
• Typed Interactions: Leverages Pydantic for data validation and serialization.
• Rust-Powered Core: High-performance core components written in Rust for speed and efficiency.

Examples

Note (Sync/Async API): The following examples demonstrate synchronous operations. To use async/await features, initialize Oparaca with oaas = Oparaca(async_mode=True). You would then use methods like get_data_async, set_data_async, commit_async, and define your object methods with async def.

Note (Remote object): When an object is created without the local=True flag (the default behavior for mock_oaas.create_object unless specified, and standard for objects managed by a live OaaS environment), method calls decorated with @your_cls_meta.func() on that object instance will result in a Remote Procedure Call (RPC) to the OaaS platform.

Basic Usage

First, define your class and its methods using the OaaS SDK decorators.

# In your_module.py
from pydantic import BaseModel
from oaas_sdk2_py import Oparaca, BaseObject

# Initialize Oparaca (default is synchronous mode)
oaas = Oparaca()
# For asynchronous operations, use: oaas = Oparaca(async_mode=True)

# Define a class metadata
sample_cls_meta = oaas.new_cls("MySampleClass")

class Msg(BaseModel):
    content: str

class Result(BaseModel):
    status: str
    message: str

@sample_cls_meta
class SampleObj(BaseObject):
    def get_intro(self) -> str:
        raw = self.get_data(0) # Key 0 for intro data
        return raw.decode("utf-8") if raw is not None else "No intro set."

    def set_intro(self, data: str):
        self.set_data(0, data.encode("utf-8"))

    @sample_cls_meta.func()
    def greet(self) -> str:
        intro = self.get_intro()
        return f"Hello from SampleObj! Intro: {intro}"

    @sample_cls_meta.func("process_message")
    def process(self, msg: Msg) -> Result:
        # Process the message
        processed_message = f"Processed: {msg.content}"
        self.set_intro(processed_message) # Example of updating object state
        self.commit()
        return Result(status="success", message=processed_message)

Interacting with Objects

from oaas_sdk2_py import Oparaca
# Assuming your_module defines oaas, sample_cls_meta, SampleObj, Msg
from your_module import oaas, sample_cls_meta, SampleObj, Msg 

def main(): 

    # For local testing, you can use the mock
    mock_oaas = oaas.mock()

    # Create an object
    # obj_id can be any unique identifier, e.g., 1
    my_object: SampleObj = mock_oaas.create_object(sample_cls_meta, 1)

    # Set initial data
    my_object.set_intro("My first OaaS object!")
    my_object.commit() # Persist changes (synchronous)

    # Invoke a simple RPC method
    greeting = my_object.greet()
    print(greeting)

    # Invoke an RPC method with input and output
    response = my_object.process(Msg(content="Important data"))
    print(f"Processing Response: {response.status} - {response.message}")

    # Verify data was updated
    updated_greeting = my_object.greet()
    print(updated_greeting)

    # Load an existing object
    loaded_object: SampleObj = mock_oaas.load_object(sample_cls_meta, 1)
    intro = loaded_object.get_intro()
    print(f"Loaded object intro: {intro}")

if __name__ == "__main__":
    main()

Using the Mock Framework for Tests

The SDK provides a oaas.mock() utility that allows you to test your object logic without connecting to a live OaaS environment. This is particularly useful for unit and integration tests.

# In your tests/test_my_sample_class.py
import unittest
# import asyncio # Not needed for synchronous example
from oaas_sdk2_py import Oparaca
# Assuming your_module defines oaas, sample_cls_meta, SampleObj, Msg
from your_module import oaas, sample_cls_meta, SampleObj, Msg

class TestMySampleClass(unittest.TestCase): # Changed from IsolatedAsyncioTestCase

    def test_greeting_with_mock(self): # Changed from async def
        mock_oaas = oaas.mock()
        obj: SampleObj = mock_oaas.create_object(sample_cls_meta, 1)
        
        obj.set_intro("Mocked Intro")
        obj.commit() # In mock, this updates the in-memory store (synchronous)
        
        result = obj.greet()
        self.assertEqual(result, "Hello from SampleObj! Intro: Mocked Intro")

    def test_process_message_with_mock(self): # Changed from async def
        mock_oaas = oaas.mock()
        obj: SampleObj = mock_oaas.create_object(sample_cls_meta, 2)
        
        response = obj.process(Msg(content="Test Message"))
        self.assertEqual(response.status, "success")
        self.assertEqual(response.message, "Processed: Test Message")
        
        # Verify state change
        intro = obj.get_intro()
        self.assertEqual(intro, "Processed: Test Message")

Refer to tests/test_mock.py and tests/sample_cls.py for more detailed examples of synchronous and asynchronous object definitions and mock usage.

Reference

Oparaca

The main entry point for interacting with the OaaS SDK.

Initialization:

• Oparaca(async_mode: bool = False)
  • Initializes the Oparaca client.
  • async_mode: If True, enables asynchronous operations for BaseObject methods (e.g., get_data_async, commit_async) and MockOaas methods (e.g., create_object_async). Defaults to False (synchronous mode).

Methods:

• new_cls(name: str) -> ClsMetadata
  • Creates metadata for a new class that will be registered with OaaS.
  • name: The name of the class.
  • Returns a ClsMetadata object, which is used as a decorator for your class and its methods.
• mock() -> MockOaas
  • Returns a MockOaas instance for local testing and development without a live OaaS environment.

---

ClsMetadata

Returned by Oparaca.new_cls(). Used as a decorator to define OaaS-compatible classes and their remotely callable methods.

Usage as Class Decorator:

oaas = Oparaca()
my_class_meta = oaas.new_cls("MyClassName")

@my_class_meta
class MyObject(BaseObject):
    # ... object implementation ...

Methods (used as decorators for methods within a BaseObject subclass):

• func(name: Optional[str] = None)

  • Decorator to mark a method within a BaseObject subclass as a remotely callable function (RPC).
  • name: Optional. If provided, this name will be used to register the function in OaaS. If None, the method's original name is used.
  • RPC Behavior: When a method decorated with func is called on an instance of BaseObject that is not local (i.e., obj.remote is True, which is the default unless local=True is specified during object creation with oaas.mock().create_object() or if the object is managed by a live OaaS environment), the SDK will automatically handle the serialization of inputs, perform an RPC to the OaaS platform, and deserialize the response. For local objects (obj.remote is False), the method is executed directly within the current process.

  @my_class_meta
  class MyObject(BaseObject):
      @my_class_meta.func()
      def my_method(self, arg: str) -> str:
          return f"Processed: {arg}"
  
      @my_class_meta.func("customFunctionName")
      def another_method(self):
          pass
  

---

BaseObject

The base class for all objects managed by OaaS. Your custom classes should inherit from BaseObject. Instances of BaseObject can be either "local" or "remote". Remote objects (where obj.remote is True) will have their methods decorated with @your_cls_meta.func() executed via RPC to the OaaS platform. Local objects execute these methods directly.

Synchronous API (when Oparaca is initialized with async_mode=False)

• get_data(key: int) -> Optional[bytes]
  • Retrieves data associated with the given key for the current object instance.
  • key: An integer key to identify the data.
  • Returns the data as bytes if found, otherwise None.
• set_data(key: int, value: bytes)
  • Sets data for the given key for the current object instance. This change is local until commit() is called.
  • key: An integer key.
  • value: The data to store, as bytes.
• commit()
  • Persists all local changes (made via set_data) to the OaaS storage. In the mock environment, this updates the in-memory store.

Asynchronous API (when Oparaca is initialized with async_mode=True)

• async get_data_async(key: int) -> Optional[bytes]
  • Asynchronously retrieves data associated with the given key.
• async set_data_async(key: int, value: bytes)
  • Asynchronously sets data for the given key. This change is local until commit_async() is called.
• async commit_async()
  • Asynchronously persists all local changes to the OaaS storage.

---

MockOaas

Provides a mock implementation of the OaaS environment for local testing. Obtained by calling oaas.mock().

Synchronous API (when Oparaca is initialized with async_mode=False)

• create_object(cls_meta: ClsMetadata, obj_id: int) -> T
  • Creates an instance of an object in the mock environment.
  • cls_meta: The ClsMetadata of the class to instantiate.
  • obj_id: A unique integer ID for the new object.
  • Returns an instance of the object (type T, where T is a subclass of BaseObject).
• load_object(cls_meta: ClsMetadata, obj_id: int) -> T
  • Loads an existing object instance from the mock environment.
  • cls_meta: The ClsMetadata of the class.
  • obj_id: The ID of the object to load.
  • Returns an instance of the object. Raises an error if the object does not exist.

Asynchronous API (when Oparaca is initialized with async_mode=True)

• async create_object_async(cls_meta: ClsMetadata, obj_id: int) -> T
  • Asynchronously creates an instance of an object in the mock environment.
• async load_object_async(cls_meta: ClsMetadata, obj_id: int) -> T
  • Asynchronously loads an existing object instance from the mock environment.

Run on OaaS

Prerequisites

• cargo (install via rust)
• oprc-cli cargo install --git https://github.com/pawissanutt/oaas-rs.git oprc-cli
• OaaS Platform (Oparaca)
  • Kubernetes Cluster (e.g., k3d with Docker runtime)

TODO

Build the project

You don't need to follow this guide unless you want to build the Python package on your own.

Prerequisites

• Python
• cargo (install via rust)
• uv (python package manager)

Build

uv sync
uv build