mirror of
https://github.com/j93es/browser-use-oauth.git
synced 2026-06-04 02:41:53 +09:00
90 lines
4.5 KiB
Python
90 lines
4.5 KiB
Python
import os
|
||
import subprocess
|
||
|
||
os.makedirs(os.path.dirname("./data"), exist_ok=True)
|
||
|
||
def create_file_from_example(target: str, example: str) -> bool:
|
||
if not os.path.exists(target):
|
||
if os.path.exists(example):
|
||
with open(example, 'r', encoding='utf-8') as example_file, \
|
||
open(target, 'w', encoding='utf-8') as target_file:
|
||
target_file.write(example_file.read())
|
||
os.startfile(target)
|
||
print(f"✅ {target} 파일이 {example}에서 생성되었습니다.")
|
||
return True
|
||
else:
|
||
print(f"⚠️ {example} 파일이 존재하지 않습니다. {target} 생성에 실패했습니다.")
|
||
else:
|
||
print(f"ℹ️ {target} 파일이 이미 존재합니다.")
|
||
return False
|
||
|
||
|
||
def install_playwright_chrome():
|
||
print("\n🛠️ Playwright의 Chrome을 설치 중입니다...")
|
||
try:
|
||
subprocess.run(['uv', 'run', 'playwright', 'install', 'chrome'], check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
||
print("✅ Playwright Chrome 설치 완료.")
|
||
except subprocess.CalledProcessError as e:
|
||
if "already" in e.stdout.decode():
|
||
print("ℹ️ Chrome이 이미 설치되어 있습니다.")
|
||
else:
|
||
print(f"❌ Playwright 설치 실패: {e}")
|
||
print("\n")
|
||
|
||
|
||
def prompt_yes_no(message: str) -> bool:
|
||
print(message, end="")
|
||
return input().strip().lower() in ['y', 'yes']
|
||
|
||
|
||
def setup_storage():
|
||
print("\n🔧 쿠키와 로컬 스토리지를 설정하시겠습니까?")
|
||
print("👀 다음 단계에서 Senstive Data를 설정할 수 있지만 쿠키와 로컬 스토리지를 더 권장합니다.")
|
||
if prompt_yes_no("\033[1m\033[33m선택하시려면 y를 입력하세요 (y/n):\033[0m "):
|
||
print("======================================================")
|
||
print("👀 원하는 OAuth Providor를 직접 모두 로그인 한 후에 브라우저를 닫으면 설정이 완료됩니다.")
|
||
os.system('uv run playwright open https://google.com/ --save-storage=./data/storage_state.json')
|
||
if os.path.exists('./data/storage_state.json'):
|
||
os.startfile('./data/storage_state.json')
|
||
print("✅ 쿠키와 로컬 스토리지 설정 완료.")
|
||
print("💾 ./data/storage_state.json 파일이 생성되었습니다.")
|
||
else:
|
||
print("🚫 쿠키와 로컬 스토리지 설정이 취소되었습니다.")
|
||
print("======================================================")
|
||
print("⚠️ 이후에 쿠키와 로컬 스토리지를 설정하려면, `uv run playwright open https://google.com/ --save-storage=./data/storage_state.json` 명령어를 사용하세요.\n")
|
||
|
||
|
||
def setup_sensitive():
|
||
print("\n🔐 Sensitive Data을 설정하시겠습니까?")
|
||
print("👉 이미 세션을 설정했다면, 이 작업은 **선택사항**입니다.")
|
||
print("⚠️ 민감 정보 파일은 오류를 유발하거나 문제가 될 수 있으므로 가급적 세션 사용을 권장합니다.")
|
||
if prompt_yes_no("\033[1m\033[33m선택하시려면 y를 입력하세요 (y/n):\033[0m "):
|
||
print("======================================================")
|
||
print("👀 .sensitive.json 파일을 생성합니다.")
|
||
print("💾 Browser Use의 문서를 참조하여 수정을 수정해주세요.")
|
||
print("https://docs.browser-use.com/customize/sensitive-data")
|
||
create_file_from_example('.sensitive.json', '.sensitive.example.json')
|
||
print("======================================================")
|
||
print("✅ .sensitive.json 파일이 생성되었습니다.")
|
||
else:
|
||
print("🚫 .sensitive.json 생성이 취소되었습니다.")
|
||
print("======================================================")
|
||
print("⚠️ 이후에 민감 정보 파일을 설정하려면, .sensitive.example.json 파일을 참고하여 .sensitive.json 파일을 생성하세요.\n")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
# 1. .env 생성
|
||
create_file_from_example('.env', '.env.example')
|
||
print("=====================================================")
|
||
# 2. Playwright용 Chrome 설치
|
||
install_playwright_chrome()
|
||
print("=====================================================")
|
||
|
||
# 3. 쿠키와 로컬 스토리지 설정
|
||
setup_storage()
|
||
print("=====================================================")
|
||
|
||
# 4. .sensitive.json 생성
|
||
setup_sensitive()
|
||
print("=====================================================")
|
||
print("🎉 초기 설정이 완료되었습니다! 이제 스크립트를 실행할 준비가 되었습니다.")
|