mirror of
https://github.com/pybricks/pybricks-api.git
synced 2026-09-12 17:46:19 +00:00
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
from docutils import nodes
|
|
from docutils.parsers.rst import Directive
|
|
from pybricks.parameters import Color
|
|
from colorsys import hsv_to_rgb
|
|
|
|
|
|
class PybricksColorDirective(Directive):
|
|
|
|
required_arguments = 1
|
|
|
|
def run(self):
|
|
# Get color name from sphinx-directive
|
|
name = self.arguments[0]
|
|
|
|
# Get Color class attribute
|
|
color = getattr(Color, name)
|
|
|
|
# Convert HSV to RGB
|
|
r, g, b = hsv_to_rgb(color.h/360, color.s/100, color.v/100)
|
|
|
|
# Convert RGB to HEX
|
|
rgbhex = '#{0:02x}{1:02x}{2:02x}'.format(
|
|
round(r*255),
|
|
round(g*255),
|
|
round(b*255)
|
|
)
|
|
|
|
# Render a small block of the given color
|
|
css = "background-color: {0}; color: {0}; width: 50px;".format(rgbhex)
|
|
|
|
if name == "WHITE":
|
|
css += "border-style: solid; border-width: 0.5px;" + \
|
|
"border-color: #666666;"
|
|
|
|
html = '<div id="test" style="{0}">_</div>'.format(css)
|
|
|
|
# Return the node
|
|
node = nodes.raw('', html, format="html")
|
|
return [node]
|
|
|
|
|
|
def setup(app):
|
|
app.add_directive_to_domain('py', 'pybricks-color', PybricksColorDirective)
|