mirror of
https://github.com/Unstructured-IO/unstructured.git
synced 2025-12-05 03:23:03 +00:00
b58d0dde3e
### Description
Add new class to wrap base config and cli mixin to help with typing:
```python
class CliConfig(BaseConfig, CliMixin):
pass
```
54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
import typing as t
|
|
from dataclasses import dataclass
|
|
|
|
import click
|
|
|
|
from unstructured.ingest.cli.base.src import BaseSrcCmd
|
|
from unstructured.ingest.cli.interfaces import (
|
|
CliConfig,
|
|
CliRecursiveConfig,
|
|
DelimitedString,
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class NotionCliConfig(CliConfig):
|
|
notion_api_key: str
|
|
page_ids: t.Optional[t.List[str]]
|
|
database_ids: t.Optional[t.List[str]]
|
|
max_retries: t.Optional[int] = None
|
|
max_time: t.Optional[float] = None
|
|
|
|
@staticmethod
|
|
def get_cli_options() -> t.List[click.Option]:
|
|
options = [
|
|
click.Option(
|
|
["--notion-api-key"],
|
|
required=True,
|
|
type=str,
|
|
help="API key for Notion api",
|
|
),
|
|
click.Option(
|
|
["--page-ids"],
|
|
default=None,
|
|
type=DelimitedString(),
|
|
help="Notion page IDs to pull text from",
|
|
),
|
|
click.Option(
|
|
["--database-ids"],
|
|
default=None,
|
|
type=DelimitedString(),
|
|
help="Notion database IDs to pull text from",
|
|
),
|
|
]
|
|
return options
|
|
|
|
|
|
def get_base_src_cmd() -> BaseSrcCmd:
|
|
cmd_cls = BaseSrcCmd(
|
|
cmd_name="notion",
|
|
cli_config=NotionCliConfig,
|
|
additional_cli_options=[CliRecursiveConfig],
|
|
)
|
|
return cmd_cls
|