upload returns immediately now, encodes in background

git-svn-id: http://yolanda.mister-muffin.de/svn@395 7eef14d0-6ed0-489d-bf55-20463b2d70db

trunk/yolanda/controllers/upload.py

@@ -1,9 +1,26 @@
+## <one line to give the program's name and a brief idea of what it does.>
+## Copyright (C) 2007, 2008 Nils Dagsson Moskopp, Johannes Schauer
+
+## This program is free software: you can redistribute it and/or modify
+## it under the terms of the GNU Affero General Public License as
+## published by the Free Software Foundation, either version 3 of the
+## License, or (at your option) any later version.
+
+## This program is distributed in the hope that it will be useful,
+## but WITHOUT ANY WARRANTY; without even the implied warranty of
+## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+## GNU Affero General Public License for more details.
+
+## You should have received a copy of the GNU Affero General Public License
+## along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
 import logging
 
 from yolanda.lib.base import *
 from yolanda.lib.gstreamer import info, snapshot, encode
 import os
 import hashlib
+import threading
 
 log = logging.getLogger(__name__)
 
@@ -11,38 +28,57 @@ class UploadController(BaseController):
 
     def index(self):
         return render('/xhtml/upload.mako')
-    
+
     def upload(self):
-        upload = request.params['file']
-        
-        #check if file is video
+        upload = request.params['file'] # request.param.get('file') is better maybe, doesn't raise a KeyError exception
+
+        # check if file is video
         videoinfo = info.Info(upload.file)
         if not videoinfo.get_info():
-            return "not a valid video"
-        
-        #check if file is duplicate
+            c.message = {
+                        'type': 'error',
+                        'text': 'Your file was not recognized as a video. Go figure.'
+                        }
+            return render('/xhtml/upload.mako')
+
+        # check if file is duplicate
         upload.file.seek(0)
         sha256 = hashlib.sha256(upload.file.read(1024*1024)).hexdigest()
-        
-        if model.Video.query.filter_by(sha256=sha256).count():
-            return "duplicate"
-        
-        #set up database entry
+
+#        if model.Video.query.filter_by(sha256=sha256).count():
+#            c.message = {
+#                        'type': 'error',
+#                        'text': 'Your file was already uploaded. Go away.'
+#                        }
+#            return render('/xhtml/upload.mako')
+
+        # set up database entry
         video = model.Video(title=request.params['title'],sha256=sha256)
         model.session.commit()
-        
-        #copy file to temp destination
+
+        # copy file to temporary destination
         temp_file = open(os.path.join(config['cache.dir'], str(video.id)), 'w')
         upload.file.seek(0)
         u.copyfileobj(upload.file, temp_file)
         upload.file.close()
         temp_file.close()
-        
-        #here begins the later daemon's work
-        videoencode = encode.Encode(
-            os.path.join(config['cache.dir'], str(video.id)),
-            os.path.join(config['pylons.paths']['static_files'], "videos", str(video.id)))
+
+        # define stuff
+        videosource=os.path.join(config['cache.dir'], str(video.id))
+        videodestination=os.path.join(config['pylons.paths']['static_files'], "videos", str(video.id))
+
+        # start encoding in background
+        threading.Thread(target=self.bgencode, args=(videosource, videodestination)).start()
+
+        # return 'Successfully uploaded: %s'%video.query.all()
+        c.message = {
+                    'type': 'information',
+                    'text': ''
+                    }
+        c.message['text']='Your file was successfully uploaded to "%s".'%videodestination
+        return render('/xhtml/index.mako')
+
+    def bgencode(self, source, destination):
+        videoencode = encode.Encode(source,destination)
         videoencode.run()
-        os.unlink(os.path.join(config['cache.dir'], str(video.id)))
-        
-        return 'Successfully uploaded: %s'%video.query.all()
+        os.unlink(source)