smart_update_db.py

import re, os, sys, traceback, cPickle
os.environ['DJANGO_SETTINGS_MODULE'] = 'settings'

from gravebook.models import Article, Category, Other

from incunabula.models import Article as IArticle
from incunabula.models import MasterArticle  as IMasterArticle
from incunabula.models import Reference as IReference

from django.db import transaction

from django.utils.encoding import smart_str, smart_unicode

import settings
settings.DEBUG = False #important otherwise run out of memory on server

tpat    = re.compile('title="(.+?)"')
idpat   = re.compile('id="(.+?)"')
imgpat  = re.compile('image="(.+?)"')
bpat    = re.compile('birth="(.+?)"')
dpat    = re.compile('death="(.+?)"')
plpat   = re.compile('people_links="(.+?)"')
olpat   = re.compile('other_links="(.+?)"')
catpat  = re.compile('categories="(.+?)"')


LIMIT = -1
START = -1

def insert_xml(path):
    f = open(path)
    count = 0

    for line in f:
        count += 1
        if count < START:
            continue

        print count,
        
        if LIMIT > 0 and count == LIMIT:
            break

        try:
            title  = tpat.search(line).group(1)
            wid    = int(idpat.search(line).group(1))
            img    = imgpat.search(line).group(1)
            birth  = int(bpat.search(line).group(1))
            death  = int(dpat.search(line).group(1))
            #cats   = catpat.search(line).group(1).split('|')
            #others = olpat.search(line).group(1).split('|')
        
            art = Article(name=title, wid=wid, image=img, birth=birth, death=death)
            art.save()

            print title
        except:
            #raise
            continue
    f.close()

def add_cats(path):
    f = open(path)
    count = 0

    for line in f:
        count += 1
        if count < START:
            continue

        print count
        
        if LIMIT > 0 and count == LIMIT:
            break

        try:
            title  = tpat.search(line).group(1)
            cats   = catpat.search(line).group(1).split('|')
            
            art = Article.objects.get(name=title)
            for cat in cats:
                if cat == '':
                    continue
                try:
                    c = Category.objects.get(name=cat)
                except:
                    c = Category(name=cat)
                    c.save()

                art.categories.add(c)
            art.save()
        except:
            continue
    f.close()


def update_category_size():
    c = 0
    for cat in Category.objects.iterator():
        cat.size = cat.article_set.count()
        cat.save()

        print c
        c+=1

def build_people_graph(path):
    f = open(path)
    count = 0

    for line in f:
        count += 1
        if count < START:
            continue

        print count
        if LIMIT > 0 and count == LIMIT:
            break

        #if count == 2:
        #   break

        try:
            title  = tpat.search(line).group(1)
            people = plpat.search(line).group(1)
        
            us = Article.objects.get(name=title)
            us.people.clear()
            us.peers.clear()

            for link in people.split('|'):
                if link == '' or link == title:
                    continue
                try:
                    them = Article.objects.get(name=link)
                except:
                    #raise
                    continue
                
                us.people.add(them)
                if (us.birth != -1 and us.death != -1 and us.death-them.birth > 15 
                    and them.death-us.birth > 15):
                    us.peers.add(them)

                us.save()
            us.save()
        except:
            #raise
            continue
    f.close()

def load(path):
    f = open(path)
    revw = cPickle.load(f)
    f.close()
    return revw


def insert_incunabula_masters(revw):
    c = 0
    for k in revw.keys():
        ma = IMasterArticle(name=k)
        #print k
        ma.save()

        print c
        c+=1
        if LIMIT > 0 and c > LIMIT:
            break
        
def insert_incunabula_articles(revw):
    c = 0
    sizes = revw['sizes']
    for k in revw.keys():
        try:
            ma = IMasterArticle.objects.get(name=k)

            for ed in revw[k]['editions']:
                a = revw[k]['editions'][ed][0]

                ia = IArticle(name=a['name'], art_id=a['id'], art_ed=ed, text=a['txt'],
                              prank=0.0, volume_score= (0.0 + len(a['txt'])) / sizes[ed],
                              match_master=ma, 
                              match_score=-1.0)
                #print a['name']
                ia.save()
        except:
            continue
            
        print c
        c+=1
        if LIMIT > 0 and c > LIMIT:
            print 'wtf>'
            break
            
def process_split(gravebook=True):
    if not gravebook:
        for name in os.listdir('_data/split'):
            revw = load('_data/split/'+name)
            insert_incunabula_masters(revw)

    c = 0
    for name in os.listdir('_data/split'):
        revw = load('_data/split/'+name)

        if gravebook:
            gr_insert_incunabula_articles(revw)
        else:
            insert_incunabula_articles(revw)

        print c
        c+=1



def update_inc_volume_score():
    eds = {3 : [], 9 : [], 11 : [], 15 : []}
    for art in IArticle.objects.all():
        eds[art.art_ed].append(len(art.text))

    print 'prepped text dict'

    avg = dict()
    for k in eds.keys():
        if eds[k] == []:
            continue
        avg[k] = sum(eds[k]) / len(eds[k])

    print 'computed averages'

    sdev = {3 : 0, 9 : 0, 11 : 0, 15 : 0}
    for k in eds.keys():
        for v in eds[k]:
            sdev[k] += (avg[k] - v)**2

    print 'computed variance'
    
    for k in sdev:
        if len(eds[k]) > 0:
            sdev[k] = (sdev[k]/len(eds[k]))**0.5

    
    for art in IArticle.objects.all():
        art.volume_score = (len(art.text)-avg[art.art_ed]) / sdev[art.art_ed]
        art.save()


def extract_people_graph(out):
    f = open(out, 'w')
    c = 0
    for art in Article.objects.iterator():
        f.write(str(art.wid)+' ')

        for ed in art.people.iterator():
            f.write(str(ed.wid)+' ')
            
        f.write('\n')

        print c
        c+=1

    f.close()

def get_top_people(pranked, lim=200):
    f = open(pranked)
    arts = []
    for line in f:
        arts.append(line.strip().split(':'))
    f.close()
    
    arts.sort(key=lambda x: float(x[1]), reverse=True)
    
    for wid, prank in arts[:lim]:
        art = Article.objects.get(wid=wid)


def get_inc_matched_edition():
    for art in IArticle.objects.iterator():
        print str(art)+'#'+str(art.art_ed)+'#'+str(art.match_master)
        

def fill_gr_linked_by():
    c = 0
    for art in Article.objects.iterator():
        for art_to in art.people.iterator():
            art_to.linked_by.add(art)
        print c
        c+=1

def fill_gr_peers():
    c = 0
    OVERLAP = 15
    for art in Article.objects.iterator():
        art.peers.clear()
        for person in art.people.iterator():
            if person.birth == -1 or person.death == -1:
                continue
        
            if person.death-art.birth > OVERLAP and art.death-person.birth > OVERLAP:
                art.peers.add(person)
                art.save()
        
        print c, art.name
        c+=1

def update_gr_vscores(path):
    f = open(path)
    c = 0
    for line in f:
        name, score = line.strip().split('\t')
        try:
            art = Article.objects.get(name=name.strip(), art_ed=1000)
            art.vscore = float(score)
            art.save()
            
            print c, 'vscores'
            c+=1
        except Article.DoesNotExist:
            continue
        except:
            traceback.print_exc(file=sys.stdout)            
            pass

    f.close()

def add_inc_wiki_articles():
    for art in IArticle.objects.iterator():
        try:
            master = art.match_master
            wa = Article.objects.get(name=master.name.replace('_', ' '))
            if art.name == wa.name:
                continue

            try:
                bla = IArticle.objects.filter(name=wa.name+'waw', art_ed=1000)
                if bla.count() > 0:
                    continue
            except:
                pass
        
            a = IArticle(name=wa.name, art_id=wa.wid, art_ed=1000, text='', prank=0.0, 
                         volume_score=wa.vscore, match_master=master, match_score=1.0)
            a.save()
            print a.name, 'incwiki'
        except:
            pass

def parse_name(name):
    c = len(name)
    while c > 0:
        try:
            name = smart_unicode(name[:c])
            return name
        except:
            c-=1
            

def gr_insert_incunabula_articles(revw):
    #clean up first
    c = 0
    '''
    for a in Article.objects.filter(art_ed=15).iterator():
        a.delete()
        print c, 'del'
        c+=1
    '''


    sizes = revw['sizes']
    for k in revw.keys():
        name = ' '.join(k.split('_'))
        try:
            ma = Article.objects.get(name=name)

            for ed in revw[k]['editions']:
                #make matches unique
                if ma.article_set.filter(art_ed=ed).count() != 0:
                    continue
                
                a = revw[k]['editions'][ed][0]
                txt = a['txt']
                if ed == 15:
                    pass
                
                name = str(ed)+'_'+str(a['id'])+'_'+a['name']
                name = parse_name(name)
                if name == None:
                    continue

                ia = Article(name=name, wid=a['id'], art_ed=ed, text=txt,
                              vscore = 0.0,
                              match_master=ma)

                ia.save()
                print ia
                '''
                for cat in ma.categories.iterator():
                    ia.categories.add(cat)
                ia.save()
                '''

            ma.match_count = ma.article_set.count()
            ma.save()

        except Article.DoesNotExist:
            continue

        except:
            raise
            #print k,'not found'
            continue
            
    
def fix_gr_years():
    for art in Article.objects.iterator():
        for cat in art.categories.iterator():
            if 'births' in cat.name:
                yr = re.search('\d+', cat.name)
                if yr != None:
                    yr = yr.group()
                    art.birth = yr
                    art.save()

            if 'deaths' in cat.name:
                yr = re.search('\d+', cat.name)
                if yr != None:
                    yr = yr.group()
                    art.death = yr
                    art.save()

                    print art

def gr_update_inc_volume_score():
    eds = {3 : [], 9 : [], 11 : [], 15 : []}
    for art in Article.objects.filter(art_ed__in=[3,9,11,15]).iterator():
        eds[art.art_ed].append(len(art.text))

    print 'prepped text dict'

    avg = dict()
    for k in eds.keys():
        if eds[k] == []:
            continue
        avg[k] = sum(eds[k]) / len(eds[k])

    print 'computed averages'

    sdev = {3 : 0, 9 : 0, 11 : 0, 15 : 0}
    for k in eds.keys():
        for v in eds[k]:
            sdev[k] += (avg[k] - v)**2

    print 'computed variance'
    
    for k in sdev:
        if len(eds[k]) > 0:
            sdev[k] = (sdev[k]/len(eds[k]))**0.5

    
    for art in Article.objects.filter(art_ed__in=[3,9,11,15]).iterator():
        art.vscore = (len(art.text)-avg[art.art_ed]) / sdev[art.art_ed]
        art.save()

def gr_add_self_master():
    for art in Article.objects.filter(art_ed=1000).iterator():
        art.match_master = art
        art.save()

def fix_names():
    for art in Article.objects.iterator():
        print art.name
        art.name = smart_str(art.name)
        art.text = smart_str(art.text)
        art.save()

        '''
        for m in art.article_set.iterator():
            m.name = str(m.name.encode("utf-8"))
            m.save()
            str(m.name)
        '''

def gr_fix_all():
    print os.getpid(), 'fixing'
    c = 0
    #filter for missing self in match master
    for art in Article.objects.iterator():
        #fix wrong years
        for cat in art.categories.iterator():
            if 'births' in cat.name:
                yr = re.search('\d+', cat.name)
                if yr != None:
                    yr = yr.group()
                    art.birth = int(yr)
                    art.save()

            if 'deaths' in cat.name:
                yr = re.search('\d+', cat.name)
                if yr != None:
                    yr = int(yr.group())
                    art.death = yr
                    art.save()        

        #fix self reference
        art.match_master = art
        art.save()

        c+=1
        if c % 100 == 0:
            print c, os.getpid(), 'fix all'

def fix_count():
    c = 0
    for art in Article.objects.iterator():
        art.match_count = art.article_set.count()
        art.save()
        print c
        c+=1

def gr_insert_ed_15(split_dir):
    #for a in Article.objects.filter(art_ed=15).iterator():
    #    a.delete()

    w = open('15notmatched', 'w')
    MAXC = 5
    c = 0
    for pkl in os.listdir(split_dir):

        pkl = split_dir+pkl
        f = open(pkl)
        arts = cPickle.load(f)
        f.close()

        for a in arts:
            match = ' '. join(a['matched'][0].split('_'))
            ma = None
            try:
                ma = Article.objects.get(name=match, art_ed=1000)
            except:
                for candid in a['candids'][:MAXC]:
                    try:
                        match = ' '. join(candids.split('_'))
                        ma = Article.objects.get(name=match, art_ed=1000)
                    except:
                        continue

            if ma == None:
                w.write(a['name']+' not matched\n')
                continue

            ed = 15
            name=str(ed)+'_'+str(a['id'])+'_'+a['name']
            name = smart_unicode(name)
            ia = Article(name=name, wid=a['id'], art_ed=ed, text=a['txt'],
                         vscore = 0.0,
                         match_master=ma)
            ia.save()

            ma.match_count = ma.article_set.count()
            ma.save()

            c+=1
            print c, 'ed15'
    w.close()


def final_db_update():
    gr_insert_ed_15('_data/ed_15/')
    process_split(gravebook=True)
    gr_update_inc_volume_score()

if __name__ == '__main__':
    #final_db_update()
    #process_split(gravebook=True)
    #gr_update_inc_volume_score()

    #-revw = load('_data/sample_revw.pkl')
    #-gr_insert_incunabula_articles(revw)
    #gr_update_inc_volume_score()
    #fix_count()
    #748587

    #gr_fix_all()
    #fill_gr_peers()
    #update_gr_vscores('_data/wiki_vol_zscores')
    #fill_gr_linked_by()

    build_people_graph('_data/people_articles_filtered.txt')