import boto3
import datetime
import pyarrow as pa
from pyiceberg.catalog import load_catalog

def lambda_handler(event, context):

    # カタログ読み込み用の変数
    region = "us-west-2"
    account_id = {アカウントID}
    s3table_bucket_name = "cm-kawanago-test"

    # テーブル指定に関連する変数
    database_name = "20250218_namespace"
    table_name = "maintenance_job_status_check"

    # カタログを読み込み
    catalog =  load_catalog(
            "S3TablesCatalog",
            **{
                "type": "rest",
                "warehouse": f"{account_id}:s3tablescatalog/{s3table_bucket_name}",
                "uri": f"https://glue.{region}.amazonaws.com/iceberg",
                "rest.sigv4-enabled": "true",
                "rest.signing-name": "glue",
                "rest.signing-region": region,
            },
        )

    # 日本時間に調整するためのタイムデルタ
    td = datetime.timedelta(hours=9)
    # 実行日時を取得
    now_jst = datetime.datetime.now() + td
    now_jst_str = datetime.datetime.strftime(now_jst, "%Y-%m-%d %H:%M:%S")

    # メンテンスジョブの履歴を取得
    s3tbl_client = boto3.client('s3tables')
    response = s3tbl_client.get_table_maintenance_job_status(
        tableBucketARN = f"arn:aws:s3tables:{region}:{account_id}:bucket/{s3table_bucket_name}",
        namespace = database_name,
        name = table_name
    )

    # テーブル作成用の辞書
    table_dict = {}

    for k, v in response["status"].items():

        try:

            # 最終実行時日時を日本時間に変換
            ts_jst = v["lastRunTimestamp"] + td
            ts_jst_str = datetime.datetime.strftime(ts_jst, "%Y-%m-%d %H:%M:%S")

        except:

            # 取得できなかった場合
            ts_jst_str = "Not_Yet_Run"

        # カラムと日時の値を格納
        table_dict[k.lower()] = ts_jst_str

    # 実行日時をデータへ格納
    table_dict["execute_timestamp"] = now_jst_str

    # 取得した結果をテーブルへ格納
    add_table = pa.Table.from_pylist([table_dict])
    tbl = catalog.load_table(f"{database_name}.{table_name}")
    tbl.append(add_table)

    return {
        "status_code": 200,
        "results": table_dict
    }