Saving thumbnails to disk, need to change the render code

Author: kracekumar

hgtv/__init__.py

@@ -34,8 +34,10 @@
 
 
 def init_for(env):
+    from hgtv.upload import configure as upload_configure
     coaster.app.init_app(app, env)
     hgtv.models.commentease.init_app(app)
     lastuser.init_app(app)
     lastuser.init_usermanager(UserManager(hgtv.models.db, hgtv.models.User))
     app.config['tz'] = timezone(app.config['TIMEZONE'])
+    upload_configure()

hgtv/views/video.py

@@ -4,6 +4,8 @@
 from urlparse import urlparse, parse_qs
 from socket import gaierror
 import requests
+from werkzeug import secure_filename
+
 from flask import render_template, flash, abort, redirect, Markup, request, escape, jsonify, g
 from coaster.views import load_models
 from baseframe.forms import render_form, render_redirect, render_delete_sqla, render_message
@@ -13,6 +15,7 @@
 from hgtv.models import db, Channel, Video, Playlist, PlaylistVideo, CHANNEL_TYPE
 from hgtv.models.channel import PLAYLIST_AUTO_TYPE
 from hgtv.views.login import lastuser
+from hgtv.upload import uploaded_thumbnails, return_werkzeug_filestorage
 
 
 class DataProcessingError(Exception):
@@ -40,7 +43,9 @@ def process_video(video, new=False):
                         video.description = escape(r.json['entry']['media$group']['media$description']['$t'])
                 for item in r.json['entry']['media$group']['media$thumbnail']:
                     if item['yt$name'] == 'mqdefault':
-                        video.thumbnail_url = item['url']  # .replace('hqdefault', 'mqdefault')
+                        thumbnail_url_request = requests.get(item['url'])
+                        filestorage = return_werkzeug_filestorage(thumbnail_url_request, filename=secure_filename(r.json['entry']['title']['$t']))
+                        video.thumbnail_url = uploaded_thumbnails.save(filestorage)
                 video.video_sourceid = video_id
                 video.video_source = u"youtube"
             except requests.ConnectionError:

instance/settings-sample.py

@@ -7,6 +7,10 @@
 SQLALCHEMY_DATABASE_URI = 'sqlite:///test.db'
 #: Secret key
 SECRET_KEY = 'make this something random'
+#: Directory to upload the video thumbnail
+UPLOAD_DIRECTORY = 'uploads/'
+#: Allowed Extensions for thumbnails
+ALLOWED_EXTENSIONS = ['jpg', 'jpeg', 'png', 'gif']
 #: Timezone
 TIMEZONE = 'Asia/Calcutta'
 #: LastUser server

requirements.txt

@@ -1,4 +1,5 @@
 Flask
+Flask-Uploads
 https://github.com/hasgeek/coaster/zipball/master
 https://github.com/hasgeek/baseframe/zipball/master
 Flask-SQLAlchemy