|
57 | 57 | 'google.cloud.vision.__init__', |
58 | 58 | 'google.cloud.vision.fixtures', |
59 | 59 | ]) |
| 60 | +PACKAGES = ( |
| 61 | + '', |
| 62 | +) |
60 | 63 |
|
61 | 64 |
|
62 | 65 | class SphinxApp(object): |
@@ -127,17 +130,24 @@ def main(build_root='_build'): |
127 | 130 | :param build_root: The root of the directory where docs are built into. |
128 | 131 | Defaults to ``_build``. |
129 | 132 | """ |
| 133 | + if build_root is None: |
| 134 | + parser = get_parser() |
| 135 | + args = parser.parse_args() |
| 136 | + build_root = args.build_root |
| 137 | + |
130 | 138 | object_inventory_relpath = os.path.join(build_root, 'html', 'objects.inv') |
131 | 139 |
|
132 | 140 | mock_uri = '' |
133 | 141 | inventory = fetch_inventory(SphinxApp, mock_uri, |
134 | 142 | object_inventory_relpath) |
135 | 143 | sphinx_mods = set(inventory['py:module'].keys()) |
136 | 144 |
|
137 | | - library_dir = os.path.join(BASE_DIR, 'google', 'cloud') |
138 | | - public_mods = get_public_modules(library_dir, |
139 | | - base_package='google.cloud') |
140 | | - public_mods = set(public_mods) |
| 145 | + public_mods = set() |
| 146 | + for package in PACKAGES: |
| 147 | + library_dir = os.path.join(BASE_DIR, package, 'google', 'cloud') |
| 148 | + package_mods = get_public_modules(library_dir, |
| 149 | + base_package='google.cloud') |
| 150 | + public_mods.update(package_mods) |
141 | 151 |
|
142 | 152 | if not sphinx_mods <= public_mods: |
143 | 153 | unexpected_mods = sphinx_mods - public_mods |
@@ -173,6 +183,4 @@ def get_parser(): |
173 | 183 |
|
174 | 184 |
|
175 | 185 | if __name__ == '__main__': |
176 | | - parser = get_parser() |
177 | | - args = parser.parse_args() |
178 | | - main(build_root=args.build_root) |
| 186 | + main(build_root=None) |
0 commit comments