AWS LambdaのpythonでDynamoDBのキーを可変で取得したい

Question

前提・実現したいこと

DynamoDBをキーで検索し、データをjsonで取得して表示するサンプルを
作成しています。以下の点で躓いていて何とかしたいのですがお助けいただけますでしょうか。

1.複数のKeyを受けた際でも対応できるようにしたい。

　以下のコードの__Key={keyname: id}__の箇所について
複号キーを受付けた場合に対応できるようにしたいと考えているのですが
どうすればいいでしょうか。

import boto3

def get_data(id,tablename,keyname):
  try:
      dynamodb = boto3.resource('dynamodb')
      table    = dynamodb.Table(tablename)
      response = table.get_item(
              Key={
                   keyname: id
              }
          )
      return response['Item']
  except Exception as e:
    print(e)
    raise e

def lambda_handler(event, context):
    category = '2'
    tablename = 'item'
    keyname = 'item-category'
    
    item = get_data(category,tablename,keyname)
    return item

(解決済)2.レスポンスを受けたitemから一部の内容を取得したい

受取った値からは以下の様なレスポンスを取得できます。itemcode__は__item-category
によって数が不変となっています。上記の__return item__の箇所については__item_codeN__内の文字列__"222-0001"__等を取得しその値を変数に格納して別の用途に使用したいと考えて
おります。どうすれば可変する項目の文字列を取得して変数に格納することができるでしょうか？

{
  "itemcode_1": "222-0001",
  "itemcode_2": "222-0002",
  "item-category": "2"
}

テーブル構造は以下の様な形です。
https://teratail.com/questions/154614

つたない説明で恐縮ですがよろしくお願いいたします。

Answer 1

何とか自己解決しました。

1について美しいやり方ではないですが複合キーで検索する関数を準備し、id2のパラメータで制御するようにして対応できました。ハッシュキーのみ分も1つにまとめられたらよかったのですが、こちらについてはいいアイデアがありませんでした。引数が可変にできてモードを付けられればできそうですが、あまり美しくありません。こうすればもっとシンプルになるよというご指摘などあればご教示いただけると幸いです。

#複合キーで検索
def get_table_val(id1,id2,key1,key2,tablename):
  try:
      table    = dynamodb.Table(tablename)
      #検索キーが指定書式の場合はid1とid2で検索する。
      if id2==0:
        response = table.query(
          KeyConditionExpression=Key(key1).eq(id1)
        )
      else:
        response = table.query(
          KeyConditionExpression=Key(key1).eq(id1) & Key(key2).eq(id2)
        )
      return response['Items']
  except Exception as e:
    raise(e)

Answer 2

2の質問については自己解決できました。
以下のコードで値が取得できたのでこちらを応用できそうです。

import boto3
import json
import os
import re

def get_data(id,tablename,keyname):
  try:
      dynamodb = boto3.resource('dynamodb')
      table    = dynamodb.Table(tablename)
      response = table.get_item(
              Key={
                   keyname: id
              }
          )
      return response['Item']
  except Exception as e:
    print(e)
    raise e

def lambda_handler(event, context):
    category = '1'
    tablename = os.getenv('TABLE_NAME')
    keyname = os.getenv('KEY_NAME')
    item = get_data(category,tablename,keyname)
    #ループで辞書のvalueを取得する
    for i_key,i_val in item.items():
      m = re.match(keyname,i_key)
      if (m is None):
        print(i_val)