migrateスクリプト追加

migrate_kids_available_type.py → migrations/migrate_kids_available_type.py

@@ -1,7 +1,15 @@
+import sys
+import os
 import boto3
+from botocore.exceptions import ClientError
 from decimal import Decimal
 import time
 from datetime import datetime, timezone, timedelta
+import argparse
+
+# Add the project root directory to the Python path
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
 from config.logger import logger
 
 class KidsAvailableTypeMigration:
@@ -15,7 +23,7 @@ class KidsAvailableTypeMigration:
         """
         self.dynamodb = boto3.resource('dynamodb', region_name=region_name)
         self.environment = environment
-        self.table_name = f"{environment}-shops"
+        self.table_name = get_table_name(environment)
         self.table = self.dynamodb.Table(self.table_name)
         logger.info(f"初期化完了: テーブル {self.table_name}")
 
@@ -79,13 +87,8 @@ class KidsAvailableTypeMigration:
             # DynamoDBのキー構成に基づいて更新
             current_time = self.get_current_time_jst_ns()
             
-            # テーブルにソートキーがある場合とない場合に対応
-            key = {'shop_id': shop_id}
-            if shop_name_ja:
-                key['shop_name_ja'] = shop_name_ja
-            
             response = self.table.update_item(
-                Key=key,
+                Key={'shop_id': shop_id},
                 UpdateExpression='SET kids_available_type = :val, updated_at = :updated_at',
                 ExpressionAttributeValues={
                     ':val': kids_available_type_value,
@@ -224,55 +227,45 @@ class KidsAvailableTypeMigration:
         return verification_stats
 
 
-def main():
+def get_table_name(environment):
+    """環境に応じたDynamoDBのテーブル名を取得する"""
+    if environment == 'production':
+        return 'prod-shops'
+    elif environment == 'development':
+        return 'dev-shops'
+    elif environment == 'staging':
+        return 'stg-shops'
+    else:
+        # デフォルトは 'dev-shops'
+        return 'dev-shops'
+
+
+def main(environment):
     """
-    メイン実行関数
-    使用例:
-    - ドライラン: python migrate_kids_available_type.py
-    - 実際の実行: python migrate_kids_available_type.py --execute
+    マイグレーションのメイン処理
     """
-    import sys
-    
-    # 環境変数から環境を取得（デフォルトはdev）
-    import os
-    environment = os.getenv('ENVIRONMENT', 'dev')
+    # 引数で実行モードを切り替える
+    is_dry_run = '--execute' not in sys.argv
     
-    # コマンドライン引数の確認
-    dry_run = '--execute' not in sys.argv
-    
-    if dry_run:
+    if is_dry_run:
         logger.info("ドライランモードで実行します。実際の更新は行われません。")
         logger.info("実際に更新を実行する場合は --execute オプションを付けてください。")
-    else:
-        logger.info("実際の更新を実行します。")
-        response = input("本当に実行しますか？ (yes/no): ")
-        if response.lower() != 'yes':
-            logger.info("実行をキャンセルしました。")
-            return
     
     # マイグレーション実行
     migration = KidsAvailableTypeMigration(environment=environment)
-    
-    try:
-        # マイグレーション実行
-        stats = migration.migrate_all_shops(dry_run=dry_run)
-        
-        # 実際に更新した場合は検証も実行
-        if not dry_run:
-            logger.info("マイグレーション完了。検証を開始します...")
-            time.sleep(2)  # DynamoDBの整合性のため少し待機
-            verification_stats = migration.verify_migration()
-            
-            # 検証結果をチェック
-            if verification_stats['incorrect_mapping'] == 0 and verification_stats['missing_kids_available_type'] == 0:
-                logger.info("✅ マイグレーションは正常に完了しました！")
-            else:
-                logger.error("❌ マイグレーションに問題があります。ログを確認してください。")
-        
-    except Exception as e:
-        logger.error(f"マイグレーション中にエラーが発生しました: {str(e)}")
-        raise
+    migration.migrate_all_shops(dry_run=is_dry_run)
 
 
 if __name__ == "__main__":
-    main() 
+    """
+    実行コマンド例:
+      - ドライラン: python migrate_kids_available_type.py
+      - 実際の実行: python migrate_kids_available_type.py --execute
+    """
+    parser = argparse.ArgumentParser(description='kids_available_type のマイグレーションスクリプト')
+    parser.add_argument('--env', type=str, default='development', choices=['development', 'production', 'staging'], help='実行環境')
+    parser.add_argument('--execute', action='store_true', help='このフラグを付けると実際にDBを更新します')
+    args = parser.parse_args()
+    
+    # 環境変数から環境を取得（デフォルトはdev）
+    main(args.env) 

migration_readme.md → migrations/migration_readme.md

@@ -12,7 +12,7 @@ DynamoDBの`shops`テーブルに新規カラム`kids_available_type`を追加
 
 ## 値の意味
 
-- `kids_available_type = -1`: 子連れ不可/非表示
+- `kids_available_type = -1`: 子連れ非表示
 - `kids_available_type = 1`: 子連れ可
 
 ## 使用方法