From 8b38d63b7d11b29382b9e432242d2fec8892e019 Mon Sep 17 00:00:00 2001
From: Mayur <mayurnewase111@gmail.com>
Date: Wed, 3 Mar 2021 17:22:54 +0530
Subject: [PATCH] changes added (#13431)

---
 .../components/controls/VizTypeControl.jsx    |  2 +-
 superset/examples/energy.py                   | 16 ++++++------
 superset/viz.py                               | 25 -------------------
 tests/fixtures/energy_dashboard.py            | 13 +++++-----
 4 files changed, 15 insertions(+), 41 deletions(-)

diff --git a/superset-frontend/src/explore/components/controls/VizTypeControl.jsx b/superset-frontend/src/explore/components/controls/VizTypeControl.jsx
index 085fdd8da64df..df3bfd35dcd95 100644
--- a/superset-frontend/src/explore/components/controls/VizTypeControl.jsx
+++ b/superset-frontend/src/explore/components/controls/VizTypeControl.jsx
@@ -85,7 +85,7 @@ const DEFAULT_ORDER = [
   'partition',
   'event_flow',
   'deck_path',
-  'directed_force',
+  'graph_chart',
   'world_map',
   'paired_ttest',
   'para',
diff --git a/superset/examples/energy.py b/superset/examples/energy.py
index cd844a55329de..b101905af9e5a 100644
--- a/superset/examples/energy.py
+++ b/superset/examples/energy.py
@@ -94,23 +94,21 @@ def load_energy(
 
     slc = Slice(
         slice_name="Energy Force Layout",
-        viz_type="directed_force",
+        viz_type="graph_chart",
         datasource_type="table",
         datasource_id=tbl.id,
         params=textwrap.dedent(
             """\
         {
-            "charge": "-500",
-            "collapsed_fieldsets": "",
-            "groupby": [
-                "source",
-                "target"
-            ],
-            "link_length": "200",
+            "source": "source",
+            "target": "target",
+            "edgeLength": 400,
+            "repulsion": 1000,
+            "layout": "force",
             "metric": "sum__value",
             "row_limit": "5000",
             "slice_name": "Force",
-            "viz_type": "directed_force"
+            "viz_type": "graph_chart"
         }
         """
         ),
diff --git a/superset/viz.py b/superset/viz.py
index 1183418877bdf..b00a945ccc31a 100644
--- a/superset/viz.py
+++ b/superset/viz.py
@@ -1898,31 +1898,6 @@ def visit(vertex: str) -> Optional[Tuple[str, str]]:
         return recs
 
 
-class DirectedForceViz(BaseViz):
-
-    """An animated directed force layout graph visualization"""
-
-    viz_type = "directed_force"
-    verbose_name = _("Directed Force Layout")
-    credits = 'd3noob @<a href="http://bl.ocks.org/d3noob/5141278">bl.ocks.org</a>'
-    is_timeseries = False
-
-    def query_obj(self) -> QueryObjectDict:
-        qry = super().query_obj()
-        if len(self.form_data["groupby"]) != 2:
-            raise QueryObjectValidationError(_("Pick exactly 2 columns to 'Group By'"))
-        qry["metrics"] = [self.form_data["metric"]]
-        if self.form_data.get("sort_by_metric", False):
-            qry["orderby"] = [(qry["metrics"][0], False)]
-        return qry
-
-    def get_data(self, df: pd.DataFrame) -> VizData:
-        if df.empty:
-            return None
-        df.columns = ["source", "target", "value"]
-        return df.to_dict(orient="records")
-
-
 class ChordViz(BaseViz):
 
     """A Chord diagram"""
diff --git a/tests/fixtures/energy_dashboard.py b/tests/fixtures/energy_dashboard.py
index 68d142f65d832..42fb46ead7aa9 100644
--- a/tests/fixtures/energy_dashboard.py
+++ b/tests/fixtures/energy_dashboard.py
@@ -141,16 +141,17 @@ def _get_energy_slices():
         },
         {
             "slice_title": "Energy Force Layout",
-            "viz_type": "directed_force",
+            "viz_type": "graph_chart",
             "params": {
-                "charge": "-500",
-                "collapsed_fieldsets": "",
-                "groupby": ["source", "target"],
-                "link_length": "200",
+                "source": "source",
+                "target": "target",
+                "edgeLength": 400,
+                "repulsion": 1000,
+                "layout": "force",
                 "metric": "sum__value",
                 "row_limit": "5000",
                 "slice_name": "Force",
-                "viz_type": "directed_force",
+                "viz_type": "graph_chart",
             },
         },
         {