rich.pretty.pprint has an argument expand_all. Is there a way to get that functionality in the Console or Style api?

[WillMeakin]

I want either Console or Style to have the behaviour of pprint's expand_all argument. Any way to do this easily?

class MyClass:
	def __init__(self, one=1, two=2, three=3):
		self.one = one
		self.two = two
		self.three = three
	def __rich__(self):
		return self.__dict__

my_obj = MyClass()


my_style = Style(color="yellow")
my_console = Console(style=my_style, max_width)

my_console.print("console print: <-- this text is yellow", my_obj)

pprint("pprint: <-- this text is not yellow (wrong)", console=my_console)
pprint(my_obj.__dict__, console=my_console, expand_all=True)

OUTPUT:

console print: <-- this text is yellow
{'one': 1, 'two': 2, 'three': 3}
'pprint: <-- this text is not yellow (wrong)'
{
│ 'one': 1,
│ 'two': 2,
│ 'three': 3
}

I want to use console instead of pprint because:

1. console has a style property and console.print() has a style arg. Passing my_console to pprint's console arg doesn't seem to obey my_console's style (text is not yellow)
2. console.print(obj) automatically calls obj.rich()
3. console.print accepts a variable number of outputs to print.

[willmcgugan]

Have a look at how pprint is implemented.

[WillMeakin]

Thanks for the suggestion.
So it seems pprint is calling console.print(Pretty(obj))
The Pretty object argument takes an expand_all argument and works as intended.

Though passing a Pretty object still leaves the final 3 points a problem:
1: Pretty objects don't seem to obey the styling of the console. Not sure why, maybe intentionally?

2: While class Pretty has a __rich_console__ method, console.print(Pretty(obj)) still prints out the object's memory address. Tracing back Pretty.__rich_console__() it seems to be something to do with traverse not calling the obj's __rich__ nor __rich_console__ method? Not sure if bug or intended.

3: Is it worth adding an argument to Console.print to let the object renderables be wrapped in a Pretty object? And then args to pass to that Pretty object?

I don't know if this is worth exploring. Having looked at the codebase I'm now interested in contributing :)

[WillMeakin]

Follow up after some further exploring: It seems console.print does indeed apply the console's style to the Pretty object. However if we have pretty = Pretty(my_obj) , rather than printing out the my_obj.rich and applying the style colour to the text, it's calling maybe pretty.repr or something and hence applying some global String styling. This can be seen in the original output by the "wrong" text is in quotes and the correct text is not in quotes. Just an interesting behavioural quirk, not really an issue I guess?