mirror of
https://github.com/skoobasteeve/useful-scripts.git
synced 2026-03-19 23:28:55 +00:00
141 lines
3.8 KiB
Python
141 lines
3.8 KiB
Python
#!/usr/bin/python3
|
|
|
|
import requests
|
|
from datetime import datetime
|
|
import os
|
|
import argparse
|
|
|
|
|
|
tmdb_api_token = os.environ.get("TMDB_API_TOKEN")
|
|
sa_api_token = os.environ.get("SA_API_TOKEN")
|
|
|
|
tmdb_url = "https://api.themoviedb.org/3"
|
|
tmdb_headers = {
|
|
'Authorization': f'Bearer {tmdb_api_token}',
|
|
'Content-Type': 'application/json;charset=utf-8',
|
|
'Accept': 'application/json;charset=utf-8'
|
|
}
|
|
|
|
sa_url = "https://streaming-availability.p.rapidapi.com/get/basic"
|
|
sa_headers = {
|
|
'x-rapidapi-host': "streaming-availability.p.rapidapi.com",
|
|
'x-rapidapi-key': sa_api_token
|
|
}
|
|
|
|
|
|
def get_args():
|
|
parser = argparse.ArgumentParser(
|
|
description='Search movie streaming availability.')
|
|
|
|
parser.add_argument('--year', type=int, help='Specify movie release year')
|
|
return parser.parse_args()
|
|
|
|
|
|
def tmdb_lookup(tmdb_url, tmdb_headers, movie, args):
|
|
tmdb_params = {
|
|
"language": "en-US",
|
|
"query": movie,
|
|
"page": 1,
|
|
"include_adult": False
|
|
}
|
|
|
|
if args.year:
|
|
tmdb_params["primary_release_year"] = args.year
|
|
|
|
tmdb_search = requests.get(f"{tmdb_url}/search/movie", params=tmdb_params,
|
|
headers=tmdb_headers).json()
|
|
|
|
if not tmdb_search["results"]:
|
|
print("I'm having trouble finding that movie. " +
|
|
"Check your spelling and try again.")
|
|
exit()
|
|
|
|
movie_id = tmdb_search['results'][0]['id']
|
|
movie_title = tmdb_search['results'][0]['title']
|
|
movie_release_check = tmdb_search['results'][0]['release_date']
|
|
|
|
if movie_release_check:
|
|
movie_release = datetime.strptime(
|
|
tmdb_search['results'][0]['release_date'], "%Y-%m-%d")
|
|
else:
|
|
movie_release = "???"
|
|
|
|
movie_rating = tmdb_search['results'][0]['vote_average']
|
|
|
|
return movie_id, movie_title, movie_release, movie_rating
|
|
|
|
|
|
def sa_lookup(sa_url, sa_headers, movie_id):
|
|
sa_params = {
|
|
"country": "us",
|
|
"tmdb_id": f"movie/{movie_id}",
|
|
"output_language": "en"
|
|
}
|
|
|
|
sa_request = requests.request("GET", sa_url, headers=sa_headers,
|
|
params=sa_params)
|
|
|
|
if sa_request.status_code == 404:
|
|
print("I'm having trouble finding that movie. " +
|
|
"Check your spelling and try again.")
|
|
exit()
|
|
|
|
sa_response = sa_request.json()
|
|
services = sa_response["streamingInfo"]
|
|
|
|
return sa_response, services
|
|
|
|
|
|
def services_speller(service):
|
|
if service == "hbo":
|
|
service_proper = "HBO Max"
|
|
elif service == "hulu":
|
|
service_proper = "Hulu"
|
|
elif service == "prime":
|
|
service_proper = "Amazon Prime"
|
|
elif service == "netflix":
|
|
service_proper = "Netflix"
|
|
elif service == "disney":
|
|
service_proper = "Disney+"
|
|
elif service == "apple":
|
|
service_proper = "Apple TV+"
|
|
elif service == "paramount":
|
|
service_proper = "Paramount+"
|
|
else:
|
|
return service
|
|
return service_proper
|
|
|
|
|
|
def main():
|
|
|
|
args = get_args()
|
|
movie = input("Enter a movie: ")
|
|
|
|
movie_id, movie_title, movie_release, movie_rating = tmdb_lookup(
|
|
tmdb_url, tmdb_headers, movie, args)
|
|
|
|
print(movie_title + f" ({movie_release.year})")
|
|
print(f"Rating: {movie_rating}\n")
|
|
|
|
sa_response, services = sa_lookup(sa_url, sa_headers, movie_id)
|
|
|
|
if not services:
|
|
print("Streaming not available :(")
|
|
|
|
for s in services:
|
|
leaving_epoch = sa_response["streamingInfo"][s]["us"]["leaving"]
|
|
leaving_date = datetime.fromtimestamp(
|
|
int(leaving_epoch)).strftime('%Y-%m-%d')
|
|
link = sa_response["streamingInfo"][s]["us"]["link"]
|
|
|
|
print(f"Available on {services_speller(s)}")
|
|
|
|
if leaving_epoch != 0:
|
|
print(f"Will be leaving on {leaving_date}")
|
|
|
|
print(f"Watch here: {link}\n")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|