Is it possible to bring your own data structure to rich and have rich handle truncation behavior?

[cpcloud]

Rich is really great. I love the way things look with it!

I'm having some trouble figuring out how to customize the rendering of a deeply nested tree data structure. The data structure represents a schema of highly nested but uniform JSON data.

It looks like this:

struct<
  a: int64,
  b: array<
      struct<z: string>,
     >,
  c: float64
>

The one I'm dealing with is much bigger and more complex and showing the entire schema is undesirable.

I don't want to print this using __rich_repr__ because this is a user-facing display, and showing the guts of the underlying objects is undesirable.

Ideally I can somehow hook into rich with this custom tree data structure and have rich automatically truncate the depth of of the structure for me.

Is there a way to do this? Can I somehow leverage the existing sequence printing code that automatically truncate by bringing my own sequence? How would I do that?

I've tried implementing the following things to no avail:

1. __rich_repr__, this is too verbose and not what I want stylistically
2. __rich_console__, this doesn't appear to automatically handle indentation for nesting and doesn't automatically truncate.

[willmcgugan]

I'll need a little more information on what the output you want, and what you've tried so far...

[willmcgugan]

You are straying quite a bit from a tradition repr format, which is what __rich_repr__ was designed to output. You may be able to copy and modify the code in pretty.py to your requirements, but it's not something you can configure out of the box.

[cpcloud]

Right, but I don't want __repr__ :)

Is it possible to have rich automatically indent/truncate in any context except __rich_repr__?

[cpcloud]

What if I hand a custom sequence type to rich? Would that be automatically indented/truncated and allow me to customize all the punctuation? I'd be happy to do that.

[willmcgugan]

Have a look at pretty.py which does all the indentation / truncation.

[cpcloud]

Here's the PR where I'm using rich.

Ideally I could bring my own sequence type instead of having to add to the big if isinstance set of checks in pretty.py, but I guess if that's the canonical way to hook a custom data structure into rich so be it!