rewrite file/instance backend

This commit is contained in:
j 2013-02-08 17:01:26 +00:00
parent 04a391a442
commit 7951c42609
3 changed files with 72 additions and 75 deletions

View file

@ -18,18 +18,31 @@ import ox
import chardet
from item import utils
import item.models
from person.models import get_name_sort
import extract
class File(models.Model):
AV_INFO = (
'duration', 'video', 'audio', 'oshash', 'size',
)
PATH_INFO = (
'episodes', 'extension', 'language', 'part', 'partTitle', 'version'
)
ITEM_INFO = (
'title', 'director', 'year',
'season', 'episode', 'episodeTitle',
'seriesTitle', 'seriesYear'
)
created = models.DateTimeField(auto_now_add=True)
modified = models.DateTimeField(auto_now=True)
oshash = models.CharField(max_length=16, unique=True)
item = models.ForeignKey("item.Item", related_name='files')
item = models.ForeignKey("item.Item", related_name='files', null=True)
path = models.CharField(max_length=2048, default="") # canoncial path/file
sort_path = models.CharField(max_length=2048, default="") # sort name
@ -127,26 +140,13 @@ class File(models.Model):
if self.framerate:
self.pixels = int(self.width * self.height * float(utils.parse_decimal(self.framerate)) * self.duration)
def parse_instance_path(self):
if self.instances.count():
path = self.instances.all()[0].path
data = ox.movie.parse_path(path)
for key in (
'normalizedPath', 'isEpisode',
'title', 'director', 'directorSort', 'year',
'season', 'episode', 'episodeTitle',
'seriesTitle', 'seriesYear'
):
del data[key]
self.path_info = data
def get_path_info(self):
data = self.path_info.copy()
for key in (
'title', 'director', 'year',
'season', 'episode', 'episodeTitle',
'seriesTitle', 'seriesYear'
):
data = {}
for key in self.PATH_INFO:
data[key] = self.info.get(key, None)
if self.item:
for key in self.ITEM_INFO:
data[key] = self.item.get(key)
data['directorSort'] = [get_name_sort(n) for n in self.item.get('director', [])]
data['isEpisode'] = data.get('season') != None \
@ -160,12 +160,10 @@ class File(models.Model):
for type in ox.movie.EXTENSIONS:
if data['extension'] in ox.movie.EXTENSIONS[type]:
data['type'] = type
if 'part' in data and not data['part']:
del data['part']
return data
def normalize_path(self):
#FIXME: always use path_info
#FIXME: always use format_path
if settings.USE_IMDB:
return ox.movie.format_path(self.get_path_info())
else:
@ -174,24 +172,31 @@ class File(models.Model):
path = self.instances.all()[0].path
return path
def save(self, *args, **kwargs):
if self.id and not self.path_info and self.instances.count():
def update_info(self, info, user):
#populate name sort with director if unknown
if info.get('director') and info.get('directorSort'):
for name, sortname in zip(info['director'], info['directorSort']):
get_name_sort(name, sortname)
self.item = item.models.get_item(info, user)
for key in self.AV_INFO + self.PATH_INFO:
if key in info:
self.info[key] = info[key]
self.parse_info()
self.parse_instance_path()
self.path = self.normalize_path()
def save(self, *args, **kwargs):
if self.id and self.info:
self.path = self.normalize_path()
if self.item:
data = self.get_path_info()
self.extension = data.get('extension')
self.language = data.get('language')
self.part = ox.sort_string(unicode(data.get('part', '')))
self.part = ox.sort_string(unicode(data.get('part') or ''))
self.part_title = ox.sort_string(unicode(data.get('partTitle')) or '')
self.type = data.get('type') or 'unknown'
self.version = data.get('version')
if self.path:
self.path = self.normalize_path()
self.sort_path = utils.sort_string(self.path)
self.is_audio = self.type == 'audio'
self.is_video = self.type == 'video'
self.is_subtitle = self.path.endswith('.srt')
@ -238,7 +243,7 @@ class File(models.Model):
p = user.get_profile()
return p.get_level() in ('admin', 'staff') or \
self.instances.filter(volume__user=user).count() > 0 or \
self.item.user == user
(not self.item or self.item.user == user)
def save_chunk(self, chunk, chunk_id=-1, done=False):
if not self.available:
@ -286,8 +291,8 @@ class File(models.Model):
'videoCodec': self.video_codec,
'wanted': self.wanted,
}
for key in ('part', 'partTitle', 'version', 'language', 'extension'):
data[key] = self.path_info.get(key)
for key in self.PATH_INFO:
data[key] = self.info.get(key)
data['users'] = list(set([i['user'] for i in data['instances']]))
if keys:
for k in data.keys():

View file

@ -7,20 +7,13 @@ import ox
from django.conf import settings
from item.models import get_item, Item
from item.models import Item
import item.tasks
from person.models import get_name_sort
import models
import extract
_INSTANCE_KEYS = ('mtime', 'path')
def get_or_create_item(volume, info, user):
item_info = ox.parse_movie_path(info['path'])
if item_info.get('director') and item_info.get('directorSort'):
for name, sortname in zip(item_info['director'], item_info['directorSort']):
get_name_sort(name, sortname)
return get_item(item_info, user)
def get_or_create_file(volume, f, user, item=None):
try:
@ -29,10 +22,11 @@ def get_or_create_file(volume, f, user, item=None):
file = models.File()
file.oshash = f['oshash']
file.path = f['path']
file.path = f['path']
if item:
file.item = item
else:
file.item = get_or_create_item(volume, f, user)
file.item = None #gets pupulated later via update_info
file.save()
return file
@ -66,6 +60,7 @@ def update_or_create_instance(volume, f):
setattr(instance, key, f[key])
instance.save()
instance.file.save()
if instance.file.item:
instance.file.item.update_wanted()
return instance

View file

@ -7,7 +7,7 @@ from datetime import datetime
from django import forms
from django.shortcuts import get_object_or_404, redirect
from django.conf import settings
from django.db.models import Count
from django.db.models import Count, Q
import ox
from ox.utils import json
@ -43,9 +43,12 @@ actions.register(removeVolume, cache=False)
@login_required_json
def update(request):
'''
2 calls possible:
volume/files
info
2 steps:
send files
{volume: 'Videos', files: [{oshash:, path:, mtime:, ,,}]}
send info about changed/new files
{volume: 'Videos', info: {oshash: {...}]}
call volume/files first and fill in requested info after that
param data {
@ -56,12 +59,13 @@ def update(request):
],
info: {oshash: object}
}
return {
status: {'code': int, 'text': string},
data: {
info: list,
data: list,
file: list
info: list, // list of files that need info
data: list, // list of flies that should be encoded to highest profile and uploaded
file: list // list of files that should be uploaded as is
}
}
'''
@ -82,23 +86,16 @@ def update(request):
user_profile.save()
if 'info' in data:
for oshash in data['info']:
info = data['info'][oshash]
f = models.File.objects.filter(oshash=oshash)
if f.count()>0:
f = f[0]
for f in models.File.objects.filter(oshash__in=data['info'].keys()):
if not f.info:
for key in ('atime', 'mtime', 'ctime'):
if key in info:
del info[key]
f.info = info
f.parse_info()
f.update_info(data['info'][f.oshash], user)
f.save()
if not upload_only:
all_files = models.Instance.objects.filter(volume__user=user)
files = all_files.filter(file__available=False)
if volume:
files = files.filter(volume=volume)
#fixme: might be better to check for file__path_info
response['data']['info'] = [f.file.oshash for f in all_files.filter(file__info='{}')]
response['data']['data'] = [f.file.oshash for f in files.filter(file__is_video=True,
file__available=False,
@ -211,11 +208,10 @@ def addFile(request):
extension = extension[-1]
else:
extension = 'webm'
f.path_info = {
'extension': extension
}
f.selected = True
f.info = data['info']
f.info['extension'] = extension
self.parse_info()
f.save()
response['data']['item'] = i.itemId
response['data']['itemUrl'] = request.build_absolute_uri('/%s' % i.itemId)
@ -408,6 +404,7 @@ def editFile(request):
f.instances.update(ignore=data['ignore'])
f.save()
#FIXME: is this to slow to run sync?
if f.item:
f.item.update_selected()
f.item.update_wanted()
for key in ('episodes', 'extension', 'language', 'part', 'partTitle', 'version'):