unstructured-ingest 0.6.4__py3-none-any.whl → 0.7.1__py3-none-any.whl

unstructured_ingest/cli/utils.py

@@ -1,205 +0,0 @@
-import typing as t
-from dataclasses import fields, is_dataclass
-from gettext import gettext as _
-
-import click
-
-from unstructured_ingest.cli.interfaces import (
-    CliChunkingConfig,
-    CliConfig,
-    CliEmbeddingConfig,
-    CliPartitionConfig,
-    CliPermissionsConfig,
-    CliProcessorConfig,
-    CliReadConfig,
-    CliRetryStrategyConfig,
-)
-from unstructured_ingest.interfaces import BaseConfig
-from unstructured_ingest.logger import logger
-
-
-def conform_click_options(options: dict):
-    # Click sets all multiple fields as tuple, this needs to be updated to list
-    for k, v in options.items():
-        if isinstance(v, tuple):
-            options[k] = list(v)
-
-
-def extract_config(flat_data: dict, config: t.Type[BaseConfig]) -> BaseConfig:
-    """
-    To be able to extract a nested dataclass from a flat dictionary (as in one coming
-    from a click-based options input), the config class is dynamically looked through for
-    nested dataclass fields and new nested dictionaries are created to conform to the
-    shape the overall class expects when parsing from a dict. During the process, this will create
-    copies of the original dictionary to avoid pruning fields but this isn't a
-    problem since the `from_dict()` method ignores unneeded values.
-
-    Not handling more complex edge cases for now such as nested types i.e Union[List[List[...]]]
-    """
-
-    def conform_dict(inner_d: dict, inner_config: t.Type[BaseConfig]):
-        # Catch edge cases (i.e. Dict[str, ...]) where underlying type is not a concrete Class,
-        # causing 'issubclass() arg 1 must be a class' errors, return False
-        def is_subclass(instance, class_type) -> bool:
-            try:
-                return issubclass(instance, class_type)
-            except Exception:
-                return False
-
-        dd = inner_d.copy()
-        for field in fields(inner_config):
-            f_type = field.type
-            # Handle the case where the type of a value if a Union (possibly optional)
-            if t.get_origin(f_type) is t.Union:
-                union_values = t.get_args(f_type)
-                # handle List types
-                union_values = [
-                    t.get_args(u)[0] if t.get_origin(u) is list else u for u in union_values
-                ]
-                # Ignore injected NoneType when optional
-                concrete_union_values = [v for v in union_values if not is_subclass(v, type(None))]
-                dataclass_union_values = [v for v in concrete_union_values if is_dataclass(v)]
-                non_dataclass_union_values = [
-                    v for v in concrete_union_values if not is_dataclass(v)
-                ]
-                if not dataclass_union_values:
-                    continue
-                # Check if the key for this field already exists in the dictionary,
-                # if so it might map to one of these non dataclass fields and this
-                # can't be enforced
-                if non_dataclass_union_values and field.name in dd:
-                    continue
-                if len(dataclass_union_values) > 1:
-                    logger.warning(
-                        "more than one dataclass type possible for field {}, "
-                        "not extracting: {}".format(field.name, ", ".join(dataclass_union_values))
-                    )
-                    continue
-                f_type = dataclass_union_values[0]
-            origin = t.get_origin(f_type)
-            if origin:
-                f_type = origin
-            if is_subclass(f_type, BaseConfig):
-                dd[field.name] = conform_dict(inner_d=dd, inner_config=f_type)
-        return dd
-
-    adjusted_dict = conform_dict(inner_d=flat_data, inner_config=config)
-    return config.from_dict(adjusted_dict, apply_name_overload=False)
-
-
-def extract_configs(
-    data: dict,
-    extras: t.Optional[t.Dict[str, t.Type[BaseConfig]]] = None,
-    validate: t.Optional[t.List[t.Type[BaseConfig]]] = None,
-    add_defaults: bool = True,
-) -> t.Dict[str, BaseConfig]:
-    """
-    Extract all common configs used across CLI command and validate that any
-    command-specific configs have all their needed information from the Click
-    options that are passed in during invocation.
-    """
-    validate = validate if validate else []
-    res = (
-        {
-            "read_config": extract_config(flat_data=data, config=CliReadConfig),
-            "partition_config": extract_config(flat_data=data, config=CliPartitionConfig),
-            "embedding_config": extract_config(flat_data=data, config=CliEmbeddingConfig),
-            "chunking_config": extract_config(flat_data=data, config=CliChunkingConfig),
-            "processor_config": extract_config(flat_data=data, config=CliProcessorConfig),
-            "permissions_config": extract_config(flat_data=data, config=CliPermissionsConfig),
-            "retry_strategy_config": extract_config(flat_data=data, config=CliRetryStrategyConfig),
-        }
-        if add_defaults
-        else {}
-    )
-    if extras:
-        for k, conf in extras.items():
-            try:
-                res[k] = extract_config(flat_data=data, config=conf)
-            except Exception as e:
-                logger.error(f"failed to extract config from {conf.__name__}")
-                raise e
-    for v in validate:
-        try:
-            extract_config(flat_data=data, config=v)
-        except Exception as e:
-            raise Exception(f"failed to validate config {v.__name__}") from e
-
-    return res
-
-
-def add_options(
-    cmd: click.Command, extras: t.List[t.Type[CliConfig]], is_src: bool = True
-) -> click.Command:
-    configs: t.List[t.Type[CliConfig]] = (
-        [
-            CliPartitionConfig,
-            CliReadConfig,
-            CliEmbeddingConfig,
-            CliChunkingConfig,
-            CliProcessorConfig,
-            CliPermissionsConfig,
-            CliRetryStrategyConfig,
-        ]
-        if is_src
-        else []
-    )
-    # make sure what's unique to this cmd appears first
-    extras.extend(configs)
-    for config in extras:
-        try:
-            config.add_cli_options(cmd=cmd)
-        except ValueError as e:
-            raise ValueError(f"failed to set configs from {config.__name__}: {e}")
-    return cmd
-
-
-class Group(click.Group):
-    def parse_args(self, ctx, args):
-        """
-        This allows for subcommands to be called with the --help flag without breaking
-        if parent command is missing any of its required parameters
-        """
-
-        try:
-            return super().parse_args(ctx, args)
-        except click.MissingParameter:
-            if "--help" not in args:
-                raise
-
-            # remove the required params so that help can display
-            for param in self.params:
-                param.required = False
-            return super().parse_args(ctx, args)
-
-    def format_commands(self, ctx: click.Context, formatter: click.HelpFormatter) -> None:
-        """
-        Copy of the original click.Group format_commands() method but replacing
-        'Commands' -> 'Destinations'
-        """
-        commands = []
-        for subcommand in self.list_commands(ctx):
-            cmd = self.get_command(ctx, subcommand)
-            # What is this, the tool lied about a command.  Ignore it
-            if cmd is None:
-                continue
-            if cmd.hidden:
-                continue
-
-            commands.append((subcommand, cmd))
-
-        # allow for 3 times the default spacing
-        if len(commands):
-            if formatter.width:
-                limit = formatter.width - 6 - max(len(cmd[0]) for cmd in commands)
-            else:
-                limit = -6 - max(len(cmd[0]) for cmd in commands)
-
-            rows = []
-            for subcommand, cmd in commands:
-                help = cmd.get_short_help_str(limit)
-                rows.append((subcommand, help))
-
-            if rows:
-                with formatter.section(_("Destinations")):
-                    formatter.write_dl(rows)

unstructured_ingest/connector/airtable.py

@@ -1,309 +0,0 @@
-import typing as t
-from dataclasses import dataclass, field
-from datetime import datetime
-from pathlib import Path
-
-from unstructured_ingest.enhanced_dataclass import enhanced_field
-from unstructured_ingest.error import SourceConnectionError, SourceConnectionNetworkError
-from unstructured_ingest.interfaces import (
-    AccessConfig,
-    BaseConnectorConfig,
-    BaseSingleIngestDoc,
-    BaseSourceConnector,
-    IngestDocCleanupMixin,
-    SourceConnectorCleanupMixin,
-    SourceMetadata,
-)
-from unstructured_ingest.logger import logger
-from unstructured_ingest.utils.dep_check import requires_dependencies
-
-if t.TYPE_CHECKING:
-    from pyairtable import Api
-
-
-@dataclass
-class AirtableAccessConfig(AccessConfig):
-    personal_access_token: str = enhanced_field(sensitive=True)
-
-
-@dataclass
-class SimpleAirtableConfig(BaseConnectorConfig):
-    """Connector config where:
-    auth_token is the authentication token to authenticate into Airtable.
-
-    Check https://support.airtable.com/docs/airtable-api-key-deprecation-notice
-    for more info on authentication.
-    """
-
-    access_config: AirtableAccessConfig
-    list_of_paths: t.Optional[str] = None
-
-
-@dataclass
-class AirtableTableMeta:
-    """Metadata specifying a table id, a base id which the table is stored in,
-    and an t.Optional view id in case particular rows and fields are to be ingested"""
-
-    base_id: str
-    table_id: str
-    view_id: t.Optional[str] = None
-
-
-@dataclass
-class AirtableIngestDoc(IngestDocCleanupMixin, BaseSingleIngestDoc):
-    """Class encapsulating fetching a doc and writing processed results (but not
-    doing the processing).
-
-    Current implementation creates an Airtable connection object
-    to fetch each document, rather than creating a it for each thread.
-    """
-
-    connector_config: SimpleAirtableConfig
-    table_meta: AirtableTableMeta
-    registry_name: str = "airtable"
-
-    @property
-    def filename(self):
-        return (
-            Path(self.read_config.download_dir)
-            / self.table_meta.base_id
-            / f"{self.table_meta.table_id}.csv"
-        ).resolve()
-
-    @property
-    def _output_filename(self):
-        """Create output file path based on output directory, base id, and table id"""
-        output_file = f"{self.table_meta.table_id}.json"
-        return Path(self.processor_config.output_dir) / self.table_meta.base_id / output_file
-
-    @property
-    def record_locator(self) -> t.Optional[t.Dict[str, t.Any]]:
-        return {
-            "base_id": self.table_meta.base_id,
-            "table_id": self.table_meta.table_id,
-            "view_id": self.table_meta.view_id,
-        }
-
-    @property
-    def version(self) -> t.Optional[str]:
-        return None
-
-    @requires_dependencies(["pyairtable"], extras="airtable")
-    def _query_table(self):
-        from pyairtable import Api
-
-        api = Api(self.connector_config.access_config.personal_access_token)
-        table = api.table(self.table_meta.base_id, self.table_meta.table_id)
-        table_url = table.url
-        rows = table.all(
-            view=self.table_meta.view_id,
-        )
-        return rows, table_url
-
-    @SourceConnectionNetworkError.wrap
-    def _get_table_rows(self):
-        rows, table_url = self._query_table()
-
-        if len(rows) == 0:
-            logger.info("Empty document, retrieved table but it has no rows.")
-        return rows, table_url
-
-    def update_source_metadata(self, **kwargs):
-        """Gets file metadata from the current table."""
-
-        rows, table_url = kwargs.get("rows_tuple", self._get_table_rows())
-        if rows is None or len(rows) < 1:
-            self.source_metadata = SourceMetadata(
-                exists=False,
-            )
-            return
-        dates = [r.get("createdTime", "") for r in rows]
-        dates.sort()
-
-        date_created = datetime.strptime(
-            dates[0],
-            "%Y-%m-%dT%H:%M:%S.%fZ",
-        ).isoformat()
-
-        date_modified = datetime.strptime(
-            dates[-1],
-            "%Y-%m-%dT%H:%M:%S.%fZ",
-        ).isoformat()
-
-        self.source_metadata = SourceMetadata(
-            date_created=date_created,
-            date_modified=date_modified,
-            source_url=table_url,
-            exists=True,
-        )
-
-    @SourceConnectionError.wrap
-    @requires_dependencies(["pandas"])
-    @BaseSingleIngestDoc.skip_if_file_exists
-    def get_file(self):
-        import pandas as pd
-
-        rows, table_url = self._get_table_rows()
-        self.update_source_metadata(rows_tuple=(rows, table_url))
-        if rows is None:
-            raise ValueError(
-                "Failed to retrieve rows from table "
-                f"{self.table_meta.base_id}/{self.table_meta.table_id}. Check logs",
-            )
-        # NOTE: Might be a good idea to add pagination for large tables
-        df = pd.DataFrame.from_dict(
-            [row["fields"] for row in rows],
-        ).sort_index(axis=1)
-
-        self.document = df.to_csv()
-        self.filename.parent.mkdir(parents=True, exist_ok=True)
-
-        with open(self.filename, "w", encoding="utf8") as f:
-            f.write(self.document)
-
-
-airtable_id_prefixes = ["app", "tbl", "viw"]
-
-
-def raise_airtable_path_error(piece):
-    if any(piece[:3] == prefix for prefix in airtable_id_prefixes):
-        raise (
-            ValueError(
-                "Path components are not correctly ordered.\
-                            Valid path structures: \
-                            - base_id/table_id/view_id , \
-                            - base_id/table_id, \
-                            - base_id .\
-                            It is also possible to leave --airtable-list-of-paths \
-                            argument empty (this will ingest everything).",
-            )
-        )
-    else:
-        raise (
-            ValueError(
-                """Path components are not valid Airtable ids.
-                        base_id should look like: appAbcDeF1ghijKlm,
-                        table_id should look like: tblAbcDeF1ghijKlm,
-                        view_id should look like:  viwAbcDeF1ghijKlm""",
-            )
-        )
-
-
-def check_path_validity(path):
-    pieces = path.split("/")
-    assert (
-        1 <= len(pieces) <= 3
-    ), "Path should be composed of between 1-3 \
-                                components (base_id, table_id, view_id)."
-
-    for i, piece in enumerate(pieces):
-        try:
-            assert piece[:3] == airtable_id_prefixes[i]
-        except AssertionError:
-            raise_airtable_path_error(piece)
-
-
-@dataclass
-class AirtableSourceConnector(SourceConnectorCleanupMixin, BaseSourceConnector):
-    """Fetches tables or views from an Airtable org."""
-
-    connector_config: SimpleAirtableConfig
-    _api: t.Optional["Api"] = field(init=False, default=None)
-
-    @property
-    def api(self):
-        if self._api is None:
-            self._api = Api(self.connector_config.access_config.personal_access_token)
-        return self._api
-
-    @api.setter
-    def api(self, api: "Api"):
-        self._api = api
-
-    def check_connection(self):
-        import requests
-
-        try:
-            self.api.request(method="HEAD", url=self.api.build_url("meta", "bases"))
-        except requests.HTTPError as http_error:
-            logger.error(f"failed to validate connection: {http_error}", exc_info=True)
-            raise SourceConnectionError(f"failed to validate connection: {http_error}")
-
-    @requires_dependencies(["pyairtable"], extras="airtable")
-    def initialize(self):
-        from pyairtable import Api
-
-        self.base_ids_to_fetch_tables_from = []
-        if self.connector_config.list_of_paths:
-            self.list_of_paths = self.connector_config.list_of_paths.split()
-
-        self.api = Api(self.connector_config.access_config.personal_access_token)
-
-    @requires_dependencies(["pyairtable"], extras="airtable")
-    def use_all_bases(self):
-        from pyairtable.metadata import get_api_bases
-
-        self.base_ids_to_fetch_tables_from = [
-            base["id"] for base in get_api_bases(self.api)["bases"]
-        ]
-
-    @requires_dependencies(["pyairtable"], extras="airtable")
-    def fetch_table_ids(self):
-        from pyairtable.metadata import get_base_schema
-
-        bases = [
-            (base_id, self.api.base(base_id)) for base_id in self.base_ids_to_fetch_tables_from
-        ]
-
-        metadata_for_each_base = [
-            (base_id, get_base_schema(base)["tables"]) for base_id, base in bases
-        ]
-
-        baseid_tableid_viewid_tuples = [
-            (base_id, table["id"], None)
-            for base_id, base_metadata in metadata_for_each_base
-            for table in base_metadata
-        ]
-
-        return baseid_tableid_viewid_tuples
-
-    def get_ingest_docs(self):
-        """Fetches documents in an Airtable org."""
-
-        # When no list of paths provided, the connector ingests everything.
-        if not self.connector_config.list_of_paths:
-            self.use_all_bases()
-            baseid_tableid_viewid_tuples = self.fetch_table_ids()
-
-        # When there is a list of paths, the connector checks the validity
-        # of the paths, and fetches table_ids to be ingested, based on the paths.
-        else:
-            self.paths = self.connector_config.list_of_paths.split()
-            self.paths = [path.strip("/") for path in self.paths]
-
-            [check_path_validity(path) for path in self.paths]
-
-            self.base_ids_to_fetch_tables_from = []
-            baseid_tableid_viewid_tuples = []
-
-            for path in self.paths:
-                components = path.split("/")
-                if len(components) == 1:  # only a base_id is provided
-                    self.base_ids_to_fetch_tables_from.append(components[0])
-                elif len(components) == 2:  # a base_id and a table_id are provided
-                    baseid_tableid_viewid_tuples.append((components[0], components[1], None))
-                elif len(components) == 3:  # a base_id, table_id, and a view_id are provided
-                    baseid_tableid_viewid_tuples.append(
-                        (components[0], components[1], components[2]),
-                    )
-
-            baseid_tableid_viewid_tuples += self.fetch_table_ids()
-        return [
-            AirtableIngestDoc(
-                processor_config=self.processor_config,
-                connector_config=self.connector_config,
-                read_config=self.read_config,
-                table_meta=AirtableTableMeta(base_id, table_id, view_id),
-            )
-            for base_id, table_id, view_id in baseid_tableid_viewid_tuples
-        ]