summaryrefslogtreecommitdiff
path: root/patchwork/fields.py
blob: 6eca1a2f150d73e7f3e0cb4ffd66b36f13c82118 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# Patchwork - automated patch tracking system
# Copyright (C) 2008 Jeremy Kerr <jk@ozlabs.org>
# Copyright (C) 2015 Intel Corporation
#
# SPDX-License-Identifier: GPL-2.0-or-later

import hashlib

from django.db import models


class HashField(models.CharField):

    def __init__(self, *args, **kwargs):
        self.n_bytes = len(hashlib.sha1().hexdigest())
        kwargs['max_length'] = self.n_bytes

        super(HashField, self).__init__(*args, **kwargs)

    def construct(self, value):
        # TODO: should this be unconditional?
        if isinstance(value, str):
            value = value.encode('utf-8')
        return hashlib.sha1(value)

    def from_db_value(self, value, *args, **kwargs):
        return self.to_python(value)

    def db_type(self, connection=None):
        return 'char(%d)' % self.n_bytes