import time

from aws_lambda_powertools import Logger
from aws_lambda_powertools.event_handler.api_gateway import Router
from layercake.dynamodb import DynamoDBPersistenceLayer, KeyPair

from boto3clients import dynamodb_client
from config import USER_TABLE

logger = Logger(__name__)
router = Router()
dyn = DynamoDBPersistenceLayer(USER_TABLE, dynamodb_client)


@router.get('/<user_id>/orgs')
def get_orgs(user_id: str):
    time.sleep(1)
    start_key = router.current_event.get_query_string_value('start_key', None)

    return dyn.collection.query(
        # Post-migration (users): rename `orgs` to `ORG`
        key=KeyPair(user_id, 'orgs#'),
        start_key=start_key,
    )