Changeset View
Changeset View
Standalone View
Standalone View
examples/hacker_news/hacker_news/resources/hn_resource.py
| Show First 20 Lines • Show All 50 Lines • ▼ Show 20 Lines | class HNSnapshotClient(HNClient): | ||||
| def fetch_max_item_id(self) -> int: | def fetch_max_item_id(self) -> int: | ||||
| return int(list(self._items.keys())[-1]) | return int(list(self._items.keys())[-1]) | ||||
| def min_item_id(self) -> int: | def min_item_id(self) -> int: | ||||
| return int(list(self._items.keys())[0]) | return int(list(self._items.keys())[0]) | ||||
| class HNAPISubsampleClient(HNClient): | |||||
| """ | |||||
| This client gets real data from the web API, but is much faster than the normal implementation, | |||||
| which is useful for testing / demoing purposes. | |||||
| """ | |||||
| def __init__(self, subsample_rate): | |||||
| self._items = {} | |||||
| self.subsample_rate = subsample_rate | |||||
| def fetch_item_by_id(self, item_id: int) -> Optional[HNItemRecord]: | |||||
| # map self.subsample_rate items to the same item_id, caching it for faster performance | |||||
| subsample_id = item_id - item_id % self.subsample_rate | |||||
| if subsample_id not in self._items: | |||||
| item_url = f"{HN_BASE_URL}/item/{subsample_id}.json" | |||||
| item = requests.get(item_url, timeout=5).json() | |||||
| self._items[subsample_id] = item | |||||
| return self._items[subsample_id] | |||||
| def fetch_max_item_id(self) -> int: | |||||
| return requests.get(f"{HN_BASE_URL}/maxitem.json", timeout=5).json() | |||||
| def min_item_id(self) -> int: | |||||
| return 1 | |||||
| @resource(description="A hackernews client that fetches results from the firebaseio web api.") | @resource(description="A hackernews client that fetches results from the firebaseio web api.") | ||||
| def hn_api_client(_): | def hn_api_client(_): | ||||
| return HNAPIClient() | return HNAPIClient() | ||||
| @resource( | @resource( | ||||
| description="A mock hackernews client powered by a json file containing sample responses." | description="A mock hackernews client powered by a json file containing sample responses." | ||||
| ) | ) | ||||
| def hn_snapshot_client(_): | def hn_snapshot_client(_): | ||||
| return HNSnapshotClient() | return HNSnapshotClient() | ||||
| @resource( | |||||
| config_schema={"sample_rate": int}, | |||||
| description="A hackernews client that sub-samples results from the firebaseio web api.", | |||||
| ) | |||||
| def hn_api_subsample_client(context): | |||||
| return HNAPISubsampleClient(context.resource_config["sample_rate"]) | |||||