def view_list(request): """View the list of all personas""" graph.update_if_needed() if len(graph) == 0: return render_to_response('geneaprove/firsttime.html', context_instance=RequestContext(request)) styles = Styles(style_rules, graph=graph, decujus=1) # ??? Why "1" all = extended_personas(nodes=None, styles=styles, event_types=event_types_for_pedigree, graph=graph, as_css=True) all = [p for p in all.itervalues()] all.sort(key=lambda x: x.surname) all = [{ 'surname': p.surname, 'given_name': p.given_name, 'birth': p.birth, 'death': p.death, 'marriage': p.marriage } for p in all] data = { 'persons': all, } return HttpResponse(to_json(data), content_type='application/json')
def __get_json_sosa_tree(graph, id, max_levels, style_rules, last_descendant_known=-1, maxdepthDescendants=1, last_gen_known=-1): """ :param last_gen_known: is the number of the last generation for which the client already has data, and thus do not need to be sent again. -1 to retrieve all. :param maxdepthDescendants: The number of generations for which we compute the children. """ decujus = graph.node_from_id(id) styles = Styles(style_rules, graph, decujus=decujus.main_id) distance = dict() ancestors = graph.people_in_tree( id=decujus.main_id, maxdepthAncestors=max_levels - 1, maxdepthDescendants=0, distance=distance) ancestors = [a for a in ancestors if distance[a] >= last_gen_known] descendants = graph.people_in_tree( id=decujus.main_id, maxdepthAncestors=0, distance=distance, maxdepthDescendants=maxdepthDescendants) descendants = [ a for a in descendants if a != decujus and distance[a] >= last_descendant_known] sosa_tree = dict() marriage = dict() children = {} persons = {} all_person_nodes = set(ancestors).union(descendants) if all_person_nodes: persons = extended_personas( all_person_nodes, styles, event_types=event_types_for_pedigree, graph=graph) def add_parents(p): p.generation = distance[graph.node_from_id(p.id)] if p.generation >= max_levels: return fathers = graph.fathers(p.id) mothers = graph.mothers(p.id) p.parents = [ None if not fathers else persons.get(fathers[0].main_id, None), None if not mothers else persons.get(mothers[0].main_id, None)] for pa in p.parents: if pa: add_parents(pa) def add_children(p, gen): p.children = [] sorted = [(persons[node.main_id] if node.main_id in persons else None, node) for node in graph.children(p.id)] sorted.sort( key=lambda c: c[0].birth.Date if c[0] and c[0].birth else None) for c in sorted: if c[0]: c[0].generation = -gen # distance[c[1]] p.children.append(c[0]) if gen < maxdepthDescendants: add_children(c[0], gen + 1) main = persons[decujus.main_id] add_parents(main) add_children(main, gen=1) # We will however return a simpler version of the information computed # above (which includes all known events for the persons) show_age = False def person_to_json_for_pedigree(obj): if isinstance(obj, models.Persona): d = obj.death if show_age and obj.birth: if d: if d.Date: d.Date += " (age %s)" % ( str(d.Date.years_since(obj.birth.Date)), ) else: d = {Date: " (age %s)" % ( str(DateRange.today().years_since(obj.birth.Date)), )} return { 'id': obj.id, 'givn': obj.given_name, 'surn': obj.surname, 'sex': obj.sex, 'generation': obj.generation, 'parents': obj.parents if hasattr(obj, 'parents') else None, 'children': obj.children if hasattr(obj, 'children') else None, 'style': obj.styles, 'birth': obj.birth, 'marriage': obj.marriage, 'death': d} return to_json( obj= {'generations': max_levels, 'descendants': maxdepthDescendants, 'decujus': main, 'styles': styles.all_styles()}, custom=person_to_json_for_pedigree)