rayepps/dynamof

View on GitHub
dynofunc/operations/find.py

Summary

Maintainability
A
0 mins
Test Coverage

from dynofunc.core import builder as ab
from dynofunc.core import args
from dynofunc.core.utils import shake
from dynofunc.core.model import Operation
from dynofunc.core.response import response


def find(table_name: str, key: dict):
    """Creates an Operation that will find an item in a table when run.

    Args:
        table_name (str): The name of the table to find the item in
        key (dict): An object with a key, value that will be used as the identifier for the item
            Examples:
                - { 'username': 'sunshie' }
                - { 'id': '9snw82h', 'type': 'car' } -> when table uses hash key and range key

    """
    build = ab.builder(
        table_name=table_name,
        key=key)
    description = shake(
        TableName=build(args.TableName),
        Key=build(args.Key))
    return Operation(description, run)

def run(client, description):
    res = client.get_item(**description)
    return response(res)