Customizable HTML title in GraphiQL

singingwolfboy · singingwolfboy · commit 6e2d7c902e62 · 2017-08-14T15:57:33.000-07:00
diff --git a/flask_graphql/graphqlview.py b/flask_graphql/graphqlview.py
@@ -19,6 +19,7 @@ class GraphQLView(View):
     graphiql = False
     graphiql_version = None
     graphiql_template = None
+    graphiql_html_title = None
     middleware = None
     batch = False
 
@@ -51,6 +52,7 @@ def render_graphiql(self, params, result):
             result=result,
             graphiql_version=self.graphiql_version,
             graphiql_template=self.graphiql_template,
+            graphiql_html_title=self.graphiql_html_title,
         )
 
     format_error = staticmethod(default_format_error)
diff --git a/flask_graphql/render_graphiql.py b/flask_graphql/render_graphiql.py
@@ -12,6 +12,7 @@
 <!DOCTYPE html>
 <html>
 <head>
+  <title>{{graphiql_html_title|default("GraphiQL", true)}}</title>
   <style>
     html, body {
       height: 100%;
@@ -123,13 +124,15 @@
 </html>'''
 
 
-def render_graphiql(params, result, graphiql_version=None, graphiql_template=None):
+def render_graphiql(params, result, graphiql_version=None,
+                    graphiql_template=None, graphiql_html_title=None):
     graphiql_version = graphiql_version or GRAPHIQL_VERSION
     template = graphiql_template or TEMPLATE
 
     return render_template_string(
         template,
         graphiql_version=graphiql_version,
+        graphiql_html_title=graphiql_html_title,
         result=result,
         params=params
     )
diff --git a/tests/test_graphiqlview.py b/tests/test_graphiqlview.py
@@ -26,3 +26,14 @@ def test_graphiql_renders_pretty(client):
     ).replace("\"","\\\"").replace("\n","\\n")
 
     assert pretty_response in response.data.decode('utf-8')
+
+
+def test_graphiql_default_title(client):
+    response = client.get(url_for('graphql'), headers={'Accept': 'text/html'})
+    assert '<title>GraphiQL</title>' in response.data.decode('utf-8')
+
+
+@pytest.mark.parametrize('app', [create_app(graphiql=True, graphiql_html_title="Awesome")])
+def test_graphiql_custom_title(client):
+    response = client.get(url_for('graphql'), headers={'Accept': 'text/html'})
+    assert '<title>Awesome</title>' in response.data.decode('utf-8')
