from django.core.urlresolvers import reverse
from django.contrib.sites.models import Site
from django.conf import settings
+from patchwork.parser import hash_patch
import django.oldforms as oldforms
import re
person.save()
else:
for person in people:
- person.user = self.user
+ person.link_to_user(self.user)
person.save()
def __str__(self):
class Meta:
ordering = ['ordering']
-class HashField(models.Field):
+class HashField(models.CharField):
__metaclass__ = models.SubfieldBase
def __init__(self, algorithm = 'sha1', *args, **kwargs):
self.algorithm = algorithm
try:
import hashlib
- self.hashlib = True
+ def _construct(string = ''):
+ return hashlib.new(self.algorithm, string)
+ self.construct = _construct
+ self.n_bytes = len(hashlib.new(self.algorithm).hexdigest())
except ImportError:
- self.hashlib = False
- if algorithm == 'sha1':
- import sha
- self.hash_constructor = sha.new
- elif algorithm == 'md5':
- import md5
- self.hash_constructor = md5.new
- else:
+ modules = { 'sha1': 'sha', 'md5': 'md5'}
+
+ if algorithm not in modules.keys():
raise NameError("Unknown algorithm '%s'" % algorithm)
-
- super(HashField, self).__init__(*args, **kwargs)
- def db_type(self):
- if self.hashlib:
- n_bytes = len(hashlib.new(self.algorithm).digest())
- else:
- n_bytes = len(self.hash_constructor().digest())
- if settings.DATABASE_ENGINE.startswith('postgresql'):
- return 'bytea'
- elif settings.DATABASE_ENGINE == 'mysql':
- return 'binary(%d)' % n_bytes
- else:
- raise Exception("Unknown database engine '%s'" % \
- settings.DATABASE_ENGINE)
+ self.construct = __import__(modules[algorithm]).new
- def to_python(self, value):
- return value
+ self.n_bytes = len(self.construct().hexdigest())
- def get_db_prep_save(self, value):
- return ''.join(map(lambda x: '\\%03o' % ord(x), value))
+ kwargs['max_length'] = self.n_bytes
+ super(HashField, self).__init__(*args, **kwargs)
- def get_manipulator_field_objs(self):
- return [oldforms.TextField]
+ def db_type(self):
+ return 'char(%d)' % self.n_bytes
class Patch(models.Model):
project = models.ForeignKey(Project)
headers = models.TextField(blank = True)
content = models.TextField()
commit_ref = models.CharField(max_length=255, null = True, blank = True)
- hash = HashField()
+ hash = HashField(null = True, db_index = True)
def __str__(self):
return self.name
s = self.state
except:
self.state = State.objects.get(ordering = 0)
- if hash is None:
- print "no hash"
+
+ if self.hash is None:
+ self.hash = hash_patch(self.content).hexdigest()
+
super(Patch, self).save()
def is_editable(self, user):