import asyncio
import os
import psutil
import resource
import subprocess
from botocore.exceptions import HTTPClientError
import aiobotocore
import aiobotocore.session
from aiobotocore.config import AioConfig
from contextlib import AsyncExitStack

__session = None

async def get_dynamodb_secret(table: str, key: dict, region_name: str = 'us-east-1'):
    global __session
    if __session is None:
        __session = aiobotocore.session.get_session()

    async with AsyncExitStack() as exit_stack:
        config = AioConfig(connect_timeout=5, read_timeout=5)
        dynamodb_client = await exit_stack.enter_async_context(
            __session.create_client('dynamodb', config=config, region_name=region_name)
        )
        await dynamodb_client.get_item(
            TableName=table,
            Key=key,
            ConsistentRead=False,
        )

async def monitor_open_files(process: psutil.Process, interval: float = 0.2):
    print("--- Starting File Descriptor Monitor ---")
    while True:
        try:
            cmd = f"lsof -p {process.pid} | wc -l"
            result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
            lsof_count = int(result.stdout.strip())
            print(f"Monitoring... lsof count: {lsof_count})")
        except psutil.NoSuchProcess:
            print("--- Monitor stopping: Process not found. ---")
            break
        await asyncio.sleep(interval)

async def retrieve_dynamodb_secrets():
    table = "<your_dynamodb_table>"
    key = {"random_key": {"S": "random_value"}}
    await get_dynamodb_secret(table, key)

async def main():
    soft_limit, hard_limit = resource.getrlimit(resource.RLIMIT_NOFILE)
    print(f"File Descriptor Limits - Soft: {soft_limit}, Hard: {hard_limit}\n")

    current_process = psutil.Process(os.getpid())

    monitor_task = asyncio.create_task(monitor_open_files(current_process))

    try:
        for i in range(100):
            print(f"dynamodb tests start: {i}")
            tasks = []
            for j in range(5):
                tasks.append(retrieve_dynamodb_secrets())
            await asyncio.gather(*tasks)
            print(f"dynamodb tests end: {i}")
    except HTTPClientError:
        print("\n⚠️  Caught 'Too many open files' error!")
    finally:
        monitor_task.cancel()
        print("\nExiting.")


if __name__ == '__main__':
    asyncio.run(main())