0

对所有人来说,

我是一个 python 初学者,我有一个 dynamoDB 的 docker 本地 docker 环境,但我想创建一个类,允许我在本地 docker dynamodb 或 aws dynamo db 上列出表或创建表。

import boto3

class DDB():

    def __init__(self):
        print("Welcome to the DynamoDB py")
    
    def ddbconnect(self,env):
        self.env = env 
        if self.env == "l":
            client = boto3.client('dynamodb',endpoint_url='http://localhost:8000')
            return env
        else:
            client = boto3.client('dynamodb')
            return env
            
    def ddbcreateTable(self,tbname,pk,sk):
        
        createResponse = client.create_table(
            AttributeDefinitions=[
                {
                    'AttributeName': pk,
                    'AttributeType': 'S',
                }, 
                {
                    'AttributeName': sk,
                    'AttributeType':'N'
                }
            ], 
            KeySchema=[
                {
                    'AttributeName': pk,
                    'KeyType':'HASH'
                },
                {
                    'AttributeName': sk,
                    'KeyType':'RANGE'
                },
            ],
            BillingMode = 'PAY_PER_REQUEST',
            TableName=tbname
        )

mydb=DDB()

env = input("enter L if it is a Local Dynamodb")
if env.lower() == "l":
    mydb.ddbconnect(env)
    
    while True:
        prompt = input("Choices are (C)reate or (L)ist Table or (E)xit -  L | C | E  : " )
        if prompt.lower() == "l":
        env=input()
        mydb.ddblist()
        continue
    elif prompt.lower() == "c":
        tbname=input("Enter the table name: ")
        pk=input("Enter the primary key: ")
        sk=input("Enter the sort key w/ numberic attribute: ")
        mydb.ddbcreateTable(tbname,pk,sk)
        continue
    elif prompt.lower() == "e" :
        print("Exiting")
        break
    else:
        print("Enter a valid selection")
        continue

我相信有一种更有效的写作方式,但这是一种逻辑

4

0 回答 0