decentral1se/pandora
1
0
Fork 0
forked from 0x2620/pandora
Code Issues Pull requests Projects Releases Wiki Activity

matches

Browse source
This commit is contained in:
j 2013-03-24 12:53:32 +00:00
parent 2aabb6f69a
commit 7c5569b18a
1 changed files with 18 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
pandora/file/models.py
View file

@ -4,7 +4,7 @@ from __future__ import division, with_statement
import os import os
import re import re
import subprocess import subprocess
from urllib import quote from urllib import quote, unquote
from django.db import models from django.db import models
from django.contrib.auth.models import User from django.contrib.auth.models import User
@ -72,7 +72,7 @@ class File(models.Model):
return username, name, extension return username, name, extension
def get_absolute_url(self): def get_absolute_url(self):
return '/files/%s' % quote(self.get_id()) return ('/files/%s' % quote(self.get_id())).replace('%3A', ':')
def get_id(self): def get_id(self):
return u'%s:%s.%s' % (self.user.username, self.name, self.extension) return u'%s:%s.%s' % (self.user.username, self.name, self.extension)
@ -175,7 +175,22 @@ class File(models.Model):
size = [1,1] size = [1,1]
self.ratio = size[0] / size[1] self.ratio = size[0] / size[1]
def update_matches(self):
import annotation.models
import item.models
import text.models
urls = [self.get_absolute_url()]
url = unquote(urls[0])
if url != urls[0]:
urls.append(url)
matches = 0
for url in urls:
matches += annotation.models.Annotation.objects.filter(value__contains=url).count()
matches += item.models.Item.objects.filter(data__contains=url).count()
matches += text.models.Text.objects.filter(text__contains=url).count()
if matches != self.matches:
File.objects.filter(id=self.id).update(matches=matches)
self.matches = matches
def delete_file(sender, **kwargs): def delete_file(sender, **kwargs):
t = kwargs['instance'] t = kwargs['instance']