How to render my attribute values as hex #255

Yep:

% cat hex.py 
class MessageType:
    BEST_MESSAGE = 0x0123
    WORST_MESSAGE = 0xB33F

% python -m ast hex.py
Module(
   body=[
      ClassDef(
         name='MessageType',
         body=[
            Assign(
               targets=[
                  Name(id='BEST_MESSAGE', ctx=Store())],
               value=Constant(value=291)),
            Assign(
               targets=[
                  Name(id='WORST_MESSAGE', ctx=Store())],
               value=Constant(value=45887))])])

I don't see how to prevent that 😕

Thanks for your response!

Damn, just tried it with griffe dump too and they don't preserve the original value either.

Fun discovery: I see that astexplorer.net does preserve the original value:

They use https://github.com/codecombat/filbert which is a js parser. So not relevant but interesting nonetheless.

I wonder if there is a way to postprocess these values. I will look into extra_javascript.

Worst case I'll add a docstring with the value manually put in. Unfortunately for the use case in mind: our team speak and reason about all these message IDs in hex. So the document (and search) isn't very useful without it.

This worked amazingly, thank you! Also it has taught me a lot about the extensions structure.

One note, I had to change the extension to

import griffe


class ToHex(griffe.Extension):
    def __init__(self, objects: list[str]) -> None:
        self._objects = set(objects)

    def on_class_members(self, *, cls: griffe.Class, **kwargs) -> None:
        if cls.path in self._objects:
            for attr in cls.attributes.values():
                if isinstance(attr, griffe.Attribute):  # just to be sure
                    attr.value = hex(int(attr.value))

RE the just to be sure line: When I inherit a baseclass of enum the type comes out as Attrribute, not ExprConstant.
RE value.value: I think maybe it was a copy paste error?

Thanks again :)

Great! I definitely meant to check the type of attr.value, which is the right operand in BEST_MESSAGE = 0x0123. attr itself will always be an Attribute instance, since we get it from its parent attributes property. Then attr.value.value was on purpose too, since if attr.value is an ExprConstant, it will have a value attribute storing the actual constant value (the integer) 🙂 See https://mkdocstrings.github.io/griffe/reference/api/expressions/#griffe.ExprConstant.value.

But your solution works too, because attribute values can be expressions (like ExprConstant) or strings directly, so you can definitely keep this shortcut, it's supported 🙂 (you just don't need the isinstance check). I'm not sure how int(attr.value) works here though 🤔

Huh, turns out attr.value is the integer (represented as a string) directly. Hmmm. I thought it would be an ExprConstant, my bad!

Right, the expression builder doesn't ever instantiate ExprConstant 😂 _build_constant returns the value directly. Well, I'll add a note to the code. Or I guess I can just remove the ExprConstant class.