File size: 3,346 Bytes
8725d0d 6e5685b 8725d0d 0102c69 6e5685b 0102c69 6e5685b 8725d0d 6e5685b 8725d0d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 |
# tvdb.py
import os
import requests
import urllib.parse
from datetime import datetime, timedelta
from dotenv import load_dotenv
import json
from hf_scrapper import get_system_proxies
load_dotenv()
THETVDB_API_KEY = os.getenv("THETVDB_API_KEY")
THETVDB_API_URL = os.getenv("THETVDB_API_URL")
CACHE_DIR = os.getenv("CACHE_DIR")
TOKEN_EXPIRY = None
THETVDB_TOKEN = None
proxies = get_system_proxies()
def authenticate_thetvdb():
global THETVDB_TOKEN, TOKEN_EXPIRY
auth_url = f"{THETVDB_API_URL}/login"
auth_data = {
"apikey": THETVDB_API_KEY
}
try:
response = requests.post(auth_url, json=auth_data, proxies=proxies)
response.raise_for_status()
response_data = response.json()
THETVDB_TOKEN = response_data['data']['token']
TOKEN_EXPIRY = datetime.now() + timedelta(days=30)
except requests.RequestException as e:
print(f"Authentication failed: {e}")
THETVDB_TOKEN = None
TOKEN_EXPIRY = None
def get_thetvdb_token():
global THETVDB_TOKEN, TOKEN_EXPIRY
if not THETVDB_TOKEN or datetime.now() >= TOKEN_EXPIRY:
authenticate_thetvdb()
return THETVDB_TOKEN
def fetch_and_cache_json(original_title, title, media_type, year=None):
if year:
search_url = f"{THETVDB_API_URL}/search?query={urllib.parse.quote(title)}&type={media_type}&year={year}"
else:
search_url = f"{THETVDB_API_URL}/search?query={urllib.parse.quote(title)}&type={media_type}"
token = get_thetvdb_token()
if not token:
print("Authentication failed")
return
headers = {
"Authorization": f"Bearer {token}",
"accept": "application/json",
}
try:
# Fetch initial search results
response = requests.get(search_url, headers=headers, proxies=proxies)
response.raise_for_status()
data = response.json()
if 'data' in data and data['data']:
# Extract the TVDB ID and type from the first result
first_result = data['data'][0]
tvdb_id = first_result.get('tvdb_id')
media_type = first_result.get('type')
if not tvdb_id:
print("TVDB ID not found in the search results")
return
# Determine the correct extended URL based on media type
if media_type == 'movie':
extended_url = f"{THETVDB_API_URL}/movies/{tvdb_id}/extended?meta=translations"
elif media_type == 'series':
extended_url = f"{THETVDB_API_URL}/series/{tvdb_id}/extended?meta=translations"
else:
print(f"Unsupported media type: {media_type}")
return
# Request the extended information using the TVDB ID
response = requests.get(extended_url, headers=headers, proxies=proxies)
response.raise_for_status()
extended_data = response.json()
# Cache the extended JSON response
json_cache_path = os.path.join(CACHE_DIR, f"{urllib.parse.quote(original_title)}.json")
with open(json_cache_path, 'w') as f:
json.dump(extended_data, f)
except requests.RequestException as e:
print(f"Error fetching data: {e}")
|