josch/yolanda
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

upload returns immediately now, encodes in background

Browse source 
git-svn-id: http://yolanda.mister-muffin.de/svn@395 7eef14d0-6ed0-489d-bf55-20463b2d70db
This commit is contained in:
erlehmann 2008-12-01 23:27:35 +00:00
parent 86f3f3df53
commit a29670d81f
1 changed files with 58 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

60
trunk/yolanda/controllers/upload.py
View file

@ -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__)
@ -13,36 +30,55 @@ class UploadController(BaseController):
return render('/xhtml/upload.mako')
def upload(self):
upload = request.params['file']
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"
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"
# 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)))
videoencode.run()
os.unlink(os.path.join(config['cache.dir'], 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))
return 'Successfully uploaded: %s'%video.query.all()
# 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(source)