wip
This commit is contained in:
@@ -1,19 +1,19 @@
|
||||
import os
|
||||
from typing import TYPE_CHECKING
|
||||
|
||||
import boto3
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from mypy_boto3_dynamodb.client import DynamoDBClient
|
||||
else:
|
||||
DynamoDBClient = object
|
||||
|
||||
def get_dynamodb_client():
|
||||
running_sam_local = os.getenv('AWS_SAM_LOCAL')
|
||||
|
||||
if os.getenv('AWS_LAMBDA_FUNCTION_NAME') and not running_sam_local:
|
||||
def get_dynamodb_client() -> DynamoDBClient:
|
||||
if os.getenv('AWS_LAMBDA_FUNCTION_NAME'):
|
||||
return boto3.client('dynamodb')
|
||||
|
||||
dockerhost = 'host.docker.internal'
|
||||
localhost = '127.0.0.1'
|
||||
host = dockerhost if running_sam_local else localhost
|
||||
|
||||
return boto3.client('dynamodb', endpoint_url=f'http://{host}:8000')
|
||||
return boto3.client('dynamodb', endpoint_url='http://127.0.0.1:8000')
|
||||
|
||||
|
||||
dynamodb_client = get_dynamodb_client()
|
||||
dynamodb_client: DynamoDBClient = get_dynamodb_client()
|
||||
|
||||
Reference in New Issue
Block a user