320 lines
12 KiB
Python
320 lines
12 KiB
Python
# -*- 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
|
|
from django.shortcuts import render_to_response, get_object_or_404, get_list_or_404
|
|
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 oxdjango.decorators import login_required_json
|
|
from oxdjango.shortcuts import render_to_json_response, get_object_or_404_json, json_response
|
|
from oxdjango.http import HttpFileResponse
|
|
import ox
|
|
|
|
import models
|
|
|
|
from backend.utils import oxid, parse_path
|
|
import backend.models
|
|
|
|
@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):
|
|
'''
|
|
2 calls possible:
|
|
volume/files
|
|
info
|
|
call volume/files first and fill in requested info after that
|
|
|
|
param data
|
|
volume: '',
|
|
files: [
|
|
{oshash:, path:, ctime:, atime:, mtime:, }
|
|
]
|
|
info: {oshash: object}
|
|
|
|
return {'status': {'code': int, 'text': string},
|
|
'data': {info: list, data: list, file: list}}
|
|
'''
|
|
print "here we go fucker", request.user
|
|
data = json.loads(request.POST['data'])
|
|
user = request.user
|
|
|
|
response = json_response({'info': [], 'data': [], 'file': []})
|
|
|
|
if 'files' in data:
|
|
volume, created = models.Volume.objects.get_or_create(user=user, name=data['volume'])
|
|
all_files = []
|
|
for f in data['files']:
|
|
print f
|
|
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)
|
|
print folder
|
|
print name
|
|
f['folder'] = folder
|
|
f['name'] = name
|
|
oshash = f['oshash']
|
|
all_files.append(oshash)
|
|
|
|
same_folder = models.FileInstance.objects.filter(folder=folder, volume=volume)
|
|
if same_folder.count() > 0:
|
|
movie = same_folder[0].file.movie
|
|
else:
|
|
movie = None
|
|
|
|
path = os.path.join(folder, name)
|
|
|
|
instance = models.FileInstance.objects.filter(file__oshash=oshash, volume=volume)
|
|
if instance.count()>0:
|
|
instance = instance[0]
|
|
updated = False
|
|
for key in ('atime', 'mtime', 'ctime', 'name', 'folder'):
|
|
if f[key] != getattr(instance, key):
|
|
setattr(instance, key, f[key])
|
|
updated=True
|
|
if updated:
|
|
instance.save()
|
|
else:
|
|
#look if oshash is known
|
|
file_objects = models.File.objects.filter(oshash=oshash)
|
|
if file_objects.count() > 0:
|
|
file_object = file_objects[0]
|
|
#new oshash, add to database
|
|
else:
|
|
if not movie:
|
|
movie_info = parse_path(folder)
|
|
movie = backend.models.getMovie(movie_info)
|
|
file_object = models.File()
|
|
file_object.oshash = oshash
|
|
file_object.name = name
|
|
file_object.movie = movie
|
|
file_object.save()
|
|
response['data']['info'].append(oshash)
|
|
instance = models.FileInstance()
|
|
instance.volume = volume
|
|
instance.file = file_object
|
|
for key in ('atime', 'mtime', 'ctime', 'name', 'folder'):
|
|
setattr(instance, key, f[key])
|
|
instance.save()
|
|
|
|
#remove deleted files
|
|
#FIXME: can this have any bad consequences? i.e. on the selction of used movie files.
|
|
models.FileInstance.objects.filter(volume=volume).exclude(file__oshash__in=all_files).delete()
|
|
|
|
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]
|
|
instance = models.FileInstance.objects.filter(file__oshash=oshash, volume__user=user)
|
|
if instance.count()>0:
|
|
instance = instance[0]
|
|
if not instance.file.info:
|
|
for key in ('atime', 'mtime', 'ctime'):
|
|
if key in info:
|
|
del info[key]
|
|
instance.file.info = info
|
|
instance.file.save()
|
|
|
|
files = models.FileInstance.objects.filter(volume__user=user, file__available=False)
|
|
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)]
|
|
|
|
return render_to_json_response(response)
|
|
|
|
|
|
@login_required_json
|
|
#FIXME: is this part of the api or does it have to be outside due to multipart?
|
|
def api_upload(request):
|
|
'''
|
|
multipart
|
|
param data
|
|
oshash: string
|
|
frame: [] //multipart frames
|
|
return {'status': {'code': int, 'text': string},
|
|
'data': {info: object, rename: object}}
|
|
'''
|
|
user = request.user
|
|
f = get_object_or_404(models.File, oshash=request.POST['oshash'])
|
|
if f.frames.count() == 0 and 'frame' in request.FILES:
|
|
for frame in request.FILES['frame']:
|
|
name = frame.name
|
|
position = float(os.path.splitext(name)[0])
|
|
fr = models.Frame(file=f, position=position)
|
|
fr.save()
|
|
fr.frame.save(frame, name)
|
|
response = json_response({})
|
|
else:
|
|
response = json_response(status=403, text='permissino denied')
|
|
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):
|
|
#handle video upload
|
|
if request.method == 'POST':
|
|
#init upload
|
|
profile = request.POST.get('profile', request.GET['profile'])
|
|
#FIXME: check for valid profile
|
|
if 'oshash' in request.POST:
|
|
#404 if oshash is not know, files must be registered via update api first
|
|
f = get_object_or_404(models.File, oshash=request.POST['oshash'])
|
|
stream, created = models.Stream.objects.get_or_create(file=file, profile=profile)
|
|
if stream.video: #FIXME: check permission here instead of just starting over
|
|
stream.video.delete()
|
|
stream.available = False
|
|
stream.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)
|
|
#post next chunk
|
|
if 'chunk' in request.FILES and 'oshash' in request.GET:
|
|
print "all chunk now"
|
|
stream = get_object_or_404(models.Stream, oshash=request.GET['oshash'], profile=profile)
|
|
|
|
form = VideoChunkForm(request.POST, request.FILES)
|
|
if form.is_valid() and stream.editable(request.user):
|
|
c = form.cleaned_data['chunk']
|
|
chunk_id = form.cleaned_data['chunkId']
|
|
response = {
|
|
'result': 1,
|
|
'resultUrl': request.build_absolute_uri('/')
|
|
}
|
|
if not stream.save_chunk(c, chunk_id):
|
|
response['result'] = -1
|
|
elif form.cleaned_data['done']:
|
|
#FIXME: send message to encode deamon to create derivates instead
|
|
stream.available = True
|
|
stream.save()
|
|
response['result'] = 1
|
|
response['done'] = 1
|
|
return render_to_json_response(response)
|
|
print request.GET, request.POST
|
|
response = json_response(status=400, text='this request requires POST')
|
|
return render_to_json_response(response)
|
|
|
|
"""
|
|
@login_required_json
|
|
def list_files(request):
|
|
'''
|
|
GET list
|
|
> {
|
|
"files": {
|
|
"a41cde31c581e11d": {"path": "E/Example, The/An Example.avi", "size":1646274},
|
|
}
|
|
}
|
|
'''
|
|
response = {}
|
|
response['files'] = {}
|
|
qs = models.UserFile.filter(user=request.user)
|
|
p = Paginator(qs, 1000)
|
|
for i in p.page_range:
|
|
page = p.page(i)
|
|
for f in page.object_list:
|
|
response['files'][f.movie_file.oshash] = {'path': f.path, 'size': f.movie_file.size}
|
|
return render_to_json_response(response)
|
|
|
|
def find_files(request):
|
|
response = {}
|
|
query = _parse_query(request)
|
|
response['files'] = {}
|
|
qs = models.UserFile.filter(user=request.user).filter(movie_file__movie__id__in=query['q'])
|
|
p = Paginator(qs, 1000)
|
|
for i in p.page_range:
|
|
page = p.page(i)
|
|
for f in page.object_list:
|
|
response['files'][f.movie_file.oshash] = {'path': f.path, 'size': f.movie_file.size}
|
|
return render_to_json_response(response)
|
|
|
|
def api_fileInfo(request):
|
|
'''
|
|
param data
|
|
oshash string
|
|
return {'status': {'code': int, 'text': string},
|
|
'data': {imdbId:string }}
|
|
'''
|
|
if 'data' in request.POST:
|
|
oshash = json.loads(request.POST['data'])
|
|
elif 'oshash' in request.GET:
|
|
oshash = request.GET['oshash']
|
|
f = models.MovieFile.objects.get(oshash=oshash)
|
|
response = {'data': f.json()}
|
|
return render_to_json_response(response)
|
|
|
|
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:
|
|
q = models.Subtitles.objects.filter(movie_file__oshash=oshash, language=language)
|
|
if q.count() > 0:
|
|
response['data']['subtitle'] = q[0].srt
|
|
return render_to_json_response(response)
|
|
l = models.Subtitles.objects.filter(movie_file__oshash=oshash).values('language')
|
|
response['data']['languages'] = [f['language'] for f in l]
|
|
return render_to_json_response(response)
|
|
"""
|