#!/usr/bin/python
#------------------------------------------------------------------------------
#
#   BiBTeX -> Pajek converter
#
#   Vladimir Batagelj, April 2006
#------------------------------------------------------------------------------

from string import strip,split,replace,lower,index
from sys import argv,exit

def processRecord(bibRec,bibType):
   global  net, nam, vec, clu, numRec
   numRec += 1
   net.write('%s : %s > %s\n' % (numRec,bibType,bibRec))
   parts = bibRec.split('=')
   desKeys = ['head']; desVals = []
   for i in range(len(parts)):
      part = strip(parts[i]); j = part.rfind(',')
      if j < 0: j = len(part)-1
      desVal = strip(part[:j]); s = desVal[0]
      if (s == '"' or s == '{'): desVal = desVal[1:-1]
      desVals.append(desVal)
      desKeys.append(lower(strip(part[j+1:])))
   desVals[0] = desVals[0].split('{')[1]
   for i in range(len(desVals)):
      net.write('   %s = %s\n' % (desKeys[i],desVals[i]))
   net.write('  --- %s\n' % (bibType))
   try:
      i = desKeys.index('title'); title = desVals[i].replace('"',"''")
   except ValueError:
      title = 'UNKNOWN'
   nam.write('%s  "%s"\n' % (numRec,title))
   try:
      i = desKeys.index('author'); authors = desVals[i]
   except ValueError:
      authors = 'UNKNOWN'
   authors = authors.split(' and ')
   for author in authors: net.write('  --- %s\n' % (author))
   try:
      i = desKeys.index('year'); year = desVals[i]
   except ValueError:
      year = 'UNKNOWN'
   net.write('  --- %s\n' % (year))


def run(workdir,input):
   global net, nam, vec, clu, numRec
   pubs=['article','book','booklet','inbook','incollection',
         'inproceedings','manual','mastersthesis','misc','phdthesis',
         'proceedings','techreport','unpublished']
   line=' '
   try: # open input BiBTeX file
      bib = open(workdir+input,'r')
   except IOError , (errno, strerror):
      print "I/O error(%s): %s" % (errno, strerror)
      exit()
   print 'bibFile   = ', workdir+input
   net = open(workdir+'bib.net','w')
   print 'networkFile = ', workdir+'bib.net'
   nam = open(workdir+'bib.nam','w')
   print 'titleFile = ', workdir+'bib.nam'
   nam.write('*vertices\n')
   vec = open(workdir+'bib.vec','w')
   print 'yearFile = ', workdir+'bib.vec'
   clu = open(workdir+'bib.clu','w')
   print 'typeFile = ', workdir+'bib.clu'
   numRec = 0; bibRec = ''; bibType = -1
   numLine = 0; last = 0
   while not last:
      line = bib.readline()
      if not line:
         line = '@'; last = 1
      else:
         line = line.strip(); numLine += 1
      if line != '':
         if line[0] == '@':     # start of record
            if bibType >= 0:
               processRecord(bibRec,bibType)
               bibRec = ''
            head = line.split('{')
            head = replace(head[0],'@','')
            try:
               bibType = pubs.index(lower(head))
            except ValueError:
               bibType = -1
               print numLine, head
         if bibType >= 0: bibRec += line
   net.write('\n')
   net.write('*records = %s lines = %s\n' % (numRec,numLine))
   net.flush(); nam.flush(); vec.flush(); clu.flush()
   bib.close(); net.close(); nam.close(); vec.close(); clu.close()

#
# run bib2pajek
#
if __name__ == '__main__': # run it from command line
   if len(sys.argv) == 3:
      run(argv[1],argv[2])
   else:
      print "Module Bib2Pajek"
      print "Two arguments (WorkDir, BiBTeXfile) required to run !"
      exit()
   print
else: # it is imported
   print "Module Bib2Pajek imported."
   print "To run, type: bib2paj03.run('D:\\vlado\\BibTeX\\','lexicon.bib')"
   print "where 'lexicon.bib' is your input BiBTeX file"

#- End -------------------------------------------------------------------------------