18F/openFEC

import networkx as nx

def get_graph():
    """Build a `DiGraph` that captures dependencies between database migration
    tasks. Each node represents a migration script, and each edge represents
    a dependency between tasks. Tasks can be ordered using topological sort.
    """
    graph = nx.DiGraph()

    MATERIALIZED_VIEWS = [
        'audit_case',
        'cand_cmte_linkage',
        'candidate_aggregates',
        'candidate_detail',
        'candidate_election',
        'candidate_flags',
        'candidate_fulltext',
        'candidate_history',
        'candidate_history_latest',
        'committee_detail',
        'committee_fulltext',
        'committee_history',
        'communication_cost',
        'communication_cost_by_candidate',
        'election_outcome',
        'electioneering',
        'electioneering_by_candidate',
        'elections_list',
        'filing_amendments_all',
        'filing_amendments_house_senate',
        'filing_amendments_pac_party',
        'filing_amendments_presidential',
        'filings',
        'large_aggregates',
        'rad_analyst',
        'reports_house_senate',
        'reports_ie',
        'reports_pac_party',
        'reports_presidential',
        'sched_a_by_size_merged',
        'sched_a_by_state_recipient_totals',
        'sched_c',
        'sched_e_by_candidate',
        'sched_f',
        'totals_candidate_committee',
        'totals_combined',
        'totals_house_senate',
        'totals_ie',
        'totals_pac_party',
        'totals_presidential',
    ]
    graph.add_nodes_from(MATERIALIZED_VIEWS)

    graph.add_edge('candidate_history', 'candidate_detail')
    graph.add_edge('candidate_detail', 'candidate_election')

    graph.add_edges_from([
        ('candidate_history', 'candidate_history_latest'),
        ('candidate_election', 'candidate_history_latest')
    ])

    graph.add_edge('committee_history', 'committee_detail')

    graph.add_edges_from([
        ('candidate_history', 'filings'),
        ('committee_history', 'filings'),
    ])

    graph.add_edges_from([
        ('filing_amendments_presidential', 'filings'),
        ('filing_amendments_house_senate', 'filings'),
        ('filing_amendments_pac_party', 'filings'),
        ('filing_amendments_all', 'filings'),
    ])

    graph.add_edges_from([
        ('candidate_history', 'candidate_flags'),
        ('candidate_aggregates', 'candidate_flags'),
    ])

    graph.add_edges_from([
        ('filing_amendments_presidential', 'reports_presidential'),
        ('filing_amendments_house_senate', 'reports_house_senate'),
        ('filing_amendments_pac_party', 'reports_pac_party'),
        ('filing_amendments_all', 'reports_presidential'),
        ('filing_amendments_all', 'reports_house_senate'),
        ('filing_amendments_all', 'reports_pac_party'),
    ])

    graph.add_edges_from([
        ('totals_combined', 'totals_house_senate'),
        ('totals_combined', 'totals_presidential'),
        ('totals_combined', 'totals_pac_party'),
        ('totals_combined', 'totals_ie'),
    ])

    graph.add_edges_from([
        ('committee_detail', 'committee_fulltext'),
        ('totals_combined', 'committee_fulltext'),
    ])

    graph.add_edge('committee_detail', 'totals_pac_party')

    graph.add_edges_from([
        ('candidate_detail', 'candidate_fulltext'),
        ('totals_combined', 'candidate_fulltext'),
    ])

    graph.add_edges_from([
        ('totals_combined', 'sched_a_by_size_merged'),
    ])

    graph.add_edges_from([
        ('totals_house_senate', 'candidate_aggregates'),
        ('totals_presidential', 'candidate_aggregates'),
        ('candidate_election', 'candidate_aggregates'),
        ('cand_cmte_linkage', 'candidate_aggregates'),
    ])

    graph.add_edges_from([
        ('filings', 'totals_combined'),
    ])

    graph.add_edges_from([
        ('totals_pac_party', 'large_aggregates'),
        ('totals_house_senate', 'large_aggregates'),
        ('totals_combined', 'large_aggregates'),
    ])

    graph.add_edges_from([
        ('candidate_election', 'totals_candidate_committee'),
        ('filings', 'totals_candidate_committee'),
        ('cand_cmte_linkage', 'totals_candidate_committee'),
    ])

    graph.add_edge('committee_history', 'communication_cost')
    graph.add_edge('committee_detail', 'sched_a_by_state_recipient_totals')

    return graph