330 lines
13 KiB
Python
330 lines
13 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
既存データ保護版移行プログラム（Location2025対応）
|
||
既存のentry、team、memberデータを削除せずに移行データを追加する
|
||
Location2025テーブルとの整合性を確認し、チェックポイント参照の妥当性を検証する
|
||
"""
|
||
|
||
import os
|
||
import sys
|
||
import psycopg2
|
||
from datetime import datetime, time, timedelta
|
||
import pytz
|
||
|
||
def get_event_date(event_code):
|
||
"""イベントコードに基づいてイベント日付を返す"""
|
||
event_dates = {
|
||
'美濃加茂': datetime(2024, 5, 19), # 修正済み
|
||
'岐阜市': datetime(2024, 4, 28),
|
||
'大垣２': datetime(2024, 4, 20),
|
||
'各務原': datetime(2024, 3, 24),
|
||
'下呂': datetime(2024, 3, 10),
|
||
'中津川': datetime(2024, 3, 2),
|
||
'揖斐川': datetime(2024, 2, 18),
|
||
'高山': datetime(2024, 2, 11),
|
||
'大垣': datetime(2024, 1, 27),
|
||
'多治見': datetime(2024, 1, 20),
|
||
# 2024年のその他のイベント
|
||
'養老ロゲ': datetime(2024, 6, 1),
|
||
'郡上': datetime(2024, 11, 3), # 郡上イベント追加
|
||
# 2025年新規イベント
|
||
'岐阜ロゲイニング2025': datetime(2025, 9, 15),
|
||
}
|
||
return event_dates.get(event_code)
|
||
|
||
def convert_utc_to_jst(utc_timestamp):
|
||
"""UTC時刻をJST時刻に変換"""
|
||
if not utc_timestamp:
|
||
return None
|
||
|
||
utc_tz = pytz.UTC
|
||
jst_tz = pytz.timezone('Asia/Tokyo')
|
||
|
||
# UTCタイムゾーン情報を付加
|
||
if utc_timestamp.tzinfo is None:
|
||
utc_timestamp = utc_tz.localize(utc_timestamp)
|
||
|
||
# JSTに変換
|
||
return utc_timestamp.astimezone(jst_tz).replace(tzinfo=None)
|
||
|
||
def parse_goal_time(goal_time_str, event_date_str):
|
||
"""goal_time文字列を適切なdatetimeに変換"""
|
||
if not goal_time_str:
|
||
return None
|
||
|
||
try:
|
||
# goal_timeが時刻のみの場合（例: "13:45:00"）
|
||
goal_time = datetime.strptime(goal_time_str, "%H:%M:%S").time()
|
||
|
||
# event_date_strからイベント日付を解析
|
||
event_date = datetime.strptime(event_date_str, "%Y-%m-%d").date()
|
||
|
||
# 日付と時刻を結合
|
||
goal_datetime = datetime.combine(event_date, goal_time)
|
||
|
||
# JSTとして解釈
|
||
jst_tz = pytz.timezone('Asia/Tokyo')
|
||
goal_datetime_jst = jst_tz.localize(goal_datetime)
|
||
|
||
# UTCに変換して返す
|
||
return goal_datetime_jst.astimezone(pytz.UTC)
|
||
|
||
except (ValueError, TypeError) as e:
|
||
print(f"goal_time変換エラー: {goal_time_str} - {e}")
|
||
return None
|
||
|
||
def clean_target_database_selective(target_cursor):
|
||
"""ターゲットデータベースの選択的クリーンアップ（既存データを保護）"""
|
||
print("=== ターゲットデータベースの選択的クリーンアップ ===")
|
||
|
||
# 外部キー制約を一時的に無効化
|
||
target_cursor.execute("SET session_replication_role = replica;")
|
||
|
||
try:
|
||
# GPSチェックインデータのみクリーンアップ（重複移行防止）
|
||
target_cursor.execute("DELETE FROM rog_gpscheckin WHERE comment = 'migrated_from_gifuroge'")
|
||
deleted_checkins = target_cursor.rowcount
|
||
print(f"過去の移行GPSチェックインデータを削除: {deleted_checkins}件")
|
||
|
||
# 注意: rog_entry, rog_team, rog_member, rog_location2025 は削除しない！
|
||
print("注意: 既存のentry、team、member、location2025データは保護されます")
|
||
|
||
finally:
|
||
# 外部キー制約を再有効化
|
||
target_cursor.execute("SET session_replication_role = DEFAULT;")
|
||
|
||
def verify_location2025_compatibility(target_cursor):
|
||
"""Location2025テーブルとの互換性を確認"""
|
||
print("\n=== Location2025互換性確認 ===")
|
||
|
||
try:
|
||
# Location2025テーブルの存在確認
|
||
target_cursor.execute("""
|
||
SELECT COUNT(*) FROM information_schema.tables
|
||
WHERE table_name = 'rog_location2025'
|
||
""")
|
||
|
||
table_exists = target_cursor.fetchone()[0] > 0
|
||
|
||
if table_exists:
|
||
# Location2025のデータ数確認
|
||
target_cursor.execute("SELECT COUNT(*) FROM rog_location2025")
|
||
location2025_count = target_cursor.fetchone()[0]
|
||
print(f"✅ rog_location2025テーブル存在: {location2025_count}件のチェックポイント")
|
||
|
||
# イベント別チェックポイント数確認
|
||
target_cursor.execute("""
|
||
SELECT e.event_code, COUNT(l.id) as checkpoint_count
|
||
FROM rog_location2025 l
|
||
JOIN rog_newevent2 e ON l.event_id = e.id
|
||
GROUP BY e.event_code
|
||
ORDER BY checkpoint_count DESC
|
||
LIMIT 10
|
||
""")
|
||
|
||
event_checkpoints = target_cursor.fetchall()
|
||
if event_checkpoints:
|
||
print("イベント別チェックポイント数（上位10件）:")
|
||
for event_code, count in event_checkpoints:
|
||
print(f" {event_code}: {count}件")
|
||
|
||
return True
|
||
else:
|
||
print("⚠️ rog_location2025テーブルが見つかりません")
|
||
print("注意: 移行は可能ですが、チェックポイント管理機能は制限されます")
|
||
return False
|
||
|
||
except Exception as e:
|
||
print(f"❌ Location2025互換性確認エラー: {e}")
|
||
return False
|
||
|
||
def backup_existing_data(target_cursor):
|
||
"""既存データのバックアップ状況を確認"""
|
||
print("\n=== 既存データ保護確認 ===")
|
||
|
||
# 既存データ数を確認
|
||
target_cursor.execute("SELECT COUNT(*) FROM rog_entry")
|
||
entry_count = target_cursor.fetchone()[0]
|
||
|
||
target_cursor.execute("SELECT COUNT(*) FROM rog_team")
|
||
team_count = target_cursor.fetchone()[0]
|
||
|
||
target_cursor.execute("SELECT COUNT(*) FROM rog_member")
|
||
member_count = target_cursor.fetchone()[0]
|
||
|
||
target_cursor.execute("SELECT COUNT(*) FROM rog_gpscheckin")
|
||
checkin_count = target_cursor.fetchone()[0]
|
||
|
||
# Location2025データ数も確認
|
||
try:
|
||
target_cursor.execute("SELECT COUNT(*) FROM rog_location2025")
|
||
location2025_count = target_cursor.fetchone()[0]
|
||
print(f" rog_location2025: {location2025_count} 件 (保護対象)")
|
||
except Exception as e:
|
||
print(f" rog_location2025: 確認エラー ({e})")
|
||
location2025_count = 0
|
||
|
||
print(f"既存データ保護状況:")
|
||
print(f" rog_entry: {entry_count} 件 (保護対象)")
|
||
print(f" rog_team: {team_count} 件 (保護対象)")
|
||
print(f" rog_member: {member_count} 件 (保護対象)")
|
||
print(f" rog_gpscheckin: {checkin_count} 件 (移行対象)")
|
||
|
||
if entry_count > 0 or team_count > 0 or member_count > 0:
|
||
print("✅ 既存のcore application dataが検出されました。これらは保護されます。")
|
||
return True
|
||
else:
|
||
print("⚠️ 既存のcore application dataが見つかりません。")
|
||
return False
|
||
|
||
def migrate_gps_data(source_cursor, target_cursor):
|
||
"""GPS記録データのみを移行（写真記録データは除外）"""
|
||
print("\n=== GPS記録データの移行 ===")
|
||
|
||
# GPS記録のみを取得（不正な写真記録データを除外）
|
||
source_cursor.execute("""
|
||
SELECT
|
||
serial_number, team_name, cp_number, record_time,
|
||
goal_time, late_point, buy_flag, image_address,
|
||
minus_photo_flag, create_user, update_user,
|
||
colabo_company_memo
|
||
FROM gps_information
|
||
WHERE serial_number < 20000 -- GPS専用データのみ
|
||
ORDER BY serial_number
|
||
""")
|
||
|
||
gps_records = source_cursor.fetchall()
|
||
print(f"移行対象GPS記録数: {len(gps_records)}件")
|
||
|
||
migrated_count = 0
|
||
error_count = 0
|
||
|
||
for record in gps_records:
|
||
try:
|
||
(serial_number, team_name, cp_number, record_time,
|
||
goal_time, late_point, buy_flag, image_address,
|
||
minus_photo_flag, create_user, update_user,
|
||
colabo_company_memo) = record
|
||
|
||
# UTC時刻をJST時刻に変換
|
||
record_time_jst = convert_utc_to_jst(record_time)
|
||
goal_time_utc = None
|
||
|
||
if goal_time:
|
||
# goal_timeをUTCに変換
|
||
if isinstance(goal_time, str):
|
||
# イベント名からイベント日付を取得
|
||
event_name = colabo_company_memo or "不明"
|
||
event_date = get_event_date(event_name)
|
||
if event_date:
|
||
goal_time_utc = parse_goal_time(goal_time, event_date.strftime("%Y-%m-%d"))
|
||
elif isinstance(goal_time, datetime):
|
||
goal_time_utc = convert_utc_to_jst(goal_time)
|
||
|
||
# rog_gpscheckinに挿入（マイグレーション用マーカー付き）
|
||
target_cursor.execute("""
|
||
INSERT INTO rog_gpscheckin
|
||
(serial_number, team_name, cp_number, record_time, goal_time,
|
||
late_point, buy_flag, image_address, minus_photo_flag,
|
||
create_user, update_user, comment)
|
||
VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)
|
||
""", (
|
||
serial_number, team_name, cp_number, record_time_jst, goal_time_utc,
|
||
late_point, buy_flag, image_address, minus_photo_flag,
|
||
create_user, update_user, 'migrated_from_gifuroge'
|
||
))
|
||
|
||
migrated_count += 1
|
||
|
||
if migrated_count % 1000 == 0:
|
||
print(f"移行進捗: {migrated_count}/{len(gps_records)}件")
|
||
|
||
except Exception as e:
|
||
error_count += 1
|
||
print(f"移行エラー (record {serial_number}): {e}")
|
||
continue
|
||
|
||
print(f"\n移行完了: {migrated_count}件成功, {error_count}件エラー")
|
||
return migrated_count
|
||
|
||
def main():
|
||
"""メイン移行処理（既存データ保護版）"""
|
||
print("=== 既存データ保護版移行プログラム開始 ===")
|
||
print("注意: 既存のentry、team、memberデータは削除されません")
|
||
|
||
# データベース接続設定
|
||
source_config = {
|
||
'host': 'postgres-db',
|
||
'port': 5432,
|
||
'database': 'gifuroge',
|
||
'user': 'admin',
|
||
'password': 'admin123456'
|
||
}
|
||
|
||
target_config = {
|
||
'host': 'postgres-db',
|
||
'port': 5432,
|
||
'database': 'rogdb',
|
||
'user': 'admin',
|
||
'password': 'admin123456'
|
||
}
|
||
|
||
source_conn = None
|
||
target_conn = None
|
||
|
||
try:
|
||
# データベース接続
|
||
print("データベースに接続中...")
|
||
source_conn = psycopg2.connect(**source_config)
|
||
target_conn = psycopg2.connect(**target_config)
|
||
|
||
source_cursor = source_conn.cursor()
|
||
target_cursor = target_conn.cursor()
|
||
|
||
# Location2025互換性確認
|
||
location2025_available = verify_location2025_compatibility(target_cursor)
|
||
|
||
# 既存データ保護確認
|
||
has_existing_data = backup_existing_data(target_cursor)
|
||
|
||
# 確認プロンプト
|
||
print(f"\nLocation2025対応: {'✅ 利用可能' if location2025_available else '⚠️ 制限あり'}")
|
||
print(f"既存データ保護: {'✅ 検出済み' if has_existing_data else '⚠️ 未検出'}")
|
||
|
||
response = input("\n移行を開始しますか？ (y/N): ")
|
||
if response.lower() != 'y':
|
||
print("移行を中止しました。")
|
||
return
|
||
|
||
# 選択的クリーンアップ（既存データを保護）
|
||
clean_target_database_selective(target_cursor)
|
||
target_conn.commit()
|
||
|
||
# GPS記録データ移行
|
||
migrated_count = migrate_gps_data(source_cursor, target_cursor)
|
||
target_conn.commit()
|
||
|
||
print(f"\n=== 移行完了 ===")
|
||
print(f"移行されたGPS記録: {migrated_count}件")
|
||
print(f"Location2025互換性: {'✅ 対応済み' if location2025_available else '⚠️ 要確認'}")
|
||
if has_existing_data:
|
||
print("✅ 既存のentry、team、member、location2025データは保護されました")
|
||
else:
|
||
print("⚠️ 既存のcore application dataがありませんでした")
|
||
print(" 別途testdb/rogdb.sqlからの復元が必要です")
|
||
|
||
except Exception as e:
|
||
print(f"移行エラー: {e}")
|
||
if target_conn:
|
||
target_conn.rollback()
|
||
sys.exit(1)
|
||
|
||
finally:
|
||
if source_conn:
|
||
source_conn.close()
|
||
if target_conn:
|
||
target_conn.close()
|
||
|
||
if __name__ == "__main__":
|
||
main()
|