blackpearl_template.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "blackpearl_template.ipynb",
      "provenance": [],
      "collapsed_sections": [
        "p-8c7KdYYnlC",
        "7lkepy9XGmMG"
      ]
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "accelerator": "GPU"
  },
  "cells": [
    {
      "cell_type": "code",
      "metadata": {
        "id": "3j-IAlGwZet3",
        "colab_type": "code",
        "cellView": "form",
        "colab": {}
      },
      "source": [
        "#@title Install additional packages and mount Google Drive { vertical-output: true }\n",
        "from IPython.display import clear_output\n",
        "import os\n",
        "os.system('apt install mediainfo')\n",
        "os.system('pip install git+git://github.com/AWConant/jikanpy.git')\n",
        "clear_output(wait=True)\n",
        "print('Installation finished. Now mount your Google Drive.')\n",
        "from google.colab import drive\n",
        "drive.mount('/content/drive')\n",
        "clear_output(wait=True)\n",
        "print('Drive is mounted. Setup is completed.')"
      ],
      "execution_count": 0,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "p-8c7KdYYnlC",
        "colab_type": "text"
      },
      "source": [
        "# <img src='https://i.imgur.com/N4bdPff.png' height=\"30\" alt=\"IMDB\"/> **Generate IMDB template**"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "lk2eFQgwZQsb",
        "colab_type": "code",
        "cellView": "form",
        "colab": {}
      },
      "source": [
        "import urllib.request\n",
        "import json\n",
        "import re\n",
        "import requests\n",
        "import sys\n",
        "import subprocess\n",
        "import os\n",
        "import time\n",
        "from IPython.display import clear_output\n",
        "from bs4 import BeautifulSoup\n",
        "\n",
        "# Variables\n",
        "headers = {'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10.8; rv:21.0) Gecko/20100101 Firefox/21.0'}\n",
        "#@markdown ###Insert your API key\n",
        "api_key = '' #@param {type:\"string\"}\n",
        "#@markdown ---\n",
        "#@markdown ###IMDB Information\n",
        "IMDB_URL_OR_ID = '' #@param {type:\"string\"}\n",
        "imdb_pattern = re.compile('(tt\\d{7,8})')\n",
        "youtube_pattern = re.compile('(https?://)?(www\\.)?(youtube|youtu|youtube-nocookie)\\.(com|be)/(watch\\?v=|embed/|v/|.+\\?v=)?(?P<id>[A-Za-z0-9\\-=_]{11})')\n",
        "imdb_id = re.findall(imdb_pattern, IMDB_URL_OR_ID)[0]\n",
        "\n",
        "def get_imdb_template(imdb_id):\n",
        "  imdb_url = f'http://www.omdbapi.com/?i={imdb_id}&apikey={api_key}&r=json&plot=full'\n",
        "  #@markdown ---\n",
        "  #@markdown ###Movie Information\n",
        "  Trailer = \"\" #@param {type:\"string\"}\n",
        "  Screenshot = \"\" #@param {type:\"string\"}\n",
        "  youtube_match = re.match(youtube_pattern, Trailer)\n",
        "  if Trailer != \"\":\n",
        "    if not youtube_match:\n",
        "      clear_output()\n",
        "      print('Your trailer link is invalid. Please only put in youtube link!')\n",
        "      sys.exit()\n",
        "    else:\n",
        "      trailer_enabled = True\n",
        "  else:\n",
        "    trailer_enabled = False\n",
        "\n",
        "  if Screenshot != \"\":\n",
        "    screenshot_list = Screenshot.split(\" \")\n",
        "    screenshot_enabled = True\n",
        "  else:\n",
        "    screenshot_enabled = False\n",
        "\n",
        "  Path = \"\" #@param {type:\"string\"}\n",
        "  if Path == \"\":\n",
        "    mediainfo_enabled = False\n",
        "  else:\n",
        "    mediainfo_enabled = True\n",
        "    tmp = subprocess.Popen('mediainfo --Logfile=/root/.nfo \"{}\"'.format(Path), stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)\n",
        "    time.sleep(5) # Wait 5s for the nfo file to populate properly\n",
        "    with open('/root/.nfo', 'r', encoding='utf-8') as nfo_file:\n",
        "      nfo = nfo_file.readlines()\n",
        "      del nfo[1]\n",
        "      nfo[1] = \"Complete name                            : {}\\n\".format(os.path.basename(Path))\n",
        "      nfo = ''.join(nfo)\n",
        "\n",
        "  Link = \"\" #@param {type:\"string\"}\n",
        "  if Link == \"\":\n",
        "    print(\"You forgot putting in the link to download!\")\n",
        "    sys.exit()\n",
        "\n",
        "  #@markdown ---\n",
        "  #@markdown ###Forum Requirements\n",
        "  Likes = 0 #@param {type: \"number\"}\n",
        "  Posts = 0 #@param {type: \"number\"}\n",
        "  Thanks = True #@param {type:\"boolean\"}\n",
        "\n",
        "  def get_story_line(imdb_id):\n",
        "    url = 'https://www.imdb.com/title/{}'.format(imdb_id)\n",
        "    r = requests.get(url, headers=headers)\n",
        "    data = r.text\n",
        "    soup = BeautifulSoup(data, 'html.parser')\n",
        "    story_line = soup.find_all('div', class_='inline canwrap')[0].find('span').text.lstrip()\n",
        "    return story_line\n",
        "\n",
        "  # Fetch data\n",
        "  with urllib.request.urlopen(imdb_url) as imdb_url:\n",
        "    imdb_data = json.loads(imdb_url.read().decode())\n",
        "\n",
        "  # Process data\n",
        "  text_dump = \"[center][img]{}[/img]\\n\".format(imdb_data['Poster'])\n",
        "  text_dump += \"[color=rgb(250, 197, 28)][b][size=6]{} ({})[/size][/b][/color]\\n\".format(imdb_data['Title'], imdb_data['Year'])\n",
        "  text_dump += \"[url=https://www.imdb.com/title/{}][img]https://i.imgur.com/rcSipDw.png[/img][/url][size=6][b] {}[/b]/10[/size]\".format(imdb_id, imdb_data['imdbRating'])\n",
        "  text_dump += \"[size=6] --- [img]https://i.imgur.com/sEpKj3O.png[/img]{}[/size][/center]\\n\".format(imdb_data['imdbVotes'])\n",
        "  text_dump += \"[hr][/hr][indent][size=6][color=rgb(250, 197, 28)][b]Plot Summary[/b][/color][/size][/indent]\\n\"\n",
        "  text_dump += \"{}\\n\".format(get_story_line(imdb_id))\n",
        "  if trailer_enabled:\n",
        "    text_dump += \"[hr][/hr][indent][size=6][color=rgb(250, 197, 28)][b]Trailer[/b][/color][/size][/indent]\\n\"\n",
        "    text_dump += \"[media=youtube]{}[/media]\\n\".format(youtube_match.group(6))\n",
        "  if screenshot_enabled:\n",
        "    text_dump += \"[hr][/hr][indent][size=6][color=rgb(250, 197, 28)][b]Screenshot[/b][/color][/size][/indent]\\n\"\n",
        "    text_dump += \"[spoiler='Click here to view screenshots']\\n\"\n",
        "    for i in screenshot_list:\n",
        "      text_dump += \"[img]{}[/img]\\n\".format(i)\n",
        "    text_dump += \"[/spoiler]\"\n",
        "  text_dump += \"[hr][/hr][indent][size=6][color=rgb(250, 197, 28)][b]Movie Info[/b][/color][/size][/indent]\\n\"\n",
        "  text_dump += \"[LIST][*][color=rgb(255, 204, 0)]Rating:[/color] {}\\n\".format(imdb_data['Rated'])\n",
        "  text_dump += \"[*][color=rgb(255, 204, 0)]Genre:[/color] {}\\n\".format(imdb_data['Genre'])\n",
        "  text_dump += \"[*][color=rgb(255, 204, 0)]Directed By:[/color] {}\\n\".format(imdb_data['Director'])\n",
        "  text_dump += \"[*][color=rgb(255, 204, 0)]Written By:[/color] {}\\n\".format(imdb_data['Writer'])\n",
        "  text_dump += \"[*][color=rgb(255, 204, 0)]Starring:[/color] {}\\n\".format(imdb_data['Actors'])\n",
        "  text_dump += \"[*][color=rgb(255, 204, 0)]Release Date:[/color] {}\\n\".format(imdb_data['Released'])\n",
        "  try:\n",
        "    text_dump += \"[*][color=rgb(255, 204, 0)]Runtime:[/color] {}\\n\".format(imdb_data['Runtime'])\n",
        "  except:\n",
        "    text_dump += \"[*][color=rgb(255, 204, 0)]Runtime:[/color] N/A\\n\"\n",
        "  try:\n",
        "    text_dump += \"[*][color=rgb(255, 204, 0)]On Disk/Streaming:[/color] {}\\n\".format(imdb_data['DVD'])\n",
        "  except:\n",
        "    text_dump += \"[*][color=rgb(255, 204, 0)]On Disk/Streaming:[/color] N/A\\n\"\n",
        "  try:\n",
        "    text_dump += \"[*][color=rgb(255, 204, 0)]Box Office:[/color] {}\\n\".format(imdb_data['BoxOffice'])\n",
        "  except:\n",
        "    text_dump += \"[*][color=rgb(255, 204, 0)]Box Office:[/color] N/A\\n\"\n",
        "  text_dump += \"[*][color=rgb(255, 204, 0)]Awards:[/color] {}\\n\".format(imdb_data['Awards'])\n",
        "  try:\n",
        "    text_dump += \"[*][color=rgb(255, 204, 0)]Studio:[/color] {}[/LIST]\\n\".format(imdb_data['Production'])\n",
        "  except:\n",
        "    text_dump += \"[*][color=rgb(255, 204, 0)]Studio:[/color] N/A[/LIST]\\n\"\n",
        "  if mediainfo_enabled:\n",
        "    text_dump += \"[hr][/hr][indent][size=6][color=rgb(250, 197, 28)][b]Media Info[/b][/color][/size][/indent]\\n\"\n",
        "    text_dump += \"[spoiler='Click here to view Media Info'][code]{}[/code][/spoiler]\\n\".format(nfo)\n",
        "  text_dump += \"[hr][/hr][center][size=6][color=rgb(250, 197, 28)][b]Download Link[/b][/color][/size][/center]\\n\"\n",
        "  if Thanks:\n",
        "    if Likes != 0 and Posts == 0:\n",
        "      text_dump += \"[center][hidereactscore={}][hidereact=1,2,3,4,5,6]{}[/hidereact][/hidereactscore][/center]\".format(Likes, Link)\n",
        "    elif Likes == 0 and Posts != 0:\n",
        "      text_dump += \"[center][hideposts={}][hidereact=1,2,3,4,5,6]{}[/hidereact][/hideposts][/center]\".format(Posts, Link)\n",
        "    elif Likes != 0 and Posts != 0:\n",
        "      text_dump += \"[center][hidereactscore={}][hideposts={}][hidereact=1,2,3,4,5,6]{}[/hidereact][/hideposts][/hidereactscore][/center]\".format(Likes, Posts, Link)\n",
        "    else:\n",
        "      text_dump += \"[center][hidereact=1,2,3,4,5,6]{}[/hidereact][/center]\".format(Link)\n",
        "  else:\n",
        "    if Likes != 0 and Posts == 0:\n",
        "      text_dump += \"[center][hidereactscore={}]{}[/hidereactscore][/center]\".format(Likes, Link)\n",
        "    elif Likes == 0 and Posts != 0:\n",
        "      text_dump += \"[center][hideposts={}]{}[/hideposts][/center]\".format(Posts, Link)\n",
        "    elif Likes != 0 and Posts != 0:\n",
        "      text_dump += \"[center][hidereactscore={}][hideposts={}]{}[/hideposts][/hidereactscore][/center]\".format(Likes, Posts, Link)\n",
        "    else:\n",
        "      text_dump += \"[center]{}[/center]\".format(Link)\n",
        "\n",
        "  return text_dump\n",
        "\n",
        "print(get_imdb_template(imdb_id))"
      ],
      "execution_count": 0,
      "outputs": []
    },
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "7lkepy9XGmMG",
        "colab_type": "text"
      },
      "source": [
        "# <img src='https://i.imgur.com/e5Thsob.jpg?1' height=\"25\" alt=\"MAL\"/> **Generate MAL template**"
      ]
    },
    {
      "cell_type": "code",
      "metadata": {
        "id": "SWW6mtxkGmk5",
        "colab_type": "code",
        "cellView": "form",
        "colab": {}
      },
      "source": [
        "import re\n",
        "import sys\n",
        "import os\n",
        "import time\n",
        "import subprocess\n",
        "from bs4 import BeautifulSoup\n",
        "from IPython.display import clear_output\n",
        "from jikanpy import Jikan\n",
        "\n",
        "jikan = Jikan()\n",
        "\n",
        "#@markdown ###MAL Information\n",
        "Anime_URL_OR_ID = '' #@param {type: \"string\"}\n",
        "mal_pattern = re.compile('(anime)?/(\\d{1,7})/(.+)?')\n",
        "if re.match(r'\\d+', Anime_URL_OR_ID):\n",
        "  mal_id = Anime_URL_OR_ID\n",
        "else:\n",
        "  mal_id = re.findall(mal_pattern, Anime_URL_OR_ID)[0][1]\n",
        "youtube_pattern = re.compile('(https?://)?(www\\.)?(youtube|youtu|youtube-nocookie)\\.(com|be)/(watch\\?v=|embed/|v/|.+\\?v=)?(?P<id>[A-Za-z0-9\\-=_]{11})')\n",
        "\n",
        "def get_mal_template(mal_id):\n",
        "  # Fetch data\n",
        "  anime_data = jikan.anime(mal_id)\n",
        "  #@markdown ---\n",
        "  #@markdown ###Anime Information\n",
        "  Screenshot = \"\" #@param {type:\"string\"}\n",
        "  if Screenshot != \"\":\n",
        "    screenshot_list = Screenshot.split(\" \")\n",
        "    screenshot_enabled = True\n",
        "  else:\n",
        "    screenshot_enabled = False\n",
        "\n",
        "  def get_info(anime_data, info):\n",
        "    text = ''\n",
        "    for i in range(len(anime_data[f'{info}'])):\n",
        "      text += anime_data[info][i]['name'] + ', '\n",
        "    return text[:-2]\n",
        "\n",
        "  Path = \"\" #@param {type:\"string\"}\n",
        "  if Path == \"\":\n",
        "    mediainfo_enabled = False\n",
        "  else:\n",
        "    mediainfo_enabled = True\n",
        "    tmp = subprocess.Popen('mediainfo --Logfile=/root/.nfo \"{}\"'.format(Path), stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)\n",
        "    time.sleep(5) # Wait 5s for the nfo file to populate properly\n",
        "    with open('/root/.nfo', 'r', encoding='utf-8') as nfo_file:\n",
        "      nfo = nfo_file.readlines()\n",
        "      del nfo[1]\n",
        "      nfo[1] = \"Complete name                            : {}\\n\".format(os.path.basename(Path))\n",
        "      nfo = ''.join(nfo)\n",
        "\n",
        "  Link = \"\" #@param {type:\"string\"}\n",
        "  if Link == \"\":\n",
        "    print(\"You forgot putting in the link to download!\")\n",
        "    sys.exit()\n",
        "\n",
        "  #@markdown ---\n",
        "  #@markdown ###Forum Requirements\n",
        "  Likes = 0 #@param {type: \"number\"}\n",
        "  Posts = 0 #@param {type: \"number\"}\n",
        "  Thanks = True #@param {type:\"boolean\"}\n",
        "\n",
        "  # Process data\n",
        "  text_dump = \"[center][img]{}[/img]\\n\".format(anime_data['image_url'])\n",
        "  text_dump += \"[color=rgb(250, 197, 28)][b][size=6]{} ({} - {})[/size][/b][/color]\\n\".format(anime_data['title'], anime_data['type'], anime_data['premiered'])\n",
        "  text_dump += \"[url={}][img]https://i.imgur.com/YJgzM4V.png[/img][/url][size=6][b] {}[/b]/10[/size]\".format(anime_data['url'], anime_data['score'])\n",
        "  text_dump += \"[size=6] --- [img]https://i.imgur.com/sEpKj3O.png[/img]{}[/size][/center]\\n\".format(anime_data['scored_by'])\n",
        "  text_dump += \"[hr][/hr][indent][size=6][color=rgb(250, 197, 28)][b]Synopsis[/b][/color][/size][/indent]\\n\"\n",
        "  text_dump += \"{}\\n\".format(anime_data['synopsis'])\n",
        "  try:\n",
        "    text_dump += \"[hr][/hr][indent][size=6][color=rgb(250, 197, 28)][b]Trailer[/b][/color][/size][/indent]\\n\"\n",
        "    text_dump += \"[media=youtube]{}[/media]\\n\".format(re.match(youtube_pattern, str(anime_data['trailer_url'])).group(6))\n",
        "  except:\n",
        "    pass\n",
        "  if screenshot_enabled:\n",
        "    text_dump += \"[hr][/hr][indent][size=6][color=rgb(250, 197, 28)][b]Screenshot[/b][/color][/size][/indent]\\n\"\n",
        "    text_dump += \"[spoiler='Click here to view screenshots']\\n\"\n",
        "    for i in screenshot_list:\n",
        "      text_dump += \"[img]{}[/img]\\n\".format(i)\n",
        "    text_dump += \"[/spoiler]\"\n",
        "  text_dump += \"[hr][/hr][indent][size=6][color=rgb(250, 197, 28)][b]Anime Info[/b][/color][/size][/indent]\\n\"\n",
        "  text_dump += \"[LIST][*][color=rgb(255, 204, 0)]Rating:[/color] {}\\n\".format(anime_data['rating'])\n",
        "  text_dump += \"[*][color=rgb(255, 204, 0)]Type:[/color] {}\\n\".format(anime_data['type'])\n",
        "  text_dump += \"[*][color=rgb(255, 204, 0)]Episodes:[/color] {}\\n\".format(anime_data['episodes'])\n",
        "  text_dump += \"[*][color=rgb(255, 204, 0)]Status:[/color] {}\\n\".format(anime_data['status'])\n",
        "  text_dump += \"[*][color=rgb(255, 204, 0)]Aired:[/color] {}\\n\".format(anime_data['aired']['string'])\n",
        "  text_dump += \"[*][color=rgb(255, 204, 0)]Premiered:[/color] {}\\n\".format(anime_data['premiered'])\n",
        "  text_dump += \"[*][color=rgb(255, 204, 0)]Producers:[/color] {}\\n\".format(get_info(anime_data, 'producers'))\n",
        "  text_dump += \"[*][color=rgb(255, 204, 0)]Studios:[/color] {}\\n\".format(get_info(anime_data, 'studios'))\n",
        "  text_dump += \"[*][color=rgb(255, 204, 0)]Source:[/color] {}\\n\".format(anime_data['source'])\n",
        "  text_dump += \"[*][color=rgb(255, 204, 0)]Genres:[/color] {}\\n\".format(get_info(anime_data, 'genres'))\n",
        "  text_dump += \"[*][color=rgb(255, 204, 0)]Duration:[/color] {}\\n\".format(anime_data['duration'])\n",
        "  if mediainfo_enabled:\n",
        "    text_dump += \"[hr][/hr][indent][size=6][color=rgb(250, 197, 28)][b]Media Info[/b][/color][/size][/indent]\\n\"\n",
        "    text_dump += \"[spoiler='Click here to view Media Info'][code]{}[/code][/spoiler]\\n\".format(nfo)\n",
        "  text_dump += \"[hr][/hr][center][size=6][color=rgb(250, 197, 28)][b]Download Link[/b][/color][/size][/center]\\n\"\n",
        "  if Thanks:\n",
        "    if Likes != 0 and Posts == 0:\n",
        "      text_dump += \"[center][hidereactscore={}][hidereact=1,2,3,4,5,6]{}[/hidereact][/hidereactscore][/center]\".format(Likes, Link)\n",
        "    elif Likes == 0 and Posts != 0:\n",
        "      text_dump += \"[center][hideposts={}][hidereact=1,2,3,4,5,6]{}[/hidereact][/hideposts][/center]\".format(Posts, Link)\n",
        "    elif Likes != 0 and Posts != 0:\n",
        "      text_dump += \"[center][hidereactscore={}][hideposts={}][hidereact=1,2,3,4,5,6]{}[/hidereact][/hideposts][/hidereactscore][/center]\".format(Likes, Posts, Link)\n",
        "    else:\n",
        "      text_dump += \"[center][hidereact=1,2,3,4,5,6]{}[/hidereact][/center]\".format(Link)\n",
        "  else:\n",
        "    if Likes != 0 and Posts == 0:\n",
        "      text_dump += \"[center][hidereactscore={}]{}[/hidereactscore][/center]\".format(Likes, Link)\n",
        "    elif Likes == 0 and Posts != 0:\n",
        "      text_dump += \"[center][hideposts={}]{}[/hideposts][/center]\".format(Posts, Link)\n",
        "    elif Likes != 0 and Posts != 0:\n",
        "      text_dump += \"[center][hidereactscore={}][hideposts={}]{}[/hideposts][/hidereactscore][/center]\".format(Likes, Posts, Link)\n",
        "    else:\n",
        "      text_dump += \"[center]{}[/center]\".format(Link)\n",
        "\n",
        "  return text_dump\n",
        "\n",
        "print(get_mal_template(mal_id))\n"
      ],
      "execution_count": 0,
      "outputs": []
    }
  ]
}