Download function added. DOWNLOAD_DIR existence is not controlled

Author: mgdelacroix

mt_scrapper.py

@@ -1,8 +1,23 @@
 from flask import Flask, request, render_template
 import string
 import requests
+import urllib2
 from bs4 import BeautifulSoup
 
+
+DOWNLOAD_DIR = 'Downloads'
+
+def downloadFile(url, directory):
+    file = urllib2.urlopen(url)
+
+    fileName = url[url.rfind('/')+1:len(url)]
+    fileName = directory + '/' + fileName
+
+    localFile = open(fileName, 'w')
+    localFile.write(file.read())
+    localFile.close()
+
+
 app = Flask(__name__)
 
 @app.route("/")
@@ -44,7 +59,6 @@ def show_chapters():
             episodios[input.get('name')] = input.get('value')
 
     # Generamos la peticion POST y parseamos los torrents
-
     payload = {'checkall': 'on', 'total_capis': len(episodios), 'tabla': 'series'}
     payload = dict(payload.items() + episodios.items())
     post = requests.post('http://mejortorrent.com/secciones.php?sec=descargas&ap=contar_varios', payload)
@@ -56,6 +70,9 @@ def show_chapters():
         if str(link.get('href'))[0:52] == 'http://www.mejortorrent.com/uploads/torrents/series/':
             torrents.append(link.get('href'))
 
+    for torrent in torrents:
+        downloadFile(torrent, DOWNLOAD_DIR)
+
     return render_template('serie.html', section=request.args.get('name', ''),
                            torrents=torrents,
                            cover=cover)

templates/serie.html

@@ -14,4 +14,6 @@ <h1>{{ section }}</h1>
   {% endfor %}
 </ul>
 
+<h2>Se han descargado los torrents</h2>
+
 {% endblock %}