import requests
import os
import shutil
from google.colab import drive
import time

# Googleドライブをマウント
drive.mount('/content/drive')

# 設定値
API_URL = "https://api.spiral-platform.com/v1"
API_KEY = "" # SPRIALのAPIキー
APP_ROLE = "" #SPIRALの設定にてロールが指定されている場合は記入
DB_ID = ""
APP_ID = ""
delete_flg = 1  # ファイル削除フラグ（1: 削除する、0: 削除しない）
file_field_name = "file"  # ファイルフィールド名
file_field_id = "1"  # ファイルフィールドID
google_drive_folder_path = '/content/drive/My Drive/fromspiral/'  # アップロード先フォルダパス

# SPIRAL APIにリクエストを送信してデータを取得する関数
def api_curl_action(method, url, headers=None, data=None, return_json=True):
    headers = headers or {
        "Authorization": f"Bearer {API_KEY}",
        "X-Spiral-Api-Version": "1.1",
        "Content-Type": "application/json"
    }

    response = requests.request(method, API_URL + url, headers=headers, json=data)

    if return_json:
        try:
            return response.json()
        except requests.exceptions.JSONDecodeError:
            if method == "DELETE":
                return None
            else:
                print("JSONデコードエラー: レスポンスがJSON形式ではありません。")
                return None
    else:
        return response.content

# ファイル情報を取得
url = f"/apps/{APP_ID}/dbs/{DB_ID}/records"
result_record_list_select = api_curl_action("GET", url)

file_infos = []
if result_record_list_select and 'items' in result_record_list_select:
    for record in result_record_list_select['items']:
        if file_field_name in record:
            for file in record[file_field_name]:
                if 'fileKey' in file and 'fileName' in file:
                    download_url = f"/apps/{APP_ID}/dbs/{DB_ID}/{file_field_id}/{record['_id']}/files/{file['fileKey']}/download"
                    file_infos.append({
                        'fileName': file['fileName'],
                        'fileKey': file['fileKey'],
                        'url': download_url,
                        'recordId': record['_id']
                    })

# ファイルのダウンロードとGoogle Driveへのアップロード
for file_info in file_infos:
    # SPIRALからファイルをバイナリデータとしてダウンロード
    file_data = api_curl_action("GET", file_info['url'], headers=None, return_json=False)
    # print(file_data) //SPIRALからファイルをダウンロードできてない場合はこちらで確認

    if file_data:
        # ファイルをローカルに保存
        local_file_path = os.path.join('/tmp', file_info['fileName'])
        with open(local_file_path, 'wb') as file:
            file.write(file_data)

        # Google Driveにファイルをコピー（リトライ機能付き）
        for attempt in range(3):  # 最大3回リトライ
            try:
                drive_file_path = os.path.join(google_drive_folder_path, file_info['fileName'])
                shutil.copy2(local_file_path, drive_file_path)
                print(f"ファイル '{file_info['fileName']}' のGoogle Driveへのアップロードが成功しました。")

                # コピーが成功したらローカルファイルを削除
                os.remove(local_file_path)
                break  # 成功したのでループを抜ける
            except Exception as e:
                print(f"アップロード中にエラーが発生しました: {e}")
                time.sleep(5)  # 5秒待ってから再試行
        else:
            print(f"ファイル '{file_info['fileName']}' のGoogle Driveへのアップロードに失敗しました。")
        #アップロード完了後の処理を追加したい場合は下記へ追加してください。
        # レコード削除
        if delete_flg == 1:
            # ファイルを削除するためにSPIRALのレコードを更新
            fileKey = ""  # 空の値を指定
            recordId = file_info['recordId']
            files = []

            if fileKey:  # fileKeyが空でない場合のみリストに追加
              files.append(fileKey)

            update_data = {
            file_field_name: files
           }

            # PATCHリクエストを送信してレコードを更新
            update_url = f"/apps/{APP_ID}/dbs/{DB_ID}/records/{recordId}"
            api_response = api_curl_action("PATCH", update_url, data=update_data)
            # print(api_response) //SPIRALの更新がうまく行っていない場合はこちらを確認
            if api_response is not None:
                print(f"レコード '{recordId}' のファイル削除が成功しました。")
            else:
                print(f"レコード '{recordId}' のファイル削除に失敗しました。")

    else:
        print(f"ファイル '{file_info['fileName']}' のダウンロードに失敗しました。")
 
            

from google.colab import drive

# Google Driveをマウント
drive.mount('/content/drive')