|
14 | 14 | "AsyncMyOffsetPageTopLevelArray", |
15 | 15 | "SyncOffsetPageClusters", |
16 | 16 | "AsyncOffsetPageClusters", |
| 17 | + "SyncOffsetPageQueryLogs", |
| 18 | + "AsyncOffsetPageQueryLogs", |
| 19 | + "SyncOffsetPageRemediations", |
| 20 | + "AsyncOffsetPageRemediations", |
17 | 21 | ] |
18 | 22 |
|
19 | 23 | _BaseModelT = TypeVar("_BaseModelT", bound=BaseModel) |
@@ -141,3 +145,123 @@ def next_page_info(self) -> Optional[PageInfo]: |
141 | 145 | return PageInfo(params={"offset": current_count}) |
142 | 146 |
|
143 | 147 | return None |
| 148 | + |
| 149 | + |
| 150 | +class SyncOffsetPageQueryLogs(BaseSyncPage[_T], BasePage[_T], Generic[_T]): |
| 151 | + query_logs: List[_T] |
| 152 | + total_count: Optional[int] = None |
| 153 | + |
| 154 | + @override |
| 155 | + def _get_page_items(self) -> List[_T]: |
| 156 | + query_logs = self.query_logs |
| 157 | + if not query_logs: |
| 158 | + return [] |
| 159 | + return query_logs |
| 160 | + |
| 161 | + @override |
| 162 | + def next_page_info(self) -> Optional[PageInfo]: |
| 163 | + offset = self._options.params.get("offset") or 0 |
| 164 | + if not isinstance(offset, int): |
| 165 | + raise ValueError(f'Expected "offset" param to be an integer but got {offset}') |
| 166 | + |
| 167 | + length = len(self._get_page_items()) |
| 168 | + current_count = offset + length |
| 169 | + |
| 170 | + total_count = self.total_count |
| 171 | + if total_count is None: |
| 172 | + return None |
| 173 | + |
| 174 | + if current_count < total_count: |
| 175 | + return PageInfo(params={"offset": current_count}) |
| 176 | + |
| 177 | + return None |
| 178 | + |
| 179 | + |
| 180 | +class AsyncOffsetPageQueryLogs(BaseAsyncPage[_T], BasePage[_T], Generic[_T]): |
| 181 | + query_logs: List[_T] |
| 182 | + total_count: Optional[int] = None |
| 183 | + |
| 184 | + @override |
| 185 | + def _get_page_items(self) -> List[_T]: |
| 186 | + query_logs = self.query_logs |
| 187 | + if not query_logs: |
| 188 | + return [] |
| 189 | + return query_logs |
| 190 | + |
| 191 | + @override |
| 192 | + def next_page_info(self) -> Optional[PageInfo]: |
| 193 | + offset = self._options.params.get("offset") or 0 |
| 194 | + if not isinstance(offset, int): |
| 195 | + raise ValueError(f'Expected "offset" param to be an integer but got {offset}') |
| 196 | + |
| 197 | + length = len(self._get_page_items()) |
| 198 | + current_count = offset + length |
| 199 | + |
| 200 | + total_count = self.total_count |
| 201 | + if total_count is None: |
| 202 | + return None |
| 203 | + |
| 204 | + if current_count < total_count: |
| 205 | + return PageInfo(params={"offset": current_count}) |
| 206 | + |
| 207 | + return None |
| 208 | + |
| 209 | + |
| 210 | +class SyncOffsetPageRemediations(BaseSyncPage[_T], BasePage[_T], Generic[_T]): |
| 211 | + remediations: List[_T] |
| 212 | + total_count: Optional[int] = None |
| 213 | + |
| 214 | + @override |
| 215 | + def _get_page_items(self) -> List[_T]: |
| 216 | + remediations = self.remediations |
| 217 | + if not remediations: |
| 218 | + return [] |
| 219 | + return remediations |
| 220 | + |
| 221 | + @override |
| 222 | + def next_page_info(self) -> Optional[PageInfo]: |
| 223 | + offset = self._options.params.get("offset") or 0 |
| 224 | + if not isinstance(offset, int): |
| 225 | + raise ValueError(f'Expected "offset" param to be an integer but got {offset}') |
| 226 | + |
| 227 | + length = len(self._get_page_items()) |
| 228 | + current_count = offset + length |
| 229 | + |
| 230 | + total_count = self.total_count |
| 231 | + if total_count is None: |
| 232 | + return None |
| 233 | + |
| 234 | + if current_count < total_count: |
| 235 | + return PageInfo(params={"offset": current_count}) |
| 236 | + |
| 237 | + return None |
| 238 | + |
| 239 | + |
| 240 | +class AsyncOffsetPageRemediations(BaseAsyncPage[_T], BasePage[_T], Generic[_T]): |
| 241 | + remediations: List[_T] |
| 242 | + total_count: Optional[int] = None |
| 243 | + |
| 244 | + @override |
| 245 | + def _get_page_items(self) -> List[_T]: |
| 246 | + remediations = self.remediations |
| 247 | + if not remediations: |
| 248 | + return [] |
| 249 | + return remediations |
| 250 | + |
| 251 | + @override |
| 252 | + def next_page_info(self) -> Optional[PageInfo]: |
| 253 | + offset = self._options.params.get("offset") or 0 |
| 254 | + if not isinstance(offset, int): |
| 255 | + raise ValueError(f'Expected "offset" param to be an integer but got {offset}') |
| 256 | + |
| 257 | + length = len(self._get_page_items()) |
| 258 | + current_count = offset + length |
| 259 | + |
| 260 | + total_count = self.total_count |
| 261 | + if total_count is None: |
| 262 | + return None |
| 263 | + |
| 264 | + if current_count < total_count: |
| 265 | + return PageInfo(params={"offset": current_count}) |
| 266 | + |
| 267 | + return None |
![stainless-app[bot]](https://avatars.githubusercontent.com/in/378072?v=4&size=40){kind=avatar}
0 commit comments