Move pageInfo and pageBuilder classes into corresponding files.

1. Split pretty_docs.py into into a sub-package.
2. Move: ClassPageInfo + ClassPageBuilder + supporting functions -> class_page.py
3. Do the same for function, module, and type_alias
6. Replace: pretty_docs.build_md_page() -> page_info.build()
7. Everything else is just renames to make that work.

Pure refactor. No behavior changes.

PiperOrigin-RevId: 420156326

Author: MarkDaoust

MANIFEST.in

@@ -1,2 +1,2 @@
 recursive-include tools/tensorflow_docs/api_generator/gen_java/ *
-recursive-include tools/tensorflow_docs/api_generator/templates/ *
+recursive-include tools/tensorflow_docs/api_generator/pretty_docs/templates/ *

tools/tensorflow_docs/api_generator/__init__.py

@@ -19,7 +19,6 @@
 from tensorflow_docs.api_generator import doc_generator_visitor
 from tensorflow_docs.api_generator import generate_lib
 from tensorflow_docs.api_generator import parser
-from tensorflow_docs.api_generator import pretty_docs
 from tensorflow_docs.api_generator import public_api
 from tensorflow_docs.api_generator import traverse
 from tensorflow_docs.api_generator import utils

tools/tensorflow_docs/api_generator/generate_lib.py

@@ -29,11 +29,14 @@
 from tensorflow_docs.api_generator import doc_controls
 from tensorflow_docs.api_generator import doc_generator_visitor
 from tensorflow_docs.api_generator import parser
-from tensorflow_docs.api_generator import pretty_docs
 from tensorflow_docs.api_generator import public_api
 from tensorflow_docs.api_generator import reference_resolver as reference_resolver_lib
 from tensorflow_docs.api_generator import signature
 from tensorflow_docs.api_generator import traverse
+from tensorflow_docs.api_generator.pretty_docs import base_page
+
+from tensorflow_docs.api_generator.pretty_docs import docs_for_object
+
 from tensorflow_docs.api_generator.report import utils
 
 import yaml
@@ -434,7 +437,8 @@ def generate(self) -> Dict[str, Any]:
     return {'toc': toc}
 
 
-def _get_headers(page_info: parser.PageInfo, search_hints: bool) -> List[str]:
+def _get_headers(page_info: base_page.PageInfo,
+                 search_hints: bool) -> List[str]:
   """Returns the list of header lines for this page."""
   hidden = doc_controls.should_hide_from_search(page_info.py_object)
   brief_no_backticks = page_info.doc.brief.replace('`', '').strip()
@@ -559,8 +563,8 @@ def write_docs(
 
     # Generate docs for `py_object`, resolving references.
     try:
-      page_info = parser.docs_for_object(full_name, py_object, parser_config,
-                                         extra_docs)
+      page_info = docs_for_object.docs_for_object(full_name, py_object,
+                                                  parser_config, extra_docs)
       if gen_report and not full_name.startswith(
           ('tf.compat.v', 'tf.keras.backend', 'tf.numpy',
            'tf.experimental.numpy')):
@@ -572,7 +576,7 @@ def write_docs(
     path = output_dir / parser.documentation_path(full_name)
 
     content = _get_headers(page_info, search_hints)
-    content.append(pretty_docs.build_md_page(page_info))
+    content.append(page_info.build())
     text = '\n'.join(content)
     try:
       path.parent.mkdir(exist_ok=True, parents=True)

tools/tensorflow_docs/api_generator/generate_lib_test.py

@@ -29,6 +29,7 @@
 from tensorflow_docs.api_generator import generate_lib
 from tensorflow_docs.api_generator import parser
 from tensorflow_docs.api_generator import reference_resolver as reference_resolver_lib
+from tensorflow_docs.api_generator.pretty_docs import function_page
 
 import yaml
 
@@ -187,9 +188,9 @@ def test_write(self):
     self.assertTrue((output_dir / 'tf/TestModule/test_function.md').exists())
 
   def _get_test_page_info(self):
-    page_info = parser.FunctionPageInfo(
+    page_info = function_page.FunctionPageInfo(
         full_name='abc', py_object=test_function)
-    docstring_info = parser._DocstringInfo(
+    docstring_info = parser.DocstringInfo(
         brief='hello `tensorflow`',
         docstring_parts=['line1', 'line2'],
         compatibility={})