cablegates/pandora/archive/views.py

318 lines
11 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4
from __future__ import division
import os.path
import re
from datetime import datetime
from urllib2 import unquote
import mimetypes
from django import forms
from django.core.paginator import Paginator
from django.contrib.auth.decorators import login_required
from django.contrib.auth.models import User
from django.db.models import Q, Avg, Count, Sum
from django.http import HttpResponse, Http404
2010-09-14 14:10:37 +00:00
from django.shortcuts import render_to_response, get_object_or_404, get_list_or_404, redirect
from django.template import RequestContext
from django.conf import settings
try:
import simplejson as json
except ImportError:
from django.utils import simplejson as json
from ox.django.decorators import login_required_json
from ox.django.shortcuts import render_to_json_response, get_object_or_404_json, json_response
from ox.django.http import HttpFileResponse
import ox
import models
2010-11-08 16:34:25 +00:00
from item.utils import oxid, parse_path
from item.models import getItem
2010-11-08 16:34:25 +00:00
import item.tasks
2010-08-10 22:01:41 +00:00
@login_required_json
def api_removeVolume(request):
data = json.loads(request.POST['data'])
user = request.user
try:
volume = models.Volume.objects.get(user=user, name=data['volume'])
volume.files.delete()
volume.delete()
response = json_response(status=200, text='ok')
except models.Volume.DoesNotExist:
response = json_response(status=404, text='volume not found')
return render_to_json_response(response)
@login_required_json
def api_update(request):
'''
2010-08-10 22:01:41 +00:00
2 calls possible:
volume/files
info
call volume/files first and fill in requested info after that
param data
2010-08-10 22:01:41 +00:00
volume: '',
2010-08-07 14:31:20 +00:00
files: [
2010-09-23 16:01:48 +00:00
{oshash:, path:, mtime:, }
2010-08-07 14:31:20 +00:00
]
info: {oshash: object}
return {'status': {'code': int, 'text': string},
2010-08-07 14:31:20 +00:00
'data': {info: list, data: list, file: list}}
'''
data = json.loads(request.POST['data'])
2010-08-07 14:31:20 +00:00
user = request.user
2010-08-10 22:01:41 +00:00
response = json_response({'info': [], 'data': [], 'file': []})
2010-09-17 21:06:01 +00:00
volume = None
2010-08-07 14:31:20 +00:00
if 'files' in data:
2010-08-10 22:01:41 +00:00
volume, created = models.Volume.objects.get_or_create(user=user, name=data['volume'])
2010-08-07 14:31:20 +00:00
all_files = []
for f in data['files']:
2010-08-10 22:01:41 +00:00
path = f['path']
folder = path.split('/')
name = folder.pop()
if folder and folder[-1] in ('Extras', 'Versions', 'DVDs'):
name = '/'.join([folder.pop(), name])
folder = '/'.join(folder)
2010-09-18 00:01:34 +00:00
#print folder
#print name
2010-08-10 22:01:41 +00:00
f['folder'] = folder
f['name'] = name
2010-08-07 14:31:20 +00:00
oshash = f['oshash']
all_files.append(oshash)
2010-11-08 17:43:59 +00:00
same_folder = models.Instance.objects.filter(folder=folder, volume=volume)
2010-08-07 14:31:20 +00:00
if same_folder.count() > 0:
2010-09-23 16:01:48 +00:00
item = same_folder[0].file.item
2010-08-07 14:31:20 +00:00
else:
2010-09-23 16:01:48 +00:00
item = None
2010-08-07 14:31:20 +00:00
path = os.path.join(folder, name)
2010-11-08 17:43:59 +00:00
instance = models.Instance.objects.filter(file__oshash=oshash, volume=volume)
if instance.count()>0:
instance = instance[0]
2010-08-07 14:31:20 +00:00
updated = False
2010-09-23 16:01:48 +00:00
for key in ('mtime', 'name', 'folder'):
2010-08-07 14:31:20 +00:00
if f[key] != getattr(instance, key):
setattr(instance, key, f[key])
updated=True
if updated:
2010-08-10 22:01:41 +00:00
instance.save()
else:
#look if oshash is known
2010-08-10 22:01:41 +00:00
file_objects = models.File.objects.filter(oshash=oshash)
if file_objects.count() > 0:
file_object = file_objects[0]
#new oshash, add to database
else:
2010-09-23 16:01:48 +00:00
if not item:
item_info = parse_path(folder)
item = getItem(item_info)
2010-08-10 22:01:41 +00:00
file_object = models.File()
file_object.oshash = oshash
file_object.name = name
2010-09-23 16:01:48 +00:00
file_object.item = item
2010-08-10 22:01:41 +00:00
file_object.save()
response['data']['info'].append(oshash)
2010-11-08 17:43:59 +00:00
instance = models.Instance()
2010-08-10 22:01:41 +00:00
instance.volume = volume
instance.file = file_object
2010-09-23 16:01:48 +00:00
for key in ('mtime', 'name', 'folder'):
2010-08-10 22:01:41 +00:00
setattr(instance, key, f[key])
instance.save()
2010-08-07 14:31:20 +00:00
#remove deleted files
2010-09-23 16:01:48 +00:00
#FIXME: can this have any bad consequences? i.e. on the selction of used item files.
2010-11-08 17:43:59 +00:00
models.Instance.objects.filter(volume=volume).exclude(file__oshash__in=all_files).delete()
2010-08-07 14:31:20 +00:00
user_profile = user.get_profile()
user_profile.files_updated = datetime.now()
user_profile.save()
if 'info' in data:
for oshash in data['info']:
info = data['info'][oshash]
2010-11-08 17:43:59 +00:00
instance = models.Instance.objects.filter(file__oshash=oshash, volume__user=user)
2010-08-07 14:31:20 +00:00
if instance.count()>0:
instance = instance[0]
if not instance.file.info:
2010-08-10 22:01:41 +00:00
for key in ('atime', 'mtime', 'ctime'):
if key in info:
del info[key]
2010-08-07 14:31:20 +00:00
instance.file.info = info
instance.file.save()
2010-11-08 17:43:59 +00:00
files = models.Instance.objects.filter(volume__user=user, file__available=False)
2010-09-17 21:06:01 +00:00
if volume:
files = files.filter(volume=volume)
2010-08-10 22:01:41 +00:00
response['data']['info'] = [f.file.oshash for f in files.filter(file__info='{}')]
#needs some flag to find those that are actually used main is to generic
response['data']['data'] = [f.file.oshash for f in files.filter(file__is_video=True, file__is_main=True)]
response['data']['file'] = [f.file.oshash for f in files.filter(file__is_subtitle=True)]
2010-08-07 14:31:20 +00:00
return render_to_json_response(response)
2010-08-07 14:31:20 +00:00
@login_required_json
def api_encodingProfile(request):
response = json_response({'profile': settings.VIDEO_PROFILE})
return render_to_json_response(response)
2010-08-10 22:01:41 +00:00
@login_required_json
2010-08-07 14:31:20 +00:00
def api_upload(request):
'''
2010-08-24 17:16:33 +00:00
oshash: string
frame: [] //multipart frames
file: [] //multipart file
return {'status': {'code': int, 'text': string},
2010-08-07 14:31:20 +00:00
'data': {info: object, rename: object}}
'''
2010-08-07 14:31:20 +00:00
user = request.user
f = get_object_or_404(models.File, oshash=request.POST['oshash'])
2010-08-24 17:16:33 +00:00
if 'frame' in request.FILES:
if f.frames.count() == 0:
for frame in request.FILES.getlist('frame'):
name = frame.name
#float required?
position = float(os.path.splitext(name)[0])
fr = models.Frame(file=f, position=position)
fr.save()
fr.frame.save(name, frame)
response = json_response({})
else:
response = json_response(status=403, text='permissino denied')
if 'file' in request.FILES:
2010-09-17 21:06:01 +00:00
if not f.available:
2010-09-03 13:28:44 +00:00
f.data.save('data.raw', request.FILES['file'])
2010-09-17 21:06:01 +00:00
f.available = True
f.save()
response = json_response(status=200, text='file saved')
2010-08-24 17:16:33 +00:00
else:
response = json_response(status=403, text='permissino denied')
2010-08-07 14:31:20 +00:00
return render_to_json_response(response)
class VideoChunkForm(forms.Form):
chunk = forms.FileField()
chunkId = forms.IntegerField(required=False)
done = forms.IntegerField(required=False)
@login_required_json
def firefogg_upload(request):
2010-08-13 17:17:23 +00:00
profile = request.GET['profile']
2010-09-05 21:08:05 +00:00
if profile.endswith('.webm'):
profile = os.path.splitext(profile)[0]
2010-08-13 17:17:23 +00:00
oshash = request.GET['oshash']
2010-08-07 14:31:20 +00:00
#handle video upload
if request.method == 'POST':
#post next chunk
2010-08-13 17:17:23 +00:00
if 'chunk' in request.FILES and oshash:
2010-09-03 13:28:44 +00:00
f = get_object_or_404(models.File, oshash=oshash)
2010-08-07 14:31:20 +00:00
form = VideoChunkForm(request.POST, request.FILES)
2010-09-03 13:28:44 +00:00
if form.is_valid() and profile == settings.VIDEO_PROFILE and f.editable(request.user):
2010-08-07 14:31:20 +00:00
c = form.cleaned_data['chunk']
chunk_id = form.cleaned_data['chunkId']
response = {
'result': 1,
'resultUrl': request.build_absolute_uri('/')
}
2010-09-03 13:28:44 +00:00
if not f.save_chunk(c, chunk_id):
2010-08-07 14:31:20 +00:00
response['result'] = -1
elif form.cleaned_data['done']:
2010-09-03 13:28:44 +00:00
f.available = True
f.save()
#FIXME: this fails badly if rabbitmq goes down
2010-11-08 16:34:25 +00:00
item.tasks.updateStreams.delay(f.item.itemId)
2010-08-07 14:31:20 +00:00
response['result'] = 1
response['done'] = 1
return render_to_json_response(response)
2010-09-03 13:28:44 +00:00
#init upload
elif oshash and profile == settings.VIDEO_PROFILE:
2010-08-13 17:17:23 +00:00
#404 if oshash is not know, files must be registered via update api first
f = get_object_or_404(models.File, oshash=oshash)
2010-09-03 13:28:44 +00:00
if f.editable(request.user):
if f.video:
f.video.delete()
2010-09-08 17:14:01 +00:00
f.available = False
2010-09-03 13:28:44 +00:00
f.save()
response = {
#is it possible to no hardcode url here?
'uploadUrl': request.build_absolute_uri('/api/upload/?oshash=%s&profile=%s' % (f.oshash, profile)),
'result': 1
}
return render_to_json_response(response)
2010-08-07 14:31:20 +00:00
response = json_response(status=400, text='this request requires POST')
return render_to_json_response(response)
@login_required_json
2010-09-13 13:46:49 +00:00
def api_editFile(request): #FIXME: should this be file.files. or part of update
'''
2010-09-13 13:46:49 +00:00
change file / imdb link
'''
2010-09-13 13:46:49 +00:00
response = json_response(status=501, text='not implemented')
2010-08-07 14:31:20 +00:00
return render_to_json_response(response)
2010-09-14 14:10:37 +00:00
def lookup_file(request, oshash):
f = get_object_or_404(models.File, oshash=oshash)
2010-09-23 16:01:48 +00:00
return redirect(f.item.get_absolute_url())
2010-09-14 14:10:37 +00:00
2010-09-13 13:46:49 +00:00
"""
2010-08-07 14:31:20 +00:00
def api_fileInfo(request):
'''
param data
2010-08-07 14:31:20 +00:00
oshash string
return {'status': {'code': int, 'text': string},
'data': {imdbId:string }}
'''
2010-08-07 14:31:20 +00:00
if 'data' in request.POST:
oshash = json.loads(request.POST['data'])
elif 'oshash' in request.GET:
oshash = request.GET['oshash']
2010-09-23 16:01:48 +00:00
f = models.ItemFile.objects.get(oshash=oshash)
2010-08-07 14:31:20 +00:00
response = {'data': f.json()}
return render_to_json_response(response)
2010-08-07 14:31:20 +00:00
def api_subtitles(request):
'''
param data
oshash string
language string
subtitle string
return
if no language is provided:
{data: {languages: array}}
if language is set:
{data: {subtitle: string}}
if subtitle is set:
saves subtitle for given language
'''
if 'data' in request.POST:
data = json.loads(request.POST['data'])
oshash = data['oshash']
language = data.get('language', None)
srt = data.get('subtitle', None)
if srt:
user = request.user
sub = models.Subtitles.objects.get_or_create(user, oshash, language)
sub.srt = srt
sub.save()
else:
response = json_response({})
if language:
2010-09-23 16:01:48 +00:00
q = models.Subtitles.objects.filter(item_file__oshash=oshash, language=language)
2010-08-07 14:31:20 +00:00
if q.count() > 0:
response['data']['subtitle'] = q[0].srt
return render_to_json_response(response)
2010-09-23 16:01:48 +00:00
l = models.Subtitles.objects.filter(item_file__oshash=oshash).values('language')
2010-08-07 14:31:20 +00:00
response['data']['languages'] = [f['language'] for f in l]
return render_to_json_response(response)
"""