from starlit.bib.bibtexdb import BibTexDB
from starlit.bib.adsdb import ADSBibDB
from starlit.bib.adscache import ADSCacheDB

from paperweight.document import FilesystemTexDocument

doc = FilesystemTexDocument("skysub.tex", recursive=False)

doc.bib_path

rich_refs = doc.rich_bib_keys
print len(rich_refs)

print(rich_refs.keys())

print(rich_refs['Taylor:2011'])

for cite_key, cites in rich_refs.iteritems():
    print cite_key, len(cites)

cachedb = ADSCacheDB(host='localhost', port=27017, ads_db=ADSBibDB())
adsdb = ADSBibDB(cache=cachedb)
bibdb = BibTexDB(doc.bib_path, ads_cache=cachedb)

pub = bibdb['Vaduvescu:2004']
print(pub.authors)
print(pub.title)
print(pub.bibcode)
print(pub.abstract)

secondary_ref_pubs = pub.references
print(len(secondary_ref_pubs))

for secondary_ref_pub in secondary_ref_pubs:
    print(secondary_ref_pub.bibcode)
    print(secondary_ref_pub.authors)
    print(secondary_ref_pub.title)
    print("")