magnetfinder.py

import bs4 as bs
import re
import random
import requests
import string
import os
import operator
import lxml
from subprocess import call
from prettytable import PrettyTable
from pathlib import Path
from configparser import ConfigParser
import platform
import getopt, sys

config = ConfigParser()
config.read('config.ini')

class Torrent_Link():

    def name(self, name):
        self.name = name

    def magnetlink(self, link):
        self.magnet = link

    def size(self, size):
        self.size = size

    def seeders(self, seeders):
        self.seeders = seeders


def choose_torrent_website():
    accepted_links = ['nyaa', 'piratebay']
    while(True):
        choice = input('Torrent Site (nyaa, piratebay, or \'all\'): ')
        choice = choice.lower()
        if(choice in accepted_links):
            return choice
        if(choice == 'all'):
            return choice
        print('Choose one of the available websites (or \'all\'):', *accepted_links)



def display_torrents_found(amount):
    x = PrettyTable()
    x.field_names = ["#", "Name", "Size", "Seeders"]
    x.align['Name'] = 'l'
    x.align['Size'] = 'l'
    x.align['Seeders'] = 'l'
    for num, t in enumerate(top_torrents[(amount-10):amount], amount-9):
        x.add_row([num, t.name, t.size, t.seeders])
        t.magnet = t.magnet.split('&tr=', 1)[0]
    print(x)
    selected = input('Torrent num(s)? (n for next 10): ')
    return selected

def grab_nyaa():
    link = 'https://nyaa.si/?f=0&c=0_0&q='
    sortbyseeders = '&s=seeders&o=desc'
    search_query = re.sub(r"\s+", "+", original_query)
    link = ''.join([link, search_query, sortbyseeders]) 
    data = requests.get(link).text
    soup = bs.BeautifulSoup(data, "lxml")

    for torrent in soup.find_all('tr'): 
        currentTorrent = Torrent_Link()
        for link in torrent.find_all('a'):
            if(link.get('title') != None):
                if(all((word.lower() in link.get('title').lower() for word in original_query.split()))):
                    currentTorrent.name(link.get('title'))
                    top_torrents.append(currentTorrent)
            if('magnet' in link['href'].lower()):
                currentTorrent.magnetlink(link['href'])
        for info in torrent.find_all('td', {'class': 'text-center'})[:4]:
            if(not '-' in info.text):
                if('B' in info.text):
                    currentTorrent.size(info.text)
                else:
                    currentTorrent.seeders = info.text

    return top_torrents

def grab_piratebay():
    link = 'https://www.tpb.party/search/'
    sortbyseeders = ''
    search_query = re.sub(r"\s+", "%20", original_query) +'/1/99/0'
    link = ''.join([link, search_query, sortbyseeders]) 
    data = requests.get(link).text
    soup = bs.BeautifulSoup(data, "lxml")

    for row in soup.find_all('tr'):
        currentTorrent = Torrent_Link()
        for torrent in row.find_all('div', {'class': 'detName'}):
            currentTorrent.name((torrent.find('a', {'class': 'detLink'})['title'])[11:])
            top_torrents.append(currentTorrent)
        for mag in row.find_all('a', href=True):
            if('magnet' in mag['href'].lower()):
                currentTorrent.magnetlink((mag['href']))
        if(row.find('font', {'class': 'detDesc'}) is not None):
            desc = row.find('font', {'class': 'detDesc'}).text
            try:
                size = re.search('Size(.+?),', desc)
                currentTorrent.size(size.group(1))
            except:
                currentTorrent.size('#')
        if(row.find('td', {'align': 'right'}) is not None):
            currentTorrent.seeders = row.find('td', {'align': 'right'}).text
        else:
            currentTorrent.seeders = '#'

    return top_torrents

def handle_anime_directories():
    if(type_of_media.lower() == 'a' or type_of_media.lower() == 'anime'):
        series_status = input('On-Going or Finished Series? (O / F): ')
        if(series_status.lower() == 'o'):
            show_list = open('ongoing_directories.txt', 'a+')
            new_show = input('New Folder? (Y/N): ')
            if new_show.lower() == 'y':
                new_show_name = input('Enter Name of New Folder: ')
                show_list.write(new_show_name +'\n')
                show_directory_name = new_show_name
                show_list.close()
                if(platform.system().lower() == 'windows'):
                    directory = Path(config['directories']['anime'] + show_directory_name)
                else:
                    directory = Path(config['directories']['anime'] + re.sub(r"\s+", "\\ ", show_directory_name))
            else:
                with open('ongoing_directories.txt', 'r') as f:
                    all_ongoing_shows = f.read().splitlines()
                for num, show in enumerate(all_ongoing_shows):
                    print(f'({num+1}) {show}')
                show_choice = input('Choose which show to add too (1-#): ')
                show_directory_name = all_ongoing_shows[int(show_choice)-1]
                if(platform.system().lower() == 'windows'):
                    directory = Path(config['directories']['anime'] + show_directory_name)
                else:
                    directory = Path(config['directories']['anime'] + re.sub(r"\s+", "\\ ", show_directory_name))
        else:
            directory = Path(config['directories']['anime'])
    elif(type_of_media.lower() == 'm' or type_of_media.lower() == 'movie'):
        directory = Path(config['directories']['movies'])

    return directory

def autodownload():
    if(vpnarg and vpnarg.lower() == 'true'):
        try:
            call(['sudo', 'nordvpn', 'connect'])
        except:
            print("failed to run NordVPN")

    elif(config['vpn']['status'].lower() == 'true'):
        try:
            call(['sudo', 'nordvpn', 'connect'])
        except:
            print("failed to run NordVPN")
    for num in selected.split():
        try:
            if(config['torrent-client']['client'].lower() == 'aria2c'):
                call(['aria2c', '-d', f'{Path.home().joinpath(directory)}', '--seed-time=0', f'{top_torrents[int(num) - 1].magnet}'])
                
            elif(config['torrent-client']['client'].lower() == 'deluge'):
                call(['sudo', 'deluge-console', 'add', '-p', f'{directory}', top_torrents[int(num) - 1].magnet])
            
            elif(config['torrent-client']['client'].lower() == 'transmission'):
                print(f'{Path.home().joinpath(directory)}')
                call(['transmission-remote', '-w', f'{Path.home().joinpath(directory)}', '-a', top_torrents[int(num) - 1].magnet])
                
        except:
            print(f"Error auto-downloading torrent from selected torrent client {config['torrent-client']['client']}")
            print(top_torrents[int(num)-1].magnet)
            



if __name__ == '__main__':
    choice = choose_torrent_website()
    type_of_media = input('Anime Series, Movie or TV Show? (A / M / T): ')
    original_query = input('Search Query: ')
    top_torrents = []
    directory = None
    full_cmd_args = sys.argv
    arg_list = full_cmd_args[1:]

    short_options = "v:"
    long_options = "vpn="
    vpnarg = None

    try:
        arguments, values = getopt.getopt(arg_list, short_options, long_options)
    except getopt.error as err:
        print(str(err))
        sys.exit(2)

    for current_argument, current_value in arguments:
        if current_argument in ("-v", "--vpn"):
            vpnarg = current_value

    if(choice == 'nyaa'):

        #oranizing folders and saves folder locations if its an ongoing series
        directory = handle_anime_directories()

        #finds the top 50 magnet links from nyaa
        grab_nyaa()

    if(choice =='piratebay'):

        if(type_of_media.lower() == 't' or type_of_media.lower() == 'tv show'):
            directory = Path(config['directories']['tvshows'])
        elif(type_of_media.lower() == 'm' or type_of_media.lower() == 'movie'):
            directory = Path(config['directories']['movies'])
        elif(type_of_media.lower() == 'a' or type_of_media.lower() == 'anime'):
            directory = handle_anime_directories()

        #finds the top 50 magnet links from piratebay
        grab_piratebay()

    if(choice == 'all'):

        if(type_of_media.lower() == 't' or type_of_media.lower() == 'tv show'):
            directory = Path(config['directories']['tvshows'])
        elif(type_of_media.lower() == 'm' or type_of_media.lower() == 'movie'):
            directory = Path(config['directories']['movies'])
        elif(type_of_media.lower() == 'a' or type_of_media.lower() == 'anime'):
            directory = handle_anime_directories()

        grab_nyaa()
        grab_piratebay()

        #manually sorts torrents if they are from multiple websites
        top_torrents.sort(key=lambda x: int(x.seeders), reverse=True)

    selected = 'n'
    amount = 10

    #displays torrents in a prettytable and gets user selection
    while(selected.lower() == 'n'):
        selected = display_torrents_found(amount)
        amount = amount + 10

        if(amount > len(top_torrents)+9):
            amount = 10
            print('All torrents displayed, please select one or multiple (Ex: 1 5 15)')

    #if user configured autodownload to be true, will call torrent client to download selected torrents
    if(config['autodownload']['status'].lower() == 'true'):
        autodownload()

    else:
        for num in selected.split():
            print(top_torrents[int(num)-1].magnet)