2009-05-13 05:35:59 +01:00
#.-*- coding: utf-8 -*-
2009-05-13 05:37:08 +01:00
import troggle . settings as settings
import troggle . expo . models as models
2009-05-13 05:35:59 +01:00
import csv
import re
import datetime
import os
# Julian: the below code was causing errors and it seems like a duplication of the above. Hope I haven't broken anything by commenting it. -Aaron
#
# if name in expoers2008:
# print "2008:", name
# expomissing.discard(name) # I got an error which I think was caused by this -- python complained that a set changed size during iteration.
# yo = models.Expedition.objects.filter(year = "2008")[0]
# pyo = models.PersonExpedition(person = pObject, expedition = yo, is_guest=is_guest)
# pyo.save()
def parseMugShotAndBlurb ( firstname , lastname , person , header , pObject ) :
#create mugshot Photo instance
mugShotPath = os . path . join ( settings . EXPOWEB , " folk " , person [ header [ " Mugshot " ] ] )
if mugShotPath [ - 3 : ] == ' jpg ' : #if person just has an image, add it
mugShotObj = models . Photo (
caption = " Mugshot for " + firstname + " " + lastname ,
is_mugshot = True ,
file = mugShotPath ,
)
mugShotObj . save ( )
mugShotObj . contains_person . add ( pObject )
mugShotObj . save ( )
elif mugShotPath [ - 3 : ] == ' htm ' : #if person has an html page, find the image(s) and add it. Also, add the text from the html page to the "blurb" field in his model instance.
personPageOld = open ( mugShotPath , ' r ' ) . read ( )
pObject . blurb = re . search ( ' <body>.*<hr ' , personPageOld , re . DOTALL ) . group ( ) #this needs to be refined, take care of the HTML and make sure it doesn't match beyond the blurb
for photoFilename in re . findall ( ' i/.*?jpg ' , personPageOld , re . DOTALL ) :
mugShotPath = settings . EXPOWEB + " folk/ " + photoFilename
mugShotObj = models . Photo (
caption = " Mugshot for " + firstname + " " + lastname ,
is_mugshot = True ,
file = mugShotPath ,
)
mugShotObj . save ( )
mugShotObj . contains_person . add ( pObject )
mugShotObj . save ( )
pObject . save ( )
def LoadPersonsExpos ( ) :
persontab = open ( os . path . join ( settings . EXPOWEB , " noinfo " , " folk.csv " ) )
personreader = csv . reader ( persontab )
headers = personreader . next ( )
header = dict ( zip ( headers , range ( len ( headers ) ) ) )
models . Expedition . objects . all ( ) . delete ( )
years = headers [ 5 : ]
years . append ( " 2008 " )
for year in years :
y = models . Expedition ( year = year , name = " CUCC expo %s " % year )
y . save ( )
print " lll " , years
models . Person . objects . all ( ) . delete ( )
models . PersonExpedition . objects . all ( ) . delete ( )
expoers2008 = """ Edvin Deadman,Kathryn Hopkins,Djuke Veldhuis,Becka Lawson,Julian Todd,Natalie Uomini,Aaron Curtis,Tony Rooke,Ollie Stevens,Frank Tully,Martin Jahnke,Mark Shinwell,Jess Stirrups,Nial Peters,Serena Povia,Olly Madge,Steve Jones,Pete Harley,Eeva Makiranta,Keith Curtis """ . split ( " , " )
expomissing = set ( expoers2008 )
for person in personreader :
name = person [ header [ " Name " ] ]
name = re . sub ( " <.*?> " , " " , name )
mname = re . match ( " ( \ w+)(?: \ s((?:van |ten )? \ w+))?(?: \ s \ (([^)]*) \ ))? " , name )
if mname . group ( 3 ) :
nickname = mname . group ( 3 )
else :
nickname = " "
firstname , lastname = mname . group ( 1 ) , mname . group ( 2 ) or " "
print firstname , lastname , " NNN " , nickname
#assert lastname == person[header[""]], person
2009-05-13 05:38:18 +01:00
href = firstname . lower ( )
if lastname :
href + = " _ " + lastname . lower ( )
2009-05-13 05:35:59 +01:00
pObject = models . Person ( first_name = firstname ,
2009-05-13 05:38:18 +01:00
last_name = lastname , href = href ,
2009-05-13 05:35:59 +01:00
is_vfho = person [ header [ " VfHO member " ] ] ,
)
is_guest = person [ header [ " Guest " ] ] == " 1 " # this is really a per-expo catagory; not a permanent state
pObject . save ( )
2009-05-13 05:44:35 +01:00
#parseMugShotAndBlurb(firstname, lastname, person, header, pObject)
2009-05-13 05:35:59 +01:00
for year , attended in zip ( headers , person ) [ 5 : ] :
yo = models . Expedition . objects . filter ( year = year ) [ 0 ]
if attended == " 1 " or attended == " -1 " :
pyo = models . PersonExpedition ( person = pObject , expedition = yo , nickname = nickname , is_guest = is_guest )
pyo . save ( )
# error
elif ( firstname , lastname ) == ( " Mike " , " Richardson " ) and year == " 2001 " :
print " Mike Richardson(2001) error "
pyo = models . PersonExpedition ( person = pObject , expedition = yo , nickname = nickname , is_guest = is_guest )
pyo . save ( )
# this fills in those people for whom 2008 was their first expo
for name in expomissing :
firstname , lastname = name . split ( )
is_guest = name in [ " Eeva Makiranta " , " Keith Curtis " ]
print " 2008: " , name
2009-05-13 05:44:35 +01:00
persons = list ( models . Person . objects . filter ( first_name = firstname , last_name = lastname ) )
if not persons :
pObject = models . Person ( first_name = firstname ,
last_name = lastname ,
is_vfho = False ,
mug_shot = " " )
pObject . href = firstname . lower ( )
if lastname :
pObject . href + = " _ " + lastname . lower ( )
pObject . save ( )
else :
pObject = persons [ 0 ]
2009-05-13 05:35:59 +01:00
yo = models . Expedition . objects . filter ( year = " 2008 " ) [ 0 ]
pyo = models . PersonExpedition ( person = pObject , expedition = yo , nickname = " " , is_guest = is_guest )
pyo . save ( )
2009-05-13 05:39:52 +01:00
# expedition name lookup cached for speed (it's a very big list)
Gpersonexpeditionnamelookup = { }
def GetPersonExpeditionNameLookup ( expedition ) :
global Gpersonexpeditionnamelookup
res = Gpersonexpeditionnamelookup . get ( expedition . name )
if res :
return res
res = { }
duplicates = set ( )
personexpeditions = models . PersonExpedition . objects . filter ( expedition = expedition )
for personexpedition in personexpeditions :
possnames = [ ]
f = personexpedition . person . first_name . lower ( )
l = personexpedition . person . last_name . lower ( )
if l :
possnames . append ( f + " " + l )
possnames . append ( f + " " + l [ 0 ] )
possnames . append ( f + l [ 0 ] )
possnames . append ( f [ 0 ] + " " + l )
possnames . append ( f )
if personexpedition . nickname :
possnames . append ( personexpedition . nickname . lower ( ) )
for possname in possnames :
if possname in res :
duplicates . add ( possname )
else :
res [ possname ] = personexpedition
for possname in duplicates :
del res [ possname ]
Gpersonexpeditionnamelookup [ expedition . name ] = res
return res
