sept_10th_final_sub_downloader.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Enter movie name: kedi\n",
      "found 1 subtitles. \n",
      "1 : Kedi - 2016\n",
      "Enter the movie number: 1\n",
      "Downloaded Subtitle in the current Directory.\n"
     ]
    }
   ],
   "source": [
    "from urllib.parse import quote\n",
    "from requests_html import HTMLSession\n",
    "import requests\n",
    "\n",
    "\n",
    "def url_to_response(url):\n",
    "    session = HTMLSession()\n",
    "    response = session.get(url)\n",
    "    return response\n",
    "\n",
    "\n",
    "def search_for_subtitle(query):\n",
    "    website_url = \"https://yts-subs.com/\"\n",
    "    query_encoded = quote(query)\n",
    "    website_search_url = website_url + \"search/\" + query_encoded\n",
    "\n",
    "    response = url_to_response(website_search_url)\n",
    "    return response\n",
    "\n",
    "\n",
    "def search_result_to_movie_info(single_search_result):\n",
    "    poster = single_search_result.find('.media-object', first=True).attrs[\"src\"]\n",
    "    title = single_search_result.find(\".media-heading\", first=True).text\n",
    "\n",
    "    year_raw = single_search_result.find(\".movinfo-section\")[0].text\n",
    "    year = int(year_raw.split()[0])\n",
    "    movie_page_raw = single_search_result.find(\".media-body\", first=True).find(\"a\", first=True)\n",
    "    movie_page = movie_page_raw.absolute_links.pop()\n",
    "\n",
    "    movie_info = {\"title\":title,\n",
    "                 \"year\": year,\n",
    "                 \"poster\": poster,\n",
    "                 \"movie_page\": movie_page}\n",
    "\n",
    "    return movie_info\n",
    "\n",
    "\n",
    "def select_movie_number(movie_info_list):\n",
    "    for i in range(len(movie_info_list)):\n",
    "        single_movie_info = movie_info_list[i]\n",
    "        print(f'{i+1} : {single_movie_info[\"title\"]} - {single_movie_info[\"year\"]}')\n",
    "\n",
    "    selected_movie = int(input(\"Enter the movie number: \"))\n",
    "    return movie_info_list[selected_movie - 1]\n",
    "              \n",
    "\n",
    "def subtitle_html_to_dict(single_subtitle):\n",
    "    subtitle_language = single_subtitle.find(\".sub-lang\", first=True).text\n",
    "\n",
    "    download_cell = single_subtitle.find('.download-cell', first=True)\n",
    "    download_page_link = download_cell.find(\"a\", first=True).absolute_links.pop()\n",
    "\n",
    "    subtitle_rating = int(single_subtitle.find(\".rating-cell\", first=True).text)\n",
    "\n",
    "    subtitle_dict = {\"language\": subtitle_language,\n",
    "                    \"download_page\": download_page_link,\n",
    "                    \"rating\": subtitle_rating}\n",
    "\n",
    "    return subtitle_dict\n",
    "\n",
    "\n",
    "def filter_subtitles(subtitle_info_list, filter_language=\"English\"):\n",
    "    results = []\n",
    "    for single_subtitle in subtitle_info_list:\n",
    "        if single_subtitle['language'] == filter_language:\n",
    "            results.append(single_subtitle)\n",
    "\n",
    "    if len(results) == 0:\n",
    "        selected_subtitle = None\n",
    "    else:\n",
    "        selected_subtitle = results[0]\n",
    "\n",
    "    return selected_subtitle\n",
    "\n",
    "\n",
    "def download_page_to_zip_url(final_subtitle):\n",
    "    response = url_to_response(final_subtitle[\"download_page\"])\n",
    "    download_link = response.html.find('.download-subtitle', first=True).attrs[\"href\"]\n",
    "    return download_link\n",
    "\n",
    "\n",
    "def download_zip_file(zip_url):\n",
    "    zip_file_data = requests.get(zip_url).content\n",
    "    filename = zip_url.split(\"/\")[-1]\n",
    "\n",
    "    with open(filename, \"wb\") as zip_file:\n",
    "        zip_file.write(zip_file_data)\n",
    "    \n",
    "    return 0\n",
    "\n",
    "\n",
    "def main():\n",
    "    query = input(\"Enter movie name: \") # take movie name from user\n",
    "    response = search_for_subtitle(query) # search for movie on website\n",
    "    search_results = response.html.find('.media-movie-clickable') # get all the results\n",
    "\n",
    "    if len(search_results) > 0:\n",
    "        print(f\"found {len(search_results)} subtitles. \")\n",
    "    else:\n",
    "        print(\"No results found.\")\n",
    "        return 0 # exit is no result is found\n",
    "\n",
    "    # take search results convert them into a dict\n",
    "    movie_info_list = list(map(search_result_to_movie_info, search_results))\n",
    "\n",
    "    # Ask user to select a movie from the list\n",
    "    selected_movie = select_movie_number(movie_info_list)\n",
    "\n",
    "    # get the selected movie page\n",
    "    movie_page_response = url_to_response(selected_movie[\"movie_page\"])\n",
    "\n",
    "    # find the list of subtitles for that movie\n",
    "    subtitle_list = movie_page_response.html.find('.high-rating')\n",
    " \n",
    "    # convert html subtitles to dict\n",
    "    subtitle_info_list = list(map(subtitle_html_to_dict, subtitle_list))\n",
    "    \n",
    "    # select subtitle based on language and rating\n",
    "    final_subtitle = filter_subtitles(subtitle_info_list)\n",
    "    \n",
    "    # get the zip file url from the movie page\n",
    "    zip_url = download_page_to_zip_url(final_subtitle)\n",
    "    \n",
    "    # download the zip file\n",
    "    download_zip_file(zip_url)\n",
    "    print(\"Downloaded Subtitle in the current Directory.\")\n",
    "          \n",
    "\n",
    "main()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "https://www.yifysubtitles.com/subtitle/fight-club-english-yify-16425.zip\n"
     ]
    }
   ],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.6.8"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}