"""Builder for CVs."""
from copy import deepcopy
from datetime import date
from regolith.builders.basebuilder import LatexBuilderBase
from regolith.fsclient import _id_key
from regolith.sorters import doc_date_key, ene_date_key, position_key
from regolith.stylers import month_fullnames, sentencecase
from regolith.tools import (
all_docs_from_collection,
awards_grants_honors,
dereference_institution,
filter_employment_for_advisees,
filter_grants,
filter_presentations,
filter_projects,
filter_publications,
fuzzy_retrieval,
make_bibtex_file,
merge_collections_superior,
remove_duplicate_docs,
)
[docs]
class CVBuilder(LatexBuilderBase):
"""Build CV from database entries"""
btype = "cv"
needed_colls = ["institutions", "people", "grants", "citations", "projects", "proposals", "presentations"]
[docs]
def construct_global_ctx(self):
"""Constructs the global context"""
super().construct_global_ctx()
gtx = self.gtx
rc = self.rc
gtx["people"] = sorted(
all_docs_from_collection(rc.client, "people"),
key=position_key,
reverse=True,
)
gtx["presentations"] = sorted(all_docs_from_collection(rc.client, "presentations"), key=_id_key)
gtx["institutions"] = sorted(all_docs_from_collection(rc.client, "institutions"), key=_id_key)
gtx["all_docs_from_collection"] = all_docs_from_collection
gtx["float"] = float
gtx["str"] = str
gtx["zip"] = zip
[docs]
def latex(self):
"""Render latex template"""
rc = self.rc
gtx = self.gtx
if rc.people:
people = [fuzzy_retrieval(gtx["people"], ["aka", "_id", "name"], rc.people[0])]
else:
people = gtx["people"]
for person in people:
# so we don't modify the dbs when de-referencing
names = frozenset(person.get("aka", []) + [person["name"]] + [person["_id"]])
begin_period = date(1650, 1, 1)
pubs = filter_publications(
all_docs_from_collection(rc.client, "citations"),
names,
reverse=True,
)
bibfile = make_bibtex_file(pubs, pid=person["_id"], person_dir=self.bldir)
emps = person.get("employment", [])
emps = [em for em in emps if not em.get("not_in_cv", False)]
for e in emps:
e["position"] = e.get("position_full", e.get("position").title())
emps.sort(key=ene_date_key, reverse=True)
edu = person.get("education", [])
edu.sort(key=ene_date_key, reverse=True)
teach = person.get("teaching", [])
for t in teach:
t["position"] = t.get("position").title()
projs = filter_projects(all_docs_from_collection(rc.client, "projects"), names)
for proj in projs:
for member in proj.get("team", []):
member.get("role", "").replace("pi", "PI")
member["role"] = member.get("role", "").title()
just_grants = list(all_docs_from_collection(rc.client, "grants"))
just_proposals = list(all_docs_from_collection(rc.client, "proposals"))
grants = merge_collections_superior(just_proposals, just_grants, "proposal_id")
presentations = filter_presentations(
self.gtx["people"],
self.gtx["presentations"],
self.gtx["institutions"],
person.get("_id"),
statuses=["accepted"],
)
for grant in grants:
for member in grant.get("team"):
dereference_institution(member, self.gtx["institutions"])
pi_grants, pi_amount, _ = filter_grants(grants, names, pi=True)
coi_grants, coi_amount, coi_sub_amount = filter_grants(grants, names, pi=False)
for grant in coi_grants:
format_pi = grant["me"].get("position", "").replace("copi", "Co-PI")
format_role = format_pi.replace("pi", "PI")
grant["me"]["position"] = format_role
aghs = awards_grants_honors(person, "honors")
service = awards_grants_honors(person, "service", funding=False)
# TODO: pull this out so we can use it everywhere
for ee in [emps, edu]:
for e in ee:
dereference_institution(e, self.gtx["institutions"])
undergrads = filter_employment_for_advisees(
self.gtx["people"], begin_period, "undergrad", person["_id"]
)
for undergrad in undergrads:
undergrad["role"] = undergrad["role"].title()
masters = filter_employment_for_advisees(self.gtx["people"], begin_period, "ms", person["_id"])
for master in masters:
master["role"] = master["role"].title()
currents = filter_employment_for_advisees(self.gtx["people"], begin_period, "phd", person["_id"])
graduateds = filter_employment_for_advisees(self.gtx["people"], begin_period, "phd", person["_id"])
postdocs = filter_employment_for_advisees(self.gtx["people"], begin_period, "postdoc", person["_id"])
postdocs = remove_duplicate_docs(postdocs, "name")
visitors = filter_employment_for_advisees(
self.gtx["people"], begin_period, "visitor-unsupported", person["_id"]
)
visitors = remove_duplicate_docs(visitors, "name")
for visitor in visitors:
visitor["role"] = visitor["role"].title()
iter = deepcopy(graduateds)
for g in iter:
if g.get("active"):
graduateds.remove(g)
iter = deepcopy(currents)
for g in iter:
if not g.get("active"):
currents.remove(g)
#############
# hindex
#############
if person.get("hindex"):
hindex = sorted(person.get("hindex"), key=doc_date_key).pop()
else:
hindex = {}
self.render(
"cv.tex",
person["_id"] + ".tex",
p=person,
title=person.get("name", ""),
aghs=aghs,
hindex=hindex,
service=service,
undergrads=undergrads,
masters=masters,
currentphds=currents,
graduatedphds=graduateds,
postdocs=postdocs,
visitors=visitors,
pubs=pubs,
names=names,
bibfile=bibfile,
education=edu,
employment=emps,
presentations=presentations,
sentencecase=sentencecase,
monthstyle=month_fullnames,
projects=projs,
pi_grants=pi_grants,
pi_amount=pi_amount,
coi_grants=coi_grants,
coi_amount=coi_amount,
coi_sub_amount=coi_sub_amount,
)
self.pdf(person["_id"])