Ticket #18501: bug18501.shar

# This is a shell archive.  Save it in a file, remove anything before
# this line, and then unpack it by entering "sh file".  Note, it may
# create directories; files and directories will be owned by you and
# have default permissions.
#
# This archive contains:
#
#       devices
#       devices/__init__.py
#       devices/admin.py
#       devices/models.py
#       devices/tests.py
#       devices/fixtures
#       devices/fixtures/msk.json
#
echo c - devices
mkdir -p devices > /dev/null 2>&1
echo x - devices/__init__.py
sed 's/^X//' >devices/__init__.py << 'cb149f71f097b0bc51fcd4315d659ec2'
cb149f71f097b0bc51fcd4315d659ec2
echo x - devices/admin.py
sed 's/^X//' >devices/admin.py << '5441acfd00fe2f753bc5553ac68101d7'
Xfrom devices import models
Xfrom django.contrib import admin
X
Xadmin.site.register(models.DevicePciId)
5441acfd00fe2f753bc5553ac68101d7
echo x - devices/models.py
sed 's/^X//' >devices/models.py << '93737618aac33064567404e01de70784'
X# vim: ts=4 sw=4 fdm=marker tw=158
Xfrom django.db import models
Xfrom django import forms
Xfrom django.utils.translation import ugettext_lazy as _
X
Xclass HexField(models.Field):
X    description = 'Integers provided and presented as hexadecimal strings'
X    
X    __metaclass__ = models.SubfieldBase
X    def __init__(self, *args, **kwargs):
X        if( kwargs['max_length'] % 2 != 0 ) :
X            kwargs['max_length'] += 1
X        if( kwargs['max_length'] > 16 ) :
X            kwargs['max_length'] = 16
X        kwargs['max_length'] /= 2
X        super(HexField, self).__init__(*args, **kwargs)
X        
X    def db_type(self, connection) :
X        if self.max_length > 3 :
X            return 'BIGINT'
X        elif self.max_length > 1 :
X            return 'INTEGER'
X        else :
X            return 'SMALLINT'
X    
X    def to_python(self, value) :
X        if isinstance(value, int) :
X            if value < 0 :
X                raise ValidationError(
X                    "Negative values are not supported ({})".format(value))
X            
X            if self.max_length > 4 :
X                fmt = '0x{:016x}'
X            elif self.max_length > 2 :
X                fmt = '0x{:08x}'
X            else :
X                fmt = '0x{:04x}'
X            valid = fmt.format(value)
X        
X        elif isinstance(value, (unicode, str)) :
X            if value == '' :
X                valid = '0x00'
X            else :
X                try :
X                    check = int(value, 16)
X                    valid = value
X                except ValueError:
X                    raise ValidationError(
X                        "Unable to convert >{}< to hexadecimal".format(value))
X        elif value is None :
X            if self.max_length > 4 :
X                valid = '0x0000000000000000'
X            elif self.max_length > 2 :
X                valid = '0x00000000'
X            else :
X                valid = '0x0000'
X        else :
X            raise ValidationError(
X                "Expected string or int and got {}".format(type(value)))
X        
X        return valid
X
X    def get_prep_value(self, value) :
X        if isinstance(value, unicode) or isinstance(value, str) :
X            val = int(value, 16)
X        elif isinstance(value, int) :
X            raise ValidationError(
X                'get_prep_value called with already prepped    value'
X            )
X        else :
X            raise ValidationError(
X                "Value {} of type {} not expected".format(
X                    value, type(value)
X                )
X            )
X        
X        return val
X
X    def get_prep_lookup(self, lookup_type, value) :
X        if lookup_type == 'exact' :
X            return self.get_prep_value(value)
X        elif lookup_type == 'in' :
X            return [self.get_prep_value(v) for v in value]
X        else :
X            raise TypeError('Unsupported lookup type')
X
X    def formfield(self, **kwargs) :
X        """OVerride this later with hex-aware widget"""
X        defaults = { 'form_class': forms.CharField }
X        defaults.update(kwargs)
X
X        return super(HexField, self).formfield(**defaults)
X
Xclass VendorHardwareIdentifier(models.Model) :
X    identifier = HexField(
X        max_length=4,
X        primary_key=True,
X        verbose_name=_('identifier')
X    )
X    name = models.CharField(
X        max_length=96,
X        default='Unknown ID',
X        db_index=True,
X        verbose_name=_('name')
X    )
X    
X    def __unicode__(self) :
X        return self.name
X    
X    class Meta:
X        abstract = True
X
Xclass HardwareDeviceIdentifier(models.Model) :
X    identifier = HexField(
X        max_length=4,
X        verbose_name=_('identifier')
X    )
X    name = models.CharField(
X        max_length=192,
X        verbose_name=_('device name')
X    )
X
X    def __unicode__(self) :
X        return self.name
X    
X    class Meta :
X        abstract = True
X
Xclass VendorPciId(VendorHardwareIdentifier) :
X    added = models.DateField(auto_now_add=True)
X
Xclass DevicePciId(HardwareDeviceIdentifier) :
X    vendor = models.ForeignKey(VendorPciId,
X        verbose_name=_('vendor'),
X    )
X    added = models.DateField(auto_now_add=True)
X
X    class Meta :
X        verbose_name = _('PCI device identifier')
X        verbose_name_plural = _('PCI device identifiers')
X        unique_together = (('vendor', 'identifier'))
X        ordering = [ 'name' ]
93737618aac33064567404e01de70784
echo x - devices/tests.py
sed 's/^X//' >devices/tests.py << '5c6e7aac64209866496e51a471254275'
Xfrom __future__ import absolute_import
X# vim: ts=4 sw=4 fdm=marker et tw=78 ai
Xfrom django.test import TestCase
Xfrom django.forms.models import modelform_factory
Xfrom .models import DevicePciId, VendorPciId
Xfrom datetime import datetime, tzinfo, timedelta
X
XPK_OUTPUT="""<select name="vendor" id="id_vendor">
X<option value="">---------</option>
X<option value="0x11ab" selected="selected">Marvell Technology Group
XLtd.</option>
X<option value="0xfeda">Broadcom Inc</option>
X</select>"""
X
Xclass UTC(tzinfo) :
X    def utcoffset(self, dt):
X        return timedelta(0)
X    def tzname(self, dt) :
X        return "UTC"
X    def dst(self, dt) :
X        return timedelta(0)
X
Xutc = UTC()
X
Xclass SimpleTest(TestCase):
X    def test_custom_field_as_pk(self) :
X        now = datetime.utcnow().replace(tzinfo=utc)
X        bc = VendorPciId.objects.create(identifier='0xfeda',
X                name='Broadcom Inc', added=now)
X        mtg = VendorPciId.objects.create(identifier='0x11ab',
X                name='Marvell Technology Group Ltd.', added=now)
X        msk = DevicePciId.objects.create(identifier='0x4363', vendor=mtg,
X                name='88E8055 PCI-E Gigabit Ethernet Controller', added=now)
X        form_class = modelform_factory(DevicePciId)
X        form = form_class(instance=msk)
X        output = unicode(form['vendor'])
X        #print output
X        self.assertHTMLEqual(output, PK_OUTPUT)
5c6e7aac64209866496e51a471254275
echo c - devices/fixtures
mkdir -p devices/fixtures > /dev/null 2>&1
echo x - devices/fixtures/msk.json
sed 's/^X//' >devices/fixtures/msk.json << '399e6afc29059243ac06a621d3380c06'
X[
X{
X  "pk": "0x11ab", 
X  "model": "devices.vendorpciid", 
X  "fields": {
X    "added": "2012-07-17", 
X    "name": "Marvell Technology Group Ltd."
X  }
X},
X{
X  "pk": "0xfeda", 
X  "model": "devices.vendorpciid", 
X  "fields": {
X    "added": "2012-07-17", 
X    "name": "Broadcom Inc"
X  }
X},
X{
X  "pk": 1, 
X  "model": "devices.devicepciid", 
X  "fields": {
X    "identifier": "0x4363", 
X    "vendor": 4523, 
X    "name": "88E8055 PCI-E Gigabit Ethernet Controller", 
X    "added": "2012-07-17"
X  }
X}
X]
399e6afc29059243ac06a621d3380c06
exit