assKick.py

#! -*- coding: utf-8 -*-

from bs4 import BeautifulSoup
import os
import re
import requests
import subprocess
import sys
import tabulate

class OutColors:
  DEFAULT='\033[0m'
  BW='\033[1m'
  LG='\033[0m\033[32m'
  LR='\033[0m\033[31m'
  SEEDER='\033[1m\033[32m'
  LEECHER='\033[1m\033[31m'

def helper():
  print(OutColors.DEFAULT+"\nSearch torrents from Kickass.to ;)")

def select_torrent():
  torrent=raw_input('>> ')
  return torrent

def download_torrent(url):
  fname=os.getcwd()+'/'+url.split('title=')[-1]+'.torrent'
  #http://stackoverflow.com/a/14114741/1302018
  try:
    r=requests.get(url)
    with open(fname,'wb') as f:
      for chunk in r.iter_content(chunk_size=1024):
        if chunk:
          f.write(chunk)
          f.flush()
  except requests.exceptions.RequestException as e:
    print('\n'+OutColors.LR+str(e))
    sys.exit(1)
 
  return fname

def aksearch():
  helper()
  tmp_url='http://kickass.to/usearch/'

  query=raw_input('Type query: ')
  url=tmp_url+query+'/'

  try:
    cont=requests.get(url)
  except requests.exceptions.RequestException as e:
    raise SystemExit('\\n'+OutColors.LR+str(e))

  if not re.findall(r'Download torrent file',str(cont.content)):
    print('Torrents found: 0')
    aksearch()
  else:
    soup=BeautifulSoup(cont.content)

    #to use by age, seeders, and leechers
    #sample:
    #700.46 MB
    #5
    #2 years
    #1852
    #130

    al=[s.get_text() for s in soup.find_all('td', {'class':'center'})]

    href=[a.get('href') for a in soup.find_all('a',{'title':'Download torrent file'})]
    size=[t.get_text() for t in soup.find_all('td',{'class':'nobr'})]
    title=[ti.get_text() for ti in soup.find_all('a',{'class':'cellMainLink'})]
    age=al[2::5]
    seeders=al[3::5]
    leechers=al[4::5]

    # for table printing
    table=[[OutColors.BW+str(i+1)+ OutColors.DEFAULT if (i+1)% 2==0 else i+1,
              OutColors.BW+title[i]+OutColors.DEFAULT if (i+1)% 2==0 else title[i],
              OutColors.BW+size[i]+OutColors.DEFAULT if (i+1)% 2==0 else size[i],
              OutColors.BW+age[i]+OutColors.DEFAULT if (i+1)% 2==0 else age[i],
              OutColors.SEEDER+seeders[i]+OutColors.DEFAULT if (i+1)% 2==0 else OutColors.LG+ seeders[i]+OutColors.DEFAULT,
              OutColors.LEECHER+leechers[i]+OutColors.DEFAULT if(i+1)% 2==0 else OutColors.LR+leechers[i]+OutColors.DEFAULT] for i in range(len(href))]

    print()
    print(tabulate.tabulate(table, headers=['No','Title','Size','Age','Seeders','Leechers']))

    #torrent selection
    if len(href)==1:
      torrent=1
    else:
      print('\nSelect torrent: [ 1 - '+str(len(href))+' ] or [ M ] to go back to main menu or [ Q ] to quit')
      torrent=select_torrent()
      if torrent=='Q' or torrent=='q':
        sys.exit(0)
      elif torrent=='M' or torrent=='m':
        aksearch()
      else:
        if int(torrent) <= 0 or int(torrent)>len(href):
          print('Use eyelasses...')
        else:
          print('Download >> '+href[int(torrent)-1].split('title=')[-1]+'.torrent')
          fname=download_torrent(href[int(torrent)-1])
          subprocess.Popen(['xdg-open',fname], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
          aksearch()

if __name__=='__main__':
  try:
    aksearch()
  except KeyboardInterrupt:
    print('\nHuha!')