blechreiz-website/musicians/models.py

from django.db import models
from django.contrib.auth.models import User
from django.utils.translation import ugettext as _

import os

INSTRUMENTS = (
             ('TR',  _('Trumpet')   ),
             ('TRB', _('Trombone')  ),
             ('HRN',_('Horn') ),
             ('TUBA',_('Tuba')      )
)
             

def musicianPictureName( musician, originalName ):
    fileExtension = os.path.splitext(originalName)[1]
    return "user_images/" + musician.user.username + fileExtension

def musicianSmallPictureName( musician, originalName ):
    fileExtension = os.path.splitext(originalName)[1]
    return "user_images/" + musician.user.username + "_thumb" + fileExtension


class Musician( models.Model ):
    # Link to user object, contains first name and last name
    user       = models.OneToOneField( User, verbose_name=_("user") )
    
    image       = models.ImageField( upload_to = musicianPictureName,      verbose_name=_("image")                )
    small_image = models.ImageField( upload_to = musicianSmallPictureName, verbose_name = _("circular thumbnail") ) 
    
    
    # Properties
    instrument = models.CharField( max_length=4, choices=INSTRUMENTS, blank=True, verbose_name=_("instrument") )
    
    birthday   = models.DateField( null=True, verbose_name=_("birthday") )
    
    street     = models.CharField( max_length=80, blank=True, verbose_name=_("street") )
    city       = models.CharField( max_length=40, blank=True, verbose_name=_("city") )
    zip_code   = models.IntegerField( null=True, verbose_name=_("zip_code") )
    
    phone_home   = models.CharField( max_length=18, blank=True, verbose_name=_("phone_home") )
    phone_mobile = models.CharField( max_length=18, blank=True, verbose_name=_("phone_mobile") )
    phone_work   = models.CharField( max_length=18, blank=True, verbose_name=_("phone_work") )
    
    @property
    def isDeepBrass(self):
        return self.instrument == 'TRB' or self.instrument == "EUPH" or self.instrument == "TUBA" or self.instrument == "HRN"
    
    @property
    def isHighBrass(self):
        return self.instrument == 'TR'
    
    public_description = models.TextField( blank=True, verbose_name=_("public_description") )
    
    def __unicode__( self ):
        return self.user.username