Docs: Custom Pygments style for dark theme. (#2290)

docs/_static/style.css

@@ -1,3 +1,9 @@
 .sidebar-brand-text, .hide-on-website {
     display: none;
 }
+
+@media (prefers-color-scheme: dark) {
+    body:not([data-theme="light"]) .highlight {
+        border-left: 2px solid #94b594;
+    }
+}

docs/conf.py

@@ -57,6 +57,7 @@
 
 # The name of the Pygments (syntax highlighting) style to use.
 pygments_style = "default"
+pygments_dark_style = "gluonts-dark"
 
 # -- Options for HTML output -------------------------------------------------
 

setup.py

@@ -254,6 +254,11 @@ def run(self):
         "style_check": StyleCheckCommand,
         **version_cmdclass,
     },
+    entry_points={
+        "pygments.styles": [
+            "gluonts-dark=gluonts.meta.style:Dark",
+        ]
+    },
 )
 
 if HAS_SPHINX:

src/gluonts/meta/style.py

@@ -0,0 +1,45 @@
+# Copyright 2018 Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License").
+# You may not use this file except in compliance with the License.
+# A copy of the License is located at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# or in the "license" file accompanying this file. This file is distributed
+# on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+# express or implied. See the License for the specific language governing
+# permissions and limitations under the License.
+
+from pygments.style import Style
+from pygments.token import Comment, Keyword, Name, String, Number, Operator
+
+from . import colors
+
+
+class Dark(Style):
+    # transparent
+    background_color = "rgba(0, 0, 0, 0.0)"
+
+    styles = {
+        Comment: "italic #888",
+        Comment.Preproc: "noitalic #888",
+        Keyword: f"bold {colors.SALMON}",
+        Keyword.Pseudo: "nobold",
+        Keyword.Type: f"nobold {colors.SALMON}",
+        Operator: f"bold {colors.SALMON}",
+        Operator.Word: f"bold {colors.SALMON}",
+        Name.Builtin: colors.YELLOW,
+        Name.Function: f"{colors.GREEN}",
+        Name.Class: f"bold {colors.GREEN}",
+        Name.Namespace: f"bold {colors.GREEN}",
+        Name.Variable: colors.YELLOW,
+        Name.Constant: colors.RED,
+        Name.Label: colors.YELLOW,
+        Name.Attribute: colors.YELLOW,
+        Name.Tag: f"bold {colors.SALMON}",
+        Name.Decorator: colors.SALMON,
+        String: colors.SALMON,
+        String.Doc: "italic",
+        Number: colors.GREEN,
+    }