def series_modal(request): content_discovery = ContentDiscovery() content_manager = ContentManager() report_modal = request.GET.get("report_modal", False) # Get the ID from the URL tmdb_id = request.GET.get("tmdb_id", None) tvdb_id = request.GET.get("tvdb_id", None) # Determine the TVDB ID if tvdb_id: pass elif tmdb_id: tvdb_id = content_discovery.get_external_ids(tmdb_id, "tv")["tvdb_id"] # Check if the show is already within Sonarr's collection requested_show = content_manager.get(tvdb_id=tvdb_id) # If it doesn't already exists, add then add it if requested_show is None: sonarr_params = obtain_sonarr_parameters(content_discovery, content_manager, tmdb_id, tvdb_id) requested_show = content_manager.add( tvdb_id=tvdb_id, quality_profile_id=sonarr_params["sonarr_profile_id"], root_dir=sonarr_params["sonarr_root"], series_type=sonarr_params["series_type"], season_folders=sonarr_params["season_folders"], ) # Keep refreshing until we get the series from Sonarr series = content_manager.get(tvdb_id=tvdb_id, obtain_season_info=True, force_update_cache=True) if series is None: series_fetch_retries = 0 while series is None: if series_fetch_retries > MAX_SERIES_FETCH_RETRIES: break series_fetch_retries = series_fetch_retries + 1 series = content_manager.get(tvdb_id=tvdb_id, obtain_season_info=True, force_update_cache=True) log.handler( "Sonarr did not have the series information! Conreq is waiting...", log.INFO, __logger, ) context = generate_context({ "seasons": series["seasons"], "tvdb_id": tvdb_id, "report_modal": report_modal }) template = loader.get_template("modal/series_selection.html") return HttpResponse(template.render(context, request))
def content_preview_modal(request): content_discovery = ContentDiscovery() # Get the ID from the URL tmdb_id = request.GET.get("tmdb_id", None) content_type = request.GET.get("content_type", None) if tmdb_id and content_type: # Fetch and process content = content_discovery.get_by_tmdb_id(tmdb_id, content_type, obtain_extras=False) set_single_availability(content) preprocess_tmdb_result(content) # Check if the user has already requested this requested = False if UserRequest.objects.filter( content_id=content["id"], source="tmdb", content_type=content["content_type"], ): requested = True context = generate_context({ "content": content, "requested": requested }) template = loader.get_template("modal/content_preview.html") return HttpResponse(template.render(context, request))
def initialization(request): conreq_config = ConreqConfig.get_solo() user_objects = get_user_model().objects # Authenticate using Organizr headers organizr_username = request.headers.get("X-WEBAUTH-USER") if conreq_config.conreq_http_header_auth and organizr_username: organizr_email = request.headers.get("X-WEBAUTH-EMAIL") organizr_group = int(request.headers.get("X-WEBAUTH-GROUP")) user = user_objects.get_or_create(username=organizr_username, )[0] user.email = organizr_email user.is_staff = False if organizr_group == 0: user.is_superuser = True if organizr_group == 0 or organizr_group == 1: user.is_staff = True user.save() login(request, user) # Run the first time initialization if needed if conreq_config.conreq_initialized is False: # User submitted the first time setup form if request.method == "POST": form = InitializationForm(request.POST) # Create the superuser and set up the database if the form is valid if form.is_valid(): form.save() username = form.cleaned_data.get("username") password = form.cleaned_data.get("password1") user = authenticate(username=username, password=password) user.is_staff = True user.is_admin = True user.is_superuser = True user.save() login(request, user) initialize_conreq(conreq_config, form) return redirect("base:index") # Form data wasn't valid, so return the error codes template = loader.get_template("registration/initialization.html") return HttpResponse(template.render({"form": form}, request)) # User needs to fill out the first time setup template = loader.get_template("registration/initialization.html") return HttpResponse(template.render({}, request)) # If a base URL is set, redirect the user to it if request.path[1:] != BASE_URL: return redirect("/" + BASE_URL) # Render the base return login_required(render)(request, "primary/base.html", generate_context({}))
def report_issue_modal(request): # Get the parameters from the URL context = generate_context( { "issues": ISSUE_LIST, "tmdb_id": request.GET.get("tmdb_id", None), "tvdb_id": request.GET.get("tvdb_id", None), "content_type": request.GET.get("content_type", None), } ) template = loader.get_template("modal/report_issue.html") return HttpResponse(template.render(context, request))
def series_modal(user, tmdb_id=None, tvdb_id=None): # Validate login status if user.is_authenticated: content_discovery = ContentDiscovery() content_manager = ContentManager() # Determine the TVDB ID if tvdb_id is not None: pass elif tmdb_id is not None: tvdb_id = content_discovery.get_external_ids(tmdb_id, "tv")["tvdb_id"] # Check if the show is already within Sonarr's collection requested_show = content_manager.get(tvdb_id=tvdb_id) # If it doesn't already exists, add then add it if requested_show is None: sonarr_params = obtain_sonarr_parameters(content_discovery, content_manager, tmdb_id, tvdb_id) requested_show = content_manager.add( tvdb_id=tvdb_id, quality_profile_id=sonarr_params["sonarr_profile_id"], root_dir=sonarr_params["sonarr_root"], series_type=sonarr_params["series_type"], season_folders=sonarr_params["season_folders"], ) # Keep refreshing until we get the series from Sonarr series = content_manager.get(tvdb_id=tvdb_id, obtain_season_info=True) if series is None: series_fetch_retries = 0 while series is None: if series_fetch_retries > MAX_SERIES_FETCH_RETRIES: break series_fetch_retries = series_fetch_retries + 1 sleep(0.5) series = content_manager.get(tvdb_id=tvdb_id, obtain_season_info=True, force_update_cache=True) log.handler( "Retrying content fetch!", log.INFO, __logger, ) context = generate_context({"seasons": series["seasons"]}) return render_to_string("modal/series_selection.html", context)
def collection(request): collection_id = request.GET.get("collection_id", None) if collection_id: content_discovery = ContentDiscovery() tmdb_collection = content_discovery.collections(collection_id) if not isinstance(tmdb_collection, dict) or not tmdb_collection: tmdb_collection = None else: set_many_availability(tmdb_collection["parts"]) context = generate_context({"collection": tmdb_collection}) template = loader.get_template("viewport/more_info_collection.html") return HttpResponse(template.render(context, request))
def recommended(request): tmdb_id = request.GET.get("tmdb_id", None) content_type = request.GET.get("content_type", None) if tmdb_id and content_type: content_discovery = ContentDiscovery() tmdb_recommended = content_discovery.similar_and_recommended( tmdb_id, content_type) if not isinstance(tmdb_recommended, dict) or not tmdb_recommended: tmdb_recommended = None else: set_many_availability(tmdb_recommended["results"]) context = generate_context({"recommended": tmdb_recommended}) template = loader.get_template("viewport/more_info_recommended.html") return HttpResponse(template.render(context, request))
def discover_movies(request): content_discovery = ContentDiscovery() template = loader.get_template("viewport/discover.html") # Get the page number from the URL page = int(request.GET.get("page", 1)) # Search for TV content tmdb_results = content_discovery.movies(page)["results"] # Set the availability for all cards set_many_availability(tmdb_results) context = generate_context({ "all_cards": tmdb_results, }) return HttpResponse(template.render(context, request))
def discover_tv(request): content_discovery = ContentDiscovery() template = loader.get_template("viewport/discover.html") # Get the page number from the URL page = int(request.GET.get("page", 1)) # Search for TV content tmdb_results = content_discovery.tv(page, page_multiplier=2)["results"] # Set conreq status for all cards set_many_conreq_status(tmdb_results) context = generate_context({ "all_cards": tmdb_results, }) return HttpResponse(template.render(context, request))
def search(request): content_discovery = ContentDiscovery() searcher = Search() # Get the ID from the URL query = request.GET.get("query", "") content_type = request.GET.get("content_type", None) template = loader.get_template("viewport/search.html") # Determine which search method to use (tv/movie/all) if content_type == "tv": arr_results = searcher.television(query) elif content_type == "movie": arr_results = searcher.movie(query) else: arr_results = searcher.all(query) # Attempt to convert cards to TMDB equivalents thread_list = [] for index in range(0, len(arr_results)): thread = Thread(target=convert_card_to_tmdb, args=[index, arr_results]) thread.start() thread_list.append(thread) # Wait for computation to complete for thread in thread_list: thread.join() # Generate conreq status content_discovery.determine_id_validity({"results": arr_results}) set_many_conreq_status(arr_results) context = generate_context( { "all_cards": arr_results, } ) return HttpResponse(template.render(context, request))
def more_info(request): content_discovery = ContentDiscovery() template = loader.get_template("viewport/more_info.html") thread_list = [] # Get the ID from the URL tmdb_id = request.GET.get("tmdb_id", None) tvdb_id = request.GET.get("tvdb_id", None) if tmdb_id: content_type = request.GET.get("content_type", None) # Get all the basic metadata for a given ID content = content_discovery.get_by_tmdb_id(tmdb_id, content_type) # Determine the availability of the current TMDB ID set_single_availability(content) # Pre-process data attributes within tmdb_result preprocess_tmdb_result(content) # Get collection information if (content.__contains__("belongs_to_collection") and content["belongs_to_collection"] is not None): tmdb_collection_id = content["belongs_to_collection"]["id"] else: tmdb_collection_id = None # Check if the user has already requested this requested = False if UserRequest.objects.filter( content_id=content["id"], source="tmdb", content_type=content["content_type"], ): requested = True # Generate context for page rendering context = generate_context({ "content": content, "collection_id": tmdb_collection_id, "content_type": content["content_type"], "requested": requested, }) elif tvdb_id: searcher = Search() # Fallback for TVDB content = searcher.television(tvdb_id)[0] thread_list = [] # Preprocess results preprocess_arr_result(content) # Determine the availability set_single_availability(content) # Generate context for page rendering context = generate_context({ "content": content, "content_type": content["contentType"], }) # Render the page return HttpResponse(template.render(context, request))
def more_info(request): content_discovery = ContentDiscovery() template = loader.get_template("viewport/more_info.html") thread_list = [] # Get the ID from the URL tmdb_id = request.GET.get("tmdb_id", None) tvdb_id = request.GET.get("tvdb_id", None) if tmdb_id: content_type = request.GET.get("content_type", None) # Get all the basic metadata for a given ID content = content_discovery.get_by_tmdb_id(tmdb_id, content_type) # Get recommended results similar_and_recommended_thread = ReturnThread( target=content_discovery.similar_and_recommended, args=[tmdb_id, content_type], ) similar_and_recommended_thread.start() # Determine the availability of the current TMDB ID thread = Thread(target=set_single_availability, args=[content]) thread.start() thread_list.append(thread) # Pre-process data attributes within tmdb_result thread = Thread(target=preprocess_tmdb_result, args=[content]) thread.start() thread_list.append(thread) # Get collection information if (content.__contains__("belongs_to_collection") and content["belongs_to_collection"] is not None): tmdb_collection = True tmdb_collection_thread = ReturnThread( target=content_discovery.collections, args=[content["belongs_to_collection"]["id"]], ) tmdb_collection_thread.start() else: tmdb_collection = None # Recommended content tmdb_recommended = similar_and_recommended_thread.join() if not isinstance(tmdb_recommended, dict) or len(tmdb_recommended) == 0: tmdb_recommended = None # Determine the availability for all recommended content thread = Thread(target=set_many_availability, args=[tmdb_recommended["results"]]) thread.start() thread_list.append(thread) # Wait for thread computation to complete for thread in thread_list: thread.join() if tmdb_collection is not None: tmdb_collection = tmdb_collection_thread.join() # Check if the user has already requested this requested = False if UserRequest.objects.filter( content_id=content["id"], source="tmdb", content_type=content["content_type"], ): requested = True # Generate context for page rendering context = generate_context({ "content": content, "recommended": tmdb_recommended, "collection": tmdb_collection, "content_type": content["content_type"], "requested": requested, }) elif tvdb_id: searcher = Search() # Fallback for TVDB content = searcher.television(tvdb_id)[0] thread_list = [] # Preprocess results thread = Thread(target=preprocess_arr_result, args=[content]) thread.start() thread_list.append(thread) # Determine the availability thread = Thread(target=set_single_availability, args=[content]) thread.start() thread_list.append(thread) # Wait for thread computation to complete for thread in thread_list: thread.join() # Generate context for page rendering context = generate_context({ "content": content, "content_type": content["contentType"], }) # Render the page return HttpResponse(template.render(context, request))
def all_requests(request): template = loader.get_template("viewport/requests.html") content_discovery = ContentDiscovery() content_manager = ContentManager() user_requests = UserRequest.objects.all().order_by("id").reverse() all_cards = [] request_dict = {} for entry in user_requests.values( "content_id", "source", "content_type", "requested_by__username", ): # Fetch TMDB entry if entry["source"] == "tmdb" and request_is_unique( entry, request_dict): card = content_discovery.get_by_tmdb_id( tmdb_id=entry["content_id"], content_type=entry["content_type"], obtain_extras=False, ) if card is not None: card["tmdbCard"] = True card["requested_by"] = entry["requested_by__username"] all_cards.append(card) # Fetch TVDB entry if entry["source"] == "tvdb" and request_is_unique( entry, request_dict): # Attempt to convert card to TMDB conversion = content_discovery.get_by_tvdb_id( tvdb_id=entry["content_id"]) # Conversion found if conversion.__contains__( "tv_results") and conversion["tv_results"]: card = conversion["tv_results"][0] card["tmdbCard"] = True card["requested_by"] = entry["requested_by__username"] all_cards.append(card) # Convert all requests to use this new ID old_requests = UserRequest.objects.filter( content_id=entry["content_id"], source="tvdb") old_requests.update(content_id=card["id"], source="tmdb") # Fallback to checking sonarr's database else: card = content_manager.get(tvdb_id=entry["content_id"]) # Determine if the card has a known poster image if isinstance(card, dict): card["contentType"] = entry["content_type"] if card.__contains__("images"): remote_poster = is_key_value_in_list("coverType", "poster", card["images"], return_item=True) if remote_poster: card["remotePoster"] = remote_poster["remoteUrl"] card["requested_by"] = entry["requested_by__username"] all_cards.append(card) if card is None and not request_is_unique(entry, request_dict): log.handler( entry["content_type"] + " from " + entry["source"] + " with ID " + entry["content_id"] + " no longer exists!", log.WARNING, __logger, ) # Set the availability content_discovery.determine_id_validity({"results": all_cards}) set_many_availability(all_cards) context = generate_context({"all_cards": all_cards}) return HttpResponse(template.render(context, request))
def homepage(request): # Generate the base template template = loader.get_template("primary/base.html") context = generate_context({}) return HttpResponse(template.render(context, request))
def manage_users(request): template = loader.get_template("viewport/manage_users.html") users = get_user_model().objects.values() context = generate_context({"users": users}) return HttpResponse(template.render(context, request))
def my_issues(request): # Get the parameters from the URL content_discovery = ContentDiscovery() content_manager = ContentManager() reported_issues = ( ReportedIssue.objects.filter(reported_by=request.user).order_by("id").reverse() ) all_cards = [] for entry in reported_issues.values( "reported_by__username", "content_id", "source", "resolved", "content_type", "issues", "seasons", "episodes", ): # Fetch TMDB entry if entry["source"] == "tmdb": card = content_discovery.get_by_tmdb_id( tmdb_id=entry["content_id"], content_type=entry["content_type"], obtain_extras=False, ) if card is not None: card["tmdbCard"] = True all_cards.append({**card, **entry}) # Fetch TVDB entry if entry["source"] == "tvdb": # Attempt to convert card to TMDB conversion = content_discovery.get_by_tvdb_id(tvdb_id=entry["content_id"]) # Conversion found if conversion.__contains__("tv_results") and conversion["tv_results"]: card = conversion["tv_results"][0] card["tmdbCard"] = True all_cards.append({**card, **entry}) # Convert all requests to use this new ID old_requests = ReportedIssue.objects.filter( content_id=entry["content_id"], source="tvdb" ) old_requests.update(content_id=card["id"], source="tmdb") # Fallback to checking sonarr's database else: card = content_manager.get(tvdb_id=entry["content_id"]) all_cards.append({**card, **entry}) if card is None: log.handler( entry["content_type"] + " from " + entry["source"] + " with ID " + entry["content_id"] + " no longer exists!", log.WARNING, __logger, ) context = generate_context({"all_cards": all_cards}) template = loader.get_template("viewport/reported_issues.html") return HttpResponse(template.render(context, request))