Commit 076a63e6 authored by Yoong Kang Lim's avatar Yoong Kang Lim Committed by Tim Graham
Browse files

Fixed #24883 -- Added MigrationGraph.__repr__()

parent bf07ba52

django/db/migrations/graph.py

+8 −4
Original line number Diff line number Diff line
@@ -247,10 +247,14 @@ class MigrationGraph(object): 
                    node = stack.pop()



    def __str__(self):

        return "Graph: %s nodes, %s edges" % (

            len(self.nodes),

            sum(len(node.parents) for node in self.node_map.values()),

        )

        return 'Graph: %s nodes, %s edges' % self._nodes_and_edges()



    def __repr__(self):

        nodes, edges = self._nodes_and_edges()

        return '<%s: nodes=%s, edges=%s>' % (self.__class__.__name__, nodes, edges)



    def _nodes_and_edges(self):

        return len(self.nodes), sum(len(node.parents) for node in self.node_map.values())



    def make_state(self, nodes=None, at_end=True, real_apps=None):

        """

tests/migrations/test_graph.py

+1 −0
Original line number Diff line number Diff line
@@ -283,3 +283,4 @@ class GraphTests(SimpleTestCase): 
        graph.add_dependency("app_a.0003", ("app_a", "0003"), ("app_b", "0002"))



        self.assertEqual(force_text(graph), "Graph: 5 nodes, 3 edges")

        self.assertEqual(repr(graph), "<MigrationGraph: nodes=5, edges=3>")