omniload 0.0.0__py3-none-any.whl

omniload/src/hostaway/client.py

@@ -0,0 +1,288 @@
+from typing import Callable, Iterable, Optional
+
+import pendulum
+from dlt.sources.helpers.requests import Client
+
+
+class HostawayClient:
+    BASE_URL = "https://api.hostaway.com"
+
+    def __init__(self, api_key: str) -> None:
+        self.session = Client(raise_for_status=False).session
+        self.session.headers.update({"Authorization": f"Bearer {api_key}"})
+
+    def _fetch_single(self, url: str, params: Optional[dict] = None) -> Iterable[dict]:
+        response = self.session.get(url, params=params, timeout=30)
+        response.raise_for_status()
+        response_data = response.json()
+
+        if isinstance(response_data, dict) and "result" in response_data:
+            items = response_data["result"]
+        elif isinstance(response_data, list):
+            items = response_data
+        else:
+            items = []
+
+        if isinstance(items, list):
+            for item in items:
+                yield item
+        elif isinstance(items, dict):
+            yield items
+
+    def _paginate(
+        self,
+        url: str,
+        params: Optional[dict] = None,
+        limit: int = 100,
+        process_item: Optional[Callable[[dict], dict]] = None,
+    ) -> Iterable[dict]:
+        offset = 0
+        if params is None:
+            params = {}
+
+        while True:
+            page_params = {**params, "limit": limit, "offset": offset}
+            response = self.session.get(url, params=page_params, timeout=30)
+            response.raise_for_status()
+            response_data = response.json()
+
+            if isinstance(response_data, dict) and "result" in response_data:
+                items = response_data["result"]
+            elif isinstance(response_data, list):
+                items = response_data
+            else:
+                items = []
+
+            if not items or (isinstance(items, list) and len(items) == 0):
+                break
+
+            if isinstance(items, list):
+                for item in items:
+                    if process_item:
+                        item = process_item(item)
+                    yield item
+            elif isinstance(items, dict):
+                if process_item:
+                    items = process_item(items)
+                yield items
+
+            if isinstance(items, list) and len(items) < limit:
+                break
+            elif isinstance(items, dict):
+                break
+
+            offset += limit
+
+    def fetch_listings(
+        self,
+        start_time: pendulum.DateTime,
+        end_time: pendulum.DateTime,
+    ) -> Iterable[dict]:
+        def process_listing(listing: dict) -> dict:
+            if "latestActivityOn" in listing and listing["latestActivityOn"]:
+                try:
+                    listing["latestActivityOn"] = pendulum.parse(
+                        listing["latestActivityOn"]
+                    )
+                except Exception:
+                    listing["latestActivityOn"] = pendulum.datetime(
+                        1970, 1, 1, tz="UTC"
+                    )
+            else:
+                listing["latestActivityOn"] = pendulum.datetime(1970, 1, 1, tz="UTC")
+            return listing
+
+        url = f"{self.BASE_URL}/v1/listings"
+        for listing in self._paginate(url, process_item=process_listing):
+            if start_time <= listing["latestActivityOn"] <= end_time:
+                yield listing
+
+    def fetch_listing_fee_settings(
+        self,
+        listing_id,
+        start_time: pendulum.DateTime,
+        end_time: pendulum.DateTime,
+    ) -> Iterable[dict]:
+        def process_fee(fee: dict) -> dict:
+            if "updatedOn" in fee and fee["updatedOn"]:
+                try:
+                    fee["updatedOn"] = pendulum.parse(fee["updatedOn"])
+                except Exception:
+                    fee["updatedOn"] = pendulum.datetime(1970, 1, 1, tz="UTC")
+            else:
+                fee["updatedOn"] = pendulum.datetime(1970, 1, 1, tz="UTC")
+            return fee
+
+        url = f"{self.BASE_URL}/v1/listingFeeSettings/{str(listing_id)}"
+        for fee in self._paginate(url, process_item=process_fee):
+            if start_time <= fee["updatedOn"] <= end_time:
+                yield fee
+
+    def fetch_all_listing_fee_settings(
+        self,
+        start_time: pendulum.DateTime,
+        end_time: pendulum.DateTime,
+    ) -> Iterable[dict]:
+        for listing in self.fetch_listings(start_time, end_time):
+            listing_id = listing.get("id")
+            if listing_id:
+                try:
+                    yield from self.fetch_listing_fee_settings(
+                        listing_id, start_time, end_time
+                    )
+                except Exception:
+                    continue
+
+    def fetch_listing_agreement(
+        self,
+        listing_id,
+    ) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/listingAgreement/{str(listing_id)}"
+        yield from self._paginate(url)
+
+    def fetch_listing_pricing_settings(
+        self,
+        listing_id,
+    ) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/listing/pricingSettings/{str(listing_id)}"
+        yield from self._paginate(url)
+
+    def fetch_all_listing_pricing_settings(
+        self,
+        start_time: pendulum.DateTime,
+        end_time: pendulum.DateTime,
+    ) -> Iterable[dict]:
+        for listing in self.fetch_listings(start_time, end_time):
+            listing_id = listing.get("id")
+            if listing_id:
+                try:
+                    yield from self.fetch_listing_pricing_settings(listing_id)
+                except Exception:
+                    continue
+
+    def fetch_all_listing_agreements(
+        self,
+        start_time: pendulum.DateTime,
+        end_time: pendulum.DateTime,
+    ) -> Iterable[dict]:
+        for listing in self.fetch_listings(start_time, end_time):
+            listing_id = listing.get("id")
+            if listing_id:
+                try:
+                    yield from self.fetch_listing_agreement(listing_id)
+                except Exception:
+                    continue
+
+    def fetch_cancellation_policies(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/cancellationPolicies"
+        yield from self._fetch_single(url)
+
+    def fetch_cancellation_policies_airbnb(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/cancellationPolicies/airbnb"
+        yield from self._fetch_single(url)
+
+    def fetch_cancellation_policies_marriott(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/cancellationPolicies/marriott"
+        yield from self._fetch_single(url)
+
+    def fetch_cancellation_policies_vrbo(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/cancellationPolicies/vrbo"
+        yield from self._fetch_single(url)
+
+    def fetch_reservations(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/reservations"
+        yield from self._paginate(url)
+
+    def fetch_finance_field(self, reservation_id) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/financeField/{str(reservation_id)}"
+        yield from self._fetch_single(url)
+
+    def fetch_all_finance_fields(self) -> Iterable[dict]:
+        for reservation in self.fetch_reservations():
+            reservation_id = reservation.get("id")
+            if reservation_id:
+                try:
+                    yield from self.fetch_finance_field(reservation_id)
+                except Exception:
+                    continue
+
+    def fetch_reservation_payment_methods(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/reservations/paymentMethods"
+        yield from self._fetch_single(url)
+
+    def fetch_reservation_rental_agreement(self, reservation_id) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/reservations/{str(reservation_id)}/rentalAgreement"
+        try:
+            yield from self._fetch_single(url)
+        except Exception:
+            return
+
+    def fetch_all_reservation_rental_agreements(self) -> Iterable[dict]:
+        for reservation in self.fetch_reservations():
+            reservation_id = reservation.get("id")
+            if reservation_id:
+                try:
+                    yield from self.fetch_reservation_rental_agreement(reservation_id)
+                except Exception:
+                    continue
+
+    def fetch_listing_calendar(self, listing_id) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/listings/{str(listing_id)}/calendar"
+        yield from self._fetch_single(url)
+
+    def fetch_all_listing_calendars(
+        self,
+        start_time: pendulum.DateTime,
+        end_time: pendulum.DateTime,
+    ) -> Iterable[dict]:
+        for listing in self.fetch_listings(start_time, end_time):
+            listing_id = listing.get("id")
+            if listing_id:
+                try:
+                    yield from self.fetch_listing_calendar(listing_id)
+                except Exception:
+                    continue
+
+    def fetch_conversations(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/conversations"
+        yield from self._paginate(url)
+
+    def fetch_message_templates(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/messageTemplates"
+        yield from self._fetch_single(url)
+
+    def fetch_bed_types(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/bedTypes"
+        yield from self._fetch_single(url)
+
+    def fetch_property_types(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/propertyTypes"
+        yield from self._fetch_single(url)
+
+    def fetch_countries(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/countries"
+        yield from self._fetch_single(url)
+
+    def fetch_account_tax_settings(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/accountTaxSettings"
+        yield from self._fetch_single(url)
+
+    def fetch_user_groups(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/userGroups"
+        yield from self._fetch_single(url)
+
+    def fetch_guest_payment_charges(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/guestPayments/charges"
+        yield from self._paginate(url)
+
+    def fetch_coupons(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/coupons"
+        yield from self._fetch_single(url)
+
+    def fetch_webhook_reservations(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/webhooks/reservations"
+        yield from self._fetch_single(url)
+
+    def fetch_tasks(self) -> Iterable[dict]:
+        url = f"{self.BASE_URL}/v1/tasks"
+        yield from self._fetch_single(url)

omniload/src/http/__init__.py

@@ -0,0 +1,38 @@
+"""HTTP source for reading CSV, JSON, and Parquet files from public URLs"""
+
+from typing import Any, Optional
+
+import dlt
+from dlt.sources import DltResource
+
+from .readers import HttpReader
+
+
+@dlt.source
+def http_source(
+    url: str,
+    file_format: Optional[str] = None,
+    column_names: Optional[list[str]] = None,
+    **kwargs: Any,
+) -> DltResource:
+    """Source for reading files from HTTP URLs.
+
+    Supports CSV, JSON, Parquet, and CSV without headers file formats.
+
+    Args:
+        url (str): The HTTP(S) URL to the file
+        file_format (str, optional): File format ('csv', 'csv_headless', 'json', 'parquet').
+            If not provided, will be inferred from URL extension.
+        column_names (list[str], optional): Column names for csv_headless format.
+            If not provided for csv_headless, columns will be named unknown_col_0, unknown_col_1, etc.
+        **kwargs: Additional arguments passed to the reader functions
+
+    Returns:
+        DltResource: A dlt resource that yields the file data
+    """
+    reader = HttpReader(url, file_format, column_names)
+
+    return dlt.resource(
+        reader.read_file(**kwargs),
+        name="http_data",
+    )

omniload/src/http/readers.py

@@ -0,0 +1,146 @@
+"""Readers for HTTP file sources"""
+
+import io
+from typing import Any, Iterator, Optional
+from urllib.parse import urlparse
+
+import requests
+from dlt.sources import TDataItems
+
+
+class HttpReader:
+    """Reader for HTTP-based file sources"""
+
+    def __init__(
+        self,
+        url: str,
+        file_format: Optional[str] = None,
+        column_names: Optional[list[str]] = None,
+    ):
+        self.url = url
+        self.file_format = file_format or self._infer_format(url)
+        self.column_names = column_names
+
+        if self.file_format not in ["csv", "csv_headless", "json", "parquet"]:
+            raise ValueError(
+                f"Unsupported file format: {self.file_format}. "
+                "Supported formats: csv, csv_headless, json, parquet"
+            )
+
+    def _infer_format(self, url: str) -> str:
+        """Infer file format from URL extension"""
+        parsed = urlparse(url)
+        path = parsed.path.lower()
+
+        if path.endswith(".csv"):
+            return "csv"
+        elif path.endswith(".json") or path.endswith(".jsonl"):
+            return "json"
+        elif path.endswith(".parquet"):
+            return "parquet"
+        else:
+            raise ValueError(
+                f"Cannot infer file format from URL: {url}. "
+                "Please specify file_format parameter."
+            )
+
+    def _download_file(self) -> bytes:
+        """Download file from URL"""
+        response = requests.get(self.url, stream=True, timeout=30)
+        response.raise_for_status()
+        return response.content
+
+    def read_file(self, **kwargs: Any) -> Iterator[TDataItems]:
+        """Read file and yield data in chunks"""
+        content = self._download_file()
+
+        if self.file_format == "csv":
+            yield from self._read_csv(content, **kwargs)
+        elif self.file_format == "csv_headless":
+            yield from self._read_csv_headless(content, **kwargs)
+        elif self.file_format == "json":
+            yield from self._read_json(content, **kwargs)
+        elif self.file_format == "parquet":
+            yield from self._read_parquet(content, **kwargs)
+
+    def _read_csv(
+        self, content: bytes, chunksize: int = 10000, **pandas_kwargs: Any
+    ) -> Iterator[TDataItems]:
+        """Read CSV file with Pandas chunk by chunk"""
+        import pandas as pd  # type: ignore
+
+        kwargs = {**{"header": "infer", "chunksize": chunksize}, **pandas_kwargs}
+
+        file_obj = io.BytesIO(content)
+        for df in pd.read_csv(file_obj, **kwargs):
+            yield df.to_dict(orient="records")
+
+    def _read_csv_headless(
+        self, content: bytes, chunksize: int = 10000, **pandas_kwargs: Any
+    ) -> Iterator[TDataItems]:
+        """Read CSV file without headers, using provided column names or generating them"""
+        import pandas as pd  # type: ignore
+
+        # Determine column names
+        if self.column_names:
+            names = self.column_names
+        else:
+            # Use pandas to count columns reliably (handles quoted commas)
+            first_row = pd.read_csv(io.BytesIO(content), header=None, nrows=1)
+            num_columns = len(first_row.columns)
+            names = [f"unknown_col_{i}" for i in range(num_columns)]
+
+        kwargs = {
+            **{"header": None, "names": names, "chunksize": chunksize},
+            **pandas_kwargs,
+        }
+
+        file_obj = io.BytesIO(content)
+        for df in pd.read_csv(file_obj, **kwargs):
+            yield df.to_dict(orient="records")
+
+    def _read_json(
+        self, content: bytes, chunksize: int = 1000, **kwargs: Any
+    ) -> Iterator[TDataItems]:
+        """Read JSON or JSONL file"""
+        from dlt.common import json
+
+        file_obj = io.BytesIO(content)
+        text = file_obj.read().decode("utf-8")
+
+        # Try to detect if it's JSONL format (one JSON object per line)
+        lines = text.strip().split("\n")
+
+        if len(lines) > 1:
+            # Likely JSONL format
+            lines_chunk = []
+            for line in lines:
+                if line.strip():
+                    lines_chunk.append(json.loads(line))
+                    if len(lines_chunk) >= chunksize:
+                        yield lines_chunk
+                        lines_chunk = []
+            if lines_chunk:
+                yield lines_chunk
+        else:
+            # Single JSON object or array
+            data = json.loads(text)
+            if isinstance(data, list):
+                # Chunk the list
+                for i in range(0, len(data), chunksize):
+                    yield data[i : i + chunksize]
+            else:
+                # Single object
+                yield [data]
+
+    def _read_parquet(
+        self, content: bytes, chunksize: int = 10000, **kwargs: Any
+    ) -> Iterator[TDataItems]:
+        """Read Parquet file"""
+        from pyarrow import parquet as pq  # type: ignore
+
+        file_obj = io.BytesIO(content)
+        parquet_file = pq.ParquetFile(file_obj)
+
+        for batch in parquet_file.iter_batches(batch_size=chunksize):
+            yield batch.to_pylist()

omniload/src/http_client.py

@@ -0,0 +1,24 @@
+import requests
+from dlt.sources.helpers.requests import Client
+
+
+def create_client(retry_status_codes: list[int] | None = None) -> requests.Session:
+    if retry_status_codes is None:
+        retry_status_codes = [502]
+    return Client(
+        raise_for_status=False,
+        retry_condition=retry_on_status_code(retry_status_codes),
+        request_max_attempts=12,
+        request_backoff_factor=10,
+    ).session
+
+
+def retry_on_status_code(retry_status_codes: list[int]):
+    def retry_on_limit(
+        response: requests.Response | None, exception: BaseException | None
+    ) -> bool:
+        if response is None:
+            return False
+        return response.status_code in retry_status_codes
+
+    return retry_on_limit