pbp_com/apps/directory/models.py

from datetime import *
from dateutil import relativedelta

from django.db import models
from django.db.models import permalink
from django.utils.translation import ugettext_lazy as _
from django.core.urlresolvers import reverse
from django.contrib.auth.models import User
from django.contrib.localflavor.us.models import PhoneNumberField, USStateField
from fields import ZipCodeField
from django_extensions.db.models import TimeStampedModel
import os
import csv
import urllib

#import pbp_com.settings as settings
from django.conf import settings

def get_lat_long(location):
    key = settings.GOOGLE_API_KEY
    output = "csv"
    location = urllib.quote_plus(location)
    request = "http://maps.google.com/maps/geo?q=%s&output=%s&key=%s" % (location, output, key)
    data = urllib.urlopen(request).read()
    dlist = data.split(',')
    if dlist[0] == '200':
        return "%s, %s" % (dlist[2], dlist[3])
    else:
        return ''

class Person(TimeStampedModel):
    """Person model."""
    GENDER_CHOICES=(
        (1, 'Male'),
        (2, 'Female'),
    )
    title=models.CharField(_('title'), blank=True, null=True, max_length=20)
    first_name=models.CharField(_('first name'), blank=True, max_length=100)
    middle_name=models.CharField(_('middle name'), blank=True, null=True, max_length=100)
    last_name=models.CharField(_('last name'), blank=True, max_length=100)
    nickname=models.CharField(_('nickname'), blank=True, null=True, max_length=100)
    suffix=models.CharField(_('suffix'), blank=True, null=True, max_length=20)
    slug=models.SlugField(_('slug'), unique=True)
    user=models.ForeignKey(User, blank=True, null=True, help_text='If the person is an existing user of your site.')
    gender=models.PositiveSmallIntegerField(_('gender'), choices=GENDER_CHOICES, blank=True, null=True)
    birth_date=models.DateField(_('birth date'), blank=True, null=True)
    death_date=models.DateField(_('death date'), blank=True, null=True)
    phone=PhoneNumberField(blank=True)
    email=models.EmailField(blank=True)
    town=models.ForeignKey('Town', blank=True, null=True)
    
    class Meta:
        verbose_name=_('person')
        verbose_name_plural=_('people')
        ordering=('last_name', 'first_name')
        
    def __unicode__(self):
        return u'%s' % self.full_name
        
    @property
    def full_name(self):
        return u'%s %s' % (self.first_name, self.last_name)
        
    @property
    def age(self):
        if self.death_date:
            DELTA=self.death_date
        else:
            DELTA=datetime.today()
        return u'%s' % relativedelta.relativedelta(DELTA, self.birth_date).years

    @permalink    
    def get_absolute_url(self):
        return ('di-person-detail', None, {'slug': self.slug})

class Town(TimeStampedModel):
    '''Town model.'''
    name=models.CharField(_('name'), max_length=100)
    state=USStateField(_('state'))
    slug=models.SlugField(_('slug'), unique=True)
    in_coverage=models.BooleanField(_('in coverage'), default=False)
    zipcode=ZipCodeField(_('Zipcode'), blank=True, null=True, help_text='Use for a primary zipcode when town has multiple.', max_length=5)
    
    class Meta:
        unique_together=(('name','state',),)
        ordering=('state', 'name',)
        
    def __unicode__(self):
        return u'%s, %s' % (self.name, self.state)
        
    @permalink
    def get_absolute_url(self):
        return ('di-town-detail', None, {'slug': self.slug})

class PopulationYear(TimeStampedModel):
    ''' Population yaer model.

    Keeps track of a population count for a town in a given (usually census) year.
    '''
    town=models.ForeignKey(Town)
    year=models.IntegerField(_('year'), max_length=4)
    population=models.IntegerField(_('population'), max_length=7)
    
    def __unicode__(self):
        return u'%s %s population' % (self.town, self.year)
        
class TownInfo(TimeStampedModel):
    ''' Town info model.

    Keeps track of various pieces of town info for our register.
    
    We create a separate model so that we can use a really simple
    setup for our legacy website and just rewrite this whole class
    when we're ready to implement some higher-level town info stuff.

    This way we do not have to delete town ids and mess up all related_name
    applications.
    '''

    town=models.ForeignKey(Town)
    office=models.ForeignKey("Place")
    po_box=models.IntegerField(_('PO Box'))
    map=models.FileField(_('Town map'), blank=True, null=True, upload_to='directory/maps/')
    time_info=models.TextField(_('Contact info'))
    description=models.TextField(_('Description'))
    officials=models.TextField(_('Officials'))
    tax_info=models.TextField(_('Tax info'))
    permits=models.TextField(_('Permits'))
    town_ordinances=models.TextField(_('Town ordinances'))
    post_offices=models.TextField(_('Post offices'))
    total_area=models.IntegerField(_('Total area'), help_text='In acres')
    land_area=models.IntegerField(_('Land'), help_text='In acres')
    bog_area=models.IntegerField(_('Bog'), help_text='In acres', default=0)
    flood_plain_area=models.IntegerField(_('Flood plain'), help_text='In acres', default=0)
    inland_water_area=models.IntegerField(_('Inland water'), help_text='In acres', default=0)
    shoreline=models.IntegerField(_('Shoreline'), help_text='In miles', default=0)
    populations=models.ManyToManyField(PopulationYear, blank=True, null=True)
    edu_contact=models.TextField(_('Education contact'))
    enrollment=models.TextField(_('Enrollment numbers'))
    fire_dept_info=models.TextField(_('Fire department'))
    cemetery_info=models.TextField(_('Cemeteries'), blank=True, null=True)
    trash_recycle_info=models.TextField(_('Trash & recycling'), blank=True, null=True)
    public_restrooms=models.TextField(_('Public restrooms'), blank=True, null=True)
    walking_trails=models.TextField(_('Walking trails'), blank=True, null=True)
    public_landings=models.TextField(_('Public landings'), blank=True, null=True)

    def __unicode__(self):
        return u'%s information' % (self.town)
        
    @permalink
    def get_absolute_url(self):
        return ('di-town-info', None, {'slug': self.town.slug})


class Committee(TimeStampedModel):
    '''Committee model.'''
    title=models.CharField(_('title'), max_length=100)
    slug=models.SlugField(_('slug'), unique=True)
    town=models.ForeignKey(Town, related_name="town_committee")
    members=models.ManyToManyField(Person, blank=True, null=True)
    
    def __unicode__(self):
        return u'%s %s' % (self.town, self.title)
        
    def get_absolute_url(self):
        args = self.slug
        return reverse('di-committee-detail', args=args)


class Point(TimeStampedModel):
    lat_lon=models.CharField(_('coordinates'), blank=True, null=True, max_length=255)
    address=models.CharField(_('address'), max_length=255)
    town=models.ForeignKey(Town)
    zipcode=models.CharField(_('zip'), max_length=10, blank=True)
    
    def __unicode__(self):
        return u'%s, %s' % (self.address, self.town)
        
    class Meta:
        verbose_name=(_('point'))
        verbose_name_plural=(_('points'))
    

    def save(self, *args, **kwargs):
        if not self.lat_lon:
            location = "%s +%s +%s +%s" % (self.address, self.town, self.town.state, self.zipcode)
            self.lat_lon = get_lat_long(location)
            if not self.lat_lon:
                location = "%s +%s +%s" % (self.town, self.town.state, self.zipcode)
                self.lat_lon = get_lat_long(location)
        super(Point, self).save(*args, **kwargs)

class Place(TimeStampedModel):
    '''Place model for Directory application.'''
    STATUS_CHOICES = (
        (0, 'Inactive'),
        (1, 'Active'),
    )
    point=models.ForeignKey(Point, blank=True, null=True)
    title=models.CharField(_('Title'), max_length=255)
    slug=models.SlugField(_('Slug'))
    description=models.TextField(_('Description'), blank=True, null=True)
    status=models.IntegerField(_('Status'), choices=STATUS_CHOICES, default=1)
    po_box=models.IntegerField(_('P.O. Box'), blank=True, null=True, max_length=5)
    po_box_town=models.ForeignKey(Town, blank=True, null=True)
    phone=PhoneNumberField(_('Phone'), blank=True, null=True)
    fax=PhoneNumberField(_('Fax'), blank=True, null=True)
    email=models.EmailField(_('Email'), blank=True, null=True)
    website=models.URLField(_('Website'), blank=True, null=True, verify_exists=False)
    
    def __unicode__(self):
        return self.title
        
    @permalink
    def get_absolute_url(self):
        return ('di-place-detail', None, {'slug': self.slug})

class SchoolAdministrativeUnit(TimeStampedModel):
    '''
    An SAU, as defined by the state of Maine.
    '''
    
    name=models.CharField(_('name'), max_length=100)
    slug=models.SlugField(_('slug'), unique=True)
    union=models.IntegerField(_('union'), max_length=2, blank=True, null=True)
    #    governor=models.ForeignKey(Board, blank=True, null=True)
    
    def __unicode__(self):
        return self.name
    
    @permalink
    def get_absolute_url(self):
        return ('di-sau-detail', None, {'slug': self.slug})
    

class School(Place):
    '''
    School model.
    A school derives from the Place model, adding a few school-specific fields
    '''
    
    TYPE_CHOICES = (
        ('E', 'Elementary School'),
        ('H', 'High School'),
        ('C', 'Combined School'),
        ('U', 'College/University'),
        ('O', 'Other'),
    )

    CLASS_CHOICES = (
        ('N', 'None'),
        ('A', 'A'),
        ('B', 'B'),
        ('C', 'C'),
        ('D', 'D'),
    )

    REGION_CHOICES = (
	('N', 'None'),
	('W',  'West'),
	('E',  'East'),
    )
    
    short_title=models.CharField(_('short title'), max_length=30)
    type=models.CharField(_('type'), choices=TYPE_CHOICES, default="O", max_length=1)
    #enrollment=models.ManyToMany(Enrollment?)
        
    administration=models.ForeignKey(SchoolAdministrativeUnit)
    title_one_eligible=models.BooleanField(_('title one eligible'), default=False)
    title_one_recipient=models.BooleanField(_('title one recipient'), default=False)

    school_class=models.CharField(_('class'), max_length=1, choices=CLASS_CHOICES, default='N')
    region=models.CharField(_('region'), max_length=2, choices=REGION_CHOICES, default='N')
    mascot=models.CharField(_('mascot'), max_length=50, blank=True, null=True)
    athletic_director=models.ForeignKey(Person, related_name='athletic_director', blank=True, null=True)
    principal=models.ForeignKey(Person, related_name='principal', blank=True, null=True)
    
    @permalink
    def get_absolute_url(self):
        return ('di-school-detail', None, {'slug': self.slug})

class PhoneBookEntry(TimeStampedModel):
    entry=models.CharField(_('entry'), max_length=100)
    phone=PhoneNumberField()
    town=models.ForeignKey(Town)
    owner=models.ForeignKey(Person, blank=True, null=True)
    related_place=models.ForeignKey(Place, blank=True, null=True)
    
    def __unicode__(self):
        return u'%s: %s' % (self.entry, self.phone)
        
    class Meta:
        verbose_name=(_('phone book entry'))
        verbose_name_plural=(_('phone book entries'))
	
class PhoneBookEntryUpload(models.Model):
    data_file = models.FileField(_('data file (.csv)'), upload_to="temp",
                                help_text=_('Select a .csv file of phone book data to upload.'))
    town = models.ForeignKey(Town)                       
    
    class Meta:
        verbose_name = _('phone book entry upload')
        verbose_name_plural = _('phone book entry uploads')
    
    def save(self):
        super(PhoneBookEntryUpload, self).save()
        self.process_datafile()
        super(PhoneBookEntryUpload, self).delete()
    
    def process_datafile(self):
        if os.path.isfile(self.data_file.path):
            entry_data = csv.reader(open(self.data_file.path), delimiter='\t')
            for row in entry_data:
                args={}
                args['town'] = self.town
                args['entry'] = row[0]
                if len(row[1]) > 8 or len(row[1]) < 4:
                    args['phone'] = row[1]
                else:
                    args['phone'] = '207-'+row[1]
                entry = PhoneBookEntry.objects.create(**args)
                try:
                    entry.save()
                except:
                    pass