529 lines
21 KiB
Python
529 lines
21 KiB
Python
#!/usr/bin/env python
|
||
"""
|
||
イベントユーザー登録スクリプト
|
||
|
||
外部システムAPI仕様書.mdを前提に、ユーザーデータCSVから、
|
||
各ユーザーごとにユーザー登録、チーム登録、エントリー登録、イベント参加を行う
|
||
docker composeで実施するPythonスクリプト
|
||
|
||
使用方法:
|
||
python register_event_users.py --event_code 大垣2509
|
||
|
||
ユーザーデータのCSVは以下の項目を持つ：
|
||
部門別数,時間,部門,チーム名,メール,パスワード,電話番号,氏名１,誕生日１,氏名２,誕生日２,氏名３,誕生日３,氏名４,誕生日４,氏名５,誕生日５,氏名６,誕生日６,氏名７,誕生日７,,
|
||
"""
|
||
|
||
import os
|
||
import sys
|
||
import csv
|
||
import requests
|
||
import argparse
|
||
import logging
|
||
from datetime import datetime, date
|
||
import time
|
||
import json
|
||
from typing import Dict, List, Optional, Tuple
|
||
|
||
# ログ設定
|
||
logging.basicConfig(
|
||
level=logging.INFO,
|
||
format='%(asctime)s - %(levelname)s - %(message)s',
|
||
handlers=[
|
||
logging.FileHandler('register_event_users.log'),
|
||
logging.StreamHandler()
|
||
]
|
||
)
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class EventUserRegistration:
|
||
def __init__(self, event_code: str, base_url: str = "http://localhost:8000", dry_run: bool = False):
|
||
"""
|
||
イベントユーザー登録クラス
|
||
|
||
Args:
|
||
event_code: イベントコード（例: 大垣2509）
|
||
base_url: APIベースURL
|
||
dry_run: テスト実行フラグ
|
||
"""
|
||
self.event_code = event_code
|
||
self.base_url = base_url.rstrip('/')
|
||
self.dry_run = dry_run
|
||
self.session = requests.Session()
|
||
self.admin_token = None
|
||
|
||
# 統計情報
|
||
self.stats = {
|
||
'processed_teams': 0,
|
||
'users_created': 0,
|
||
'users_updated': 0,
|
||
'teams_registered': 0,
|
||
'entries_created': 0,
|
||
'participations_created': 0,
|
||
'errors': []
|
||
}
|
||
|
||
logger.info(f"Event User Registration initialized for event: {event_code}")
|
||
if dry_run:
|
||
logger.info("DRY RUN MODE - No actual API calls will be made")
|
||
|
||
def get_or_create_user(self, email: str, password: str, firstname: str, lastname: str,
|
||
date_of_birth: str, phone: str) -> Tuple[bool, Optional[str], Optional[str]]:
|
||
"""
|
||
メールアドレスをキーに既存ユーザーを取得、存在しなければ新規作成
|
||
|
||
Args:
|
||
email: メールアドレス
|
||
password: パスワード
|
||
firstname: 名前
|
||
lastname: 姓
|
||
date_of_birth: 生年月日 (YYYY/MM/DD形式)
|
||
phone: 電話番号
|
||
|
||
Returns:
|
||
Tuple[success, user_id, token]
|
||
"""
|
||
if self.dry_run:
|
||
logger.info(f"[DRY RUN] Would get or create user: {email}")
|
||
return True, "dummy_user_id", "dummy_token"
|
||
|
||
try:
|
||
# まずログインを試行して既存ユーザーかチェック
|
||
login_data = {
|
||
"identifier": email,
|
||
"password": password
|
||
}
|
||
|
||
response = self.session.post(f"{self.base_url}/login/", json=login_data)
|
||
|
||
if response.status_code == 200:
|
||
# 既存ユーザーの場合、パスワード更新（実際にはパスワード更新APIが必要）
|
||
result = response.json()
|
||
token = result.get('token')
|
||
user_id = result.get('user', {}).get('id')
|
||
|
||
logger.info(f"既存ユーザーでログイン成功: {email}")
|
||
self.stats['users_updated'] += 1
|
||
return True, str(user_id), token
|
||
|
||
elif response.status_code == 401:
|
||
# ユーザーが存在しないか、パスワードが間違っている場合、新規登録を試行
|
||
return self._create_new_user(email, password, firstname, lastname, date_of_birth)
|
||
|
||
else:
|
||
logger.error(f"ログイン試行でエラー: {response.status_code} - {response.text}")
|
||
return False, None, None
|
||
|
||
except Exception as e:
|
||
logger.error(f"ユーザー認証エラー: {str(e)}")
|
||
return False, None, None
|
||
|
||
def _create_new_user(self, email: str, password: str, firstname: str, lastname: str,
|
||
date_of_birth: str) -> Tuple[bool, Optional[str], Optional[str]]:
|
||
"""
|
||
新規ユーザーを作成
|
||
"""
|
||
try:
|
||
# 生年月日をYYYY-MM-DD形式に変換
|
||
if '/' in date_of_birth:
|
||
date_parts = date_of_birth.split('/')
|
||
if len(date_parts) == 3:
|
||
birth_date = f"{date_parts[0]}-{date_parts[1].zfill(2)}-{date_parts[2].zfill(2)}"
|
||
else:
|
||
birth_date = "1990-01-01" # デフォルト値
|
||
else:
|
||
birth_date = date_of_birth
|
||
|
||
# 仮ユーザー登録
|
||
register_data = {
|
||
"email": email,
|
||
"password": password,
|
||
"firstname": firstname,
|
||
"lastname": lastname,
|
||
"date_of_birth": birth_date,
|
||
"female": False, # デフォルト値
|
||
"is_rogaining": True
|
||
}
|
||
|
||
response = self.session.post(f"{self.base_url}/register/", json=register_data)
|
||
|
||
if response.status_code in [200, 201]:
|
||
logger.info(f"仮ユーザー登録成功: {email}")
|
||
|
||
# 実際のシステムでは、メール認証コードを使って本登録を完了する必要があります
|
||
# ここでは簡略化のため、直接ログインを試行します
|
||
time.sleep(1) # 少し待機
|
||
|
||
login_data = {
|
||
"identifier": email,
|
||
"password": password
|
||
}
|
||
|
||
login_response = self.session.post(f"{self.base_url}/login/", json=login_data)
|
||
|
||
if login_response.status_code == 200:
|
||
result = login_response.json()
|
||
token = result.get('token')
|
||
user_id = result.get('user', {}).get('id')
|
||
|
||
logger.info(f"新規ユーザーのログイン成功: {email}")
|
||
self.stats['users_created'] += 1
|
||
return True, str(user_id), token
|
||
else:
|
||
logger.warning(f"新規ユーザーのログインに失敗: {email}")
|
||
# メール認証が必要な可能性があります
|
||
self.stats['users_created'] += 1
|
||
return True, "pending_verification", None
|
||
|
||
else:
|
||
error_msg = response.text
|
||
logger.error(f"ユーザー登録失敗: {email} - {error_msg}")
|
||
return False, None, None
|
||
|
||
except Exception as e:
|
||
logger.error(f"新規ユーザー作成エラー: {str(e)}")
|
||
return False, None, None
|
||
|
||
def register_team_and_members(self, team_data: Dict, zekken_number: int) -> Tuple[bool, Optional[str]]:
|
||
"""
|
||
チーム登録とメンバー登録
|
||
|
||
Args:
|
||
team_data: チームデータ（CSVの1行分）
|
||
zekken_number: ゼッケン番号
|
||
|
||
Returns:
|
||
Tuple[success, team_id]
|
||
"""
|
||
if self.dry_run:
|
||
logger.info(f"[DRY RUN] Would register team: {team_data['チーム名']} with zekken: {zekken_number}")
|
||
return True, "dummy_team_id"
|
||
|
||
try:
|
||
# チーム登録データを準備
|
||
register_data = {
|
||
"zekken_number": zekken_number,
|
||
"event_code": self.event_code,
|
||
"team_name": team_data['チーム名'],
|
||
"class_name": team_data['部門'],
|
||
"password": team_data['パスワード']
|
||
}
|
||
|
||
# チーム登録API呼び出し
|
||
response = self.session.post(f"{self.base_url}/register_team", json=register_data)
|
||
|
||
if response.status_code in [200, 201]:
|
||
result = response.json()
|
||
if result.get('status') == 'OK':
|
||
team_id = result.get('team_id')
|
||
logger.info(f"チーム登録成功: {team_data['チーム名']} (zekken: {zekken_number})")
|
||
self.stats['teams_registered'] += 1
|
||
|
||
# メンバー登録
|
||
success = self._register_team_members(team_data, team_id)
|
||
return success, str(team_id)
|
||
else:
|
||
logger.error(f"チーム登録エラー: {result.get('message')}")
|
||
return False, None
|
||
else:
|
||
logger.error(f"チーム登録API呼び出し失敗: {response.status_code} - {response.text}")
|
||
return False, None
|
||
|
||
except Exception as e:
|
||
logger.error(f"チーム登録エラー: {str(e)}")
|
||
return False, None
|
||
|
||
def _register_team_members(self, team_data: Dict, team_id: str) -> bool:
|
||
"""
|
||
チームメンバーを登録（最大7名）
|
||
|
||
Args:
|
||
team_data: チームデータ
|
||
team_id: チームID
|
||
|
||
Returns:
|
||
成功フラグ
|
||
"""
|
||
if self.dry_run:
|
||
logger.info(f"[DRY RUN] Would register team members for team: {team_id}")
|
||
return True
|
||
|
||
try:
|
||
success_count = 0
|
||
|
||
# メンバー1-7を順番に処理
|
||
for i in range(1, 8):
|
||
name_key = f'氏名{i}'
|
||
birth_key = f'誕生日{i}'
|
||
|
||
if name_key in team_data and team_data[name_key].strip():
|
||
name = team_data[name_key].strip()
|
||
birth_date = team_data.get(birth_key, '1990/01/01')
|
||
|
||
# ダミーメールアドレスを生成
|
||
dummy_email = f"{name.replace(' ', '')}_{team_id}_{i}@dummy.local"
|
||
|
||
# メンバー追加データ
|
||
member_data = {
|
||
"email": dummy_email,
|
||
"firstname": name.split()[0] if ' ' in name else name,
|
||
"lastname": name.split()[-1] if ' ' in name else "",
|
||
"date_of_birth": birth_date.replace('/', '-'),
|
||
"female": False # デフォルト値
|
||
}
|
||
|
||
# メンバー追加API呼び出し
|
||
response = self.session.post(
|
||
f"{self.base_url}/teams/{team_id}/members/",
|
||
json=member_data
|
||
)
|
||
|
||
if response.status_code in [200, 201]:
|
||
logger.info(f"メンバー追加成功: {name} -> チーム{team_id}")
|
||
success_count += 1
|
||
else:
|
||
logger.warning(f"メンバー追加失敗: {name} - {response.text}")
|
||
|
||
logger.info(f"チーム{team_id}のメンバー登録完了: {success_count}名")
|
||
return success_count > 0
|
||
|
||
except Exception as e:
|
||
logger.error(f"メンバー登録エラー: {str(e)}")
|
||
return False
|
||
|
||
def create_event_entry(self, team_id: str, category_id: int = 1) -> Tuple[bool, Optional[str]]:
|
||
"""
|
||
イベントエントリー登録
|
||
|
||
Args:
|
||
team_id: チームID
|
||
category_id: カテゴリID
|
||
|
||
Returns:
|
||
Tuple[success, entry_id]
|
||
"""
|
||
if self.dry_run:
|
||
logger.info(f"[DRY RUN] Would create event entry for team: {team_id}")
|
||
return True, "dummy_entry_id"
|
||
|
||
try:
|
||
# エントリーデータ準備
|
||
entry_data = {
|
||
"team_id": team_id,
|
||
"event_code": self.event_code,
|
||
"category": category_id,
|
||
"entry_date": datetime.now().strftime("%Y-%m-%d")
|
||
}
|
||
|
||
# エントリー登録API呼び出し
|
||
response = self.session.post(f"{self.base_url}/entry/", json=entry_data)
|
||
|
||
if response.status_code in [200, 201]:
|
||
result = response.json()
|
||
entry_id = result.get('id') or result.get('entry_id')
|
||
logger.info(f"エントリー登録成功: team_id={team_id}, entry_id={entry_id}")
|
||
self.stats['entries_created'] += 1
|
||
return True, str(entry_id)
|
||
else:
|
||
logger.error(f"エントリー登録失敗: {response.status_code} - {response.text}")
|
||
return False, None
|
||
|
||
except Exception as e:
|
||
logger.error(f"エントリー登録エラー: {str(e)}")
|
||
return False, None
|
||
|
||
def participate_in_event(self, entry_id: str, zekken_number: int) -> bool:
|
||
"""
|
||
イベント参加処理
|
||
|
||
Args:
|
||
entry_id: エントリーID
|
||
zekken_number: ゼッケン番号
|
||
|
||
Returns:
|
||
成功フラグ
|
||
"""
|
||
if self.dry_run:
|
||
logger.info(f"[DRY RUN] Would participate in event: entry_id={entry_id}, zekken={zekken_number}")
|
||
return True
|
||
|
||
try:
|
||
# イベント参加データ準備
|
||
participation_data = {
|
||
"entry_id": entry_id,
|
||
"event_code": self.event_code,
|
||
"zekken_number": zekken_number,
|
||
"participation_date": datetime.now().strftime("%Y-%m-%d")
|
||
}
|
||
|
||
# イベント参加API呼び出し（実際のAPIエンドポイントに合わせて調整が必要）
|
||
response = self.session.post(f"{self.base_url}/start_from_rogapp", json=participation_data)
|
||
|
||
if response.status_code in [200, 201]:
|
||
logger.info(f"イベント参加成功: entry_id={entry_id}, zekken={zekken_number}")
|
||
self.stats['participations_created'] += 1
|
||
return True
|
||
else:
|
||
logger.warning(f"イベント参加API呼び出し結果: {response.status_code} - {response.text}")
|
||
# 参加処理は必須ではないため、警告のみでTrueを返す
|
||
return True
|
||
|
||
except Exception as e:
|
||
logger.error(f"イベント参加エラー: {str(e)}")
|
||
return True # 参加処理は必須ではないため、エラーでもTrueを返す
|
||
|
||
def process_csv_file(self, csv_file_path: str) -> bool:
|
||
"""
|
||
CSVファイルを処理してユーザー登録からイベント参加まで実行
|
||
|
||
Args:
|
||
csv_file_path: CSVファイルパス
|
||
|
||
Returns:
|
||
成功フラグ
|
||
"""
|
||
try:
|
||
if not os.path.exists(csv_file_path):
|
||
logger.error(f"CSVファイルが見つかりません: {csv_file_path}")
|
||
return False
|
||
|
||
with open(csv_file_path, 'r', encoding='utf-8') as file:
|
||
csv_reader = csv.DictReader(file)
|
||
|
||
for row_num, row in enumerate(csv_reader, start=1):
|
||
try:
|
||
self._process_team_row(row, row_num)
|
||
|
||
# API呼び出し間隔を空ける
|
||
if not self.dry_run:
|
||
time.sleep(0.5)
|
||
|
||
except Exception as e:
|
||
error_msg = f"行{row_num}の処理でエラー: {str(e)}"
|
||
logger.error(error_msg)
|
||
self.stats['errors'].append(error_msg)
|
||
continue
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
logger.error(f"CSVファイル処理エラー: {str(e)}")
|
||
return False
|
||
|
||
def _process_team_row(self, row: Dict, row_num: int):
|
||
"""
|
||
CSVの1行（1チーム）を処理
|
||
|
||
Args:
|
||
row: CSV行データ
|
||
row_num: 行番号
|
||
"""
|
||
team_name = row.get('チーム名', '').strip()
|
||
email = row.get('メール', '').strip()
|
||
password = row.get('password', '').strip()
|
||
phone = row.get('電話番号', '').strip()
|
||
|
||
if not all([team_name, email, password]):
|
||
logger.warning(f"行{row_num}: 必須項目が不足 - チーム名={team_name}, メール={email}")
|
||
return
|
||
|
||
logger.info(f"行{row_num}の処理開始: チーム={team_name}")
|
||
|
||
# ゼッケン番号を生成（行番号ベース、実際の運用では別途管理が必要）
|
||
zekken_number = row_num
|
||
|
||
# 2-1. カスタムユーザー登録
|
||
# 最初のメンバー（氏名１）をメインユーザーとして使用
|
||
firstname = row.get('氏名１', team_name).strip()
|
||
lastname = ""
|
||
if ' ' in firstname:
|
||
parts = firstname.split(' ', 1)
|
||
firstname = parts[0]
|
||
lastname = parts[1]
|
||
|
||
date_of_birth = row.get('誕生日１', '1990/01/01')
|
||
|
||
user_success, user_id, token = self.get_or_create_user(
|
||
email, password, firstname, lastname, date_of_birth, phone
|
||
)
|
||
|
||
if not user_success:
|
||
logger.error(f"行{row_num}: ユーザー登録/取得失敗")
|
||
return
|
||
|
||
# 2-2. チーム登録、メンバー登録
|
||
team_success, team_id = self.register_team_and_members(row, zekken_number)
|
||
|
||
if not team_success:
|
||
logger.error(f"行{row_num}: チーム登録失敗")
|
||
return
|
||
|
||
# 2-3. エントリー登録
|
||
entry_success, entry_id = self.create_event_entry(team_id)
|
||
|
||
if not entry_success:
|
||
logger.error(f"行{row_num}: エントリー登録失敗")
|
||
return
|
||
|
||
# 2-4. イベント参加
|
||
participation_success = self.participate_in_event(entry_id, zekken_number)
|
||
|
||
if participation_success:
|
||
logger.info(f"行{row_num}: 全処理完了 - チーム={team_name}, zekken={zekken_number}")
|
||
self.stats['processed_teams'] += 1
|
||
else:
|
||
logger.warning(f"行{row_num}: イベント参加処理で警告")
|
||
|
||
def print_statistics(self):
|
||
"""
|
||
処理統計を出力
|
||
"""
|
||
logger.info("=== 処理統計 ===")
|
||
logger.info(f"処理完了チーム数: {self.stats['processed_teams']}")
|
||
logger.info(f"作成ユーザー数: {self.stats['users_created']}")
|
||
logger.info(f"更新ユーザー数: {self.stats['users_updated']}")
|
||
logger.info(f"登録チーム数: {self.stats['teams_registered']}")
|
||
logger.info(f"作成エントリー数: {self.stats['entries_created']}")
|
||
logger.info(f"参加登録数: {self.stats['participations_created']}")
|
||
logger.info(f"エラー数: {len(self.stats['errors'])}")
|
||
|
||
if self.stats['errors']:
|
||
logger.error("エラー詳細:")
|
||
for error in self.stats['errors']:
|
||
logger.error(f" - {error}")
|
||
|
||
|
||
def main():
|
||
parser = argparse.ArgumentParser(description='イベントユーザー登録スクリプト')
|
||
parser.add_argument('--event_code', required=True, help='イベントコード（例: 大垣2509）')
|
||
parser.add_argument('--csv_file', default='CPLIST/input/team2025.csv', help='CSVファイルパス')
|
||
parser.add_argument('--base_url', default='http://localhost:8000', help='APIベースURL')
|
||
parser.add_argument('--dry_run', action='store_true', help='テスト実行（実際のAPI呼び出しなし）')
|
||
|
||
args = parser.parse_args()
|
||
|
||
logger.info(f"イベントユーザー登録処理開始: event_code={args.event_code}")
|
||
|
||
# 登録処理実行
|
||
registration = EventUserRegistration(
|
||
event_code=args.event_code,
|
||
base_url=args.base_url,
|
||
dry_run=args.dry_run
|
||
)
|
||
|
||
success = registration.process_csv_file(args.csv_file)
|
||
|
||
# 統計出力
|
||
registration.print_statistics()
|
||
|
||
if success:
|
||
logger.info("処理が正常に完了しました")
|
||
return 0
|
||
else:
|
||
logger.error("処理中にエラーが発生しました")
|
||
return 1
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|