Merge pull request #6 from taskbadger/sk/update-gen-code

update gen code + better error handling

Author: snopoke

taskbadger.yaml

@@ -1,6 +1,6 @@
 openapi: 3.0.3
 info:
-  title: Task Badger
+  title: Taskbadger API
   version: '0.1'
   description: Documentation for the Task Badger API
 paths:
@@ -675,6 +675,15 @@ components:
         status:
           allOf:
           - $ref: '#/components/schemas/StatusEnum'
+          description: |-
+            * `pending` - pending
+            * `pre_processing` - pre_processing
+            * `processing` - processing
+            * `post_processing` - post_processing
+            * `success` - success
+            * `error` - error
+            * `cancelled` - cancelled
+            * `stale` - stale
           default: pending
         value:
           type: integer
@@ -712,14 +721,14 @@ components:
           minimum: 1
           nullable: true
           description: Maximum duration the task can be running for before being considered
-            failed. (minutes)
+            failed. (seconds)
         stale_timeout:
           type: integer
           maximum: 2147483647
           minimum: 1
           nullable: true
           description: Maximum time to allow between task updates before considering
-            the task stale. Only applies when task is in a running state. (minutes)
+            the task stale. (seconds)
     StatusEnum:
       enum:
       - pending
@@ -751,6 +760,15 @@ components:
         status:
           allOf:
           - $ref: '#/components/schemas/StatusEnum'
+          description: |-
+            * `pending` - pending
+            * `pre_processing` - pre_processing
+            * `processing` - processing
+            * `post_processing` - post_processing
+            * `success` - success
+            * `error` - error
+            * `cancelled` - cancelled
+            * `stale` - stale
           default: pending
         value:
           type: integer
@@ -800,14 +818,14 @@ components:
           minimum: 1
           nullable: true
           description: Maximum duration the task can be running for before being considered
-            failed. (minutes)
+            failed. (seconds)
         stale_timeout:
           type: integer
           maximum: 2147483647
           minimum: 1
           nullable: true
           description: Maximum time to allow between task updates before considering
-            the task stale. Only applies when task is in a running state. (minutes)
+            the task stale. (seconds)
         url:
           type: string
           readOnly: true
@@ -835,6 +853,15 @@ components:
         status:
           allOf:
           - $ref: '#/components/schemas/StatusEnum'
+          description: |-
+            * `pending` - pending
+            * `pre_processing` - pre_processing
+            * `processing` - processing
+            * `post_processing` - post_processing
+            * `success` - success
+            * `error` - error
+            * `cancelled` - cancelled
+            * `stale` - stale
           default: pending
         value:
           type: integer
@@ -872,14 +899,14 @@ components:
           minimum: 1
           nullable: true
           description: Maximum duration the task can be running for before being considered
-            failed. (minutes)
+            failed. (seconds)
         stale_timeout:
           type: integer
           maximum: 2147483647
           minimum: 1
           nullable: true
           description: Maximum time to allow between task updates before considering
-            the task stale. Only applies when task is in a running state. (minutes)
+            the task stale. (seconds)
       required:
       - name
   securitySchemes:

taskbadger/errors.py

@@ -0,0 +1,18 @@
+class TaskbadgerException(Exception):
+    pass
+
+
+class Unauthorized(TaskbadgerException):
+    pass
+
+
+class UnexpectedStatus(TaskbadgerException):
+    def __init__(self, status_code: int, content: bytes):
+        self.status_code = status_code
+        self.content = content
+
+        super().__init__(f"Unexpected status code: {status_code}")
+
+
+class ServerError(UnexpectedStatus):
+    pass

taskbadger/integrations.py

@@ -1,5 +1,4 @@
 import dataclasses
-from enum import Enum
 from typing import Any, Dict
 
 from taskbadger.exceptions import TaskbadgerException

taskbadger/internal/__init__.py

@@ -1,4 +1,4 @@
-""" A client library for accessing Task Badger """
+""" A client library for accessing Taskbadger API """
 from .client import AuthenticatedClient, Client
 
 __all__ = (

taskbadger/internal/api/action_endpoints/action_cancel.py

@@ -29,14 +29,15 @@ def _get_kwargs(
         "headers": headers,
         "cookies": cookies,
         "timeout": client.get_timeout(),
+        "follow_redirects": client.follow_redirects,
     }
 
 
 def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Any]:
     if response.status_code == HTTPStatus.NO_CONTENT:
         return None
     if client.raise_on_unexpected_status:
-        raise errors.UnexpectedStatus(f"Unexpected status code: {response.status_code}")
+        raise errors.UnexpectedStatus(response.status_code, response.content)
     else:
         return None
 

taskbadger/internal/api/action_endpoints/action_create.py

@@ -33,6 +33,7 @@ def _get_kwargs(
         "headers": headers,
         "cookies": cookies,
         "timeout": client.get_timeout(),
+        "follow_redirects": client.follow_redirects,
         "json": json_json_body,
     }
 
@@ -43,7 +44,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Act
 
         return response_201
     if client.raise_on_unexpected_status:
-        raise errors.UnexpectedStatus(f"Unexpected status code: {response.status_code}")
+        raise errors.UnexpectedStatus(response.status_code, response.content)
     else:
         return None
 
@@ -122,7 +123,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Action]
+        Action
     """
 
     return sync_detailed(
@@ -197,7 +198,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Action]
+        Action
     """
 
     return (

taskbadger/internal/api/action_endpoints/action_get.py

@@ -30,6 +30,7 @@ def _get_kwargs(
         "headers": headers,
         "cookies": cookies,
         "timeout": client.get_timeout(),
+        "follow_redirects": client.follow_redirects,
     }
 
 
@@ -39,7 +40,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Act
 
         return response_200
     if client.raise_on_unexpected_status:
-        raise errors.UnexpectedStatus(f"Unexpected status code: {response.status_code}")
+        raise errors.UnexpectedStatus(response.status_code, response.content)
     else:
         return None
 
@@ -118,7 +119,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Action]
+        Action
     """
 
     return sync_detailed(
@@ -193,7 +194,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Action]
+        Action
     """
 
     return (

taskbadger/internal/api/action_endpoints/action_list.py

@@ -29,6 +29,7 @@ def _get_kwargs(
         "headers": headers,
         "cookies": cookies,
         "timeout": client.get_timeout(),
+        "follow_redirects": client.follow_redirects,
     }
 
 
@@ -43,7 +44,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Lis
 
         return response_200
     if client.raise_on_unexpected_status:
-        raise errors.UnexpectedStatus(f"Unexpected status code: {response.status_code}")
+        raise errors.UnexpectedStatus(response.status_code, response.content)
     else:
         return None
 
@@ -117,7 +118,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['Action']]
+        List['Action']
     """
 
     return sync_detailed(
@@ -186,7 +187,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[List['Action']]
+        List['Action']
     """
 
     return (

taskbadger/internal/api/action_endpoints/action_partial_update.py

@@ -34,6 +34,7 @@ def _get_kwargs(
         "headers": headers,
         "cookies": cookies,
         "timeout": client.get_timeout(),
+        "follow_redirects": client.follow_redirects,
         "json": json_json_body,
     }
 
@@ -44,7 +45,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Act
 
         return response_200
     if client.raise_on_unexpected_status:
-        raise errors.UnexpectedStatus(f"Unexpected status code: {response.status_code}")
+        raise errors.UnexpectedStatus(response.status_code, response.content)
     else:
         return None
 
@@ -128,7 +129,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Action]
+        Action
     """
 
     return sync_detailed(
@@ -209,7 +210,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Action]
+        Action
     """
 
     return (

taskbadger/internal/api/action_endpoints/action_update.py

@@ -34,6 +34,7 @@ def _get_kwargs(
         "headers": headers,
         "cookies": cookies,
         "timeout": client.get_timeout(),
+        "follow_redirects": client.follow_redirects,
         "json": json_json_body,
     }
 
@@ -44,7 +45,7 @@ def _parse_response(*, client: Client, response: httpx.Response) -> Optional[Act
 
         return response_200
     if client.raise_on_unexpected_status:
-        raise errors.UnexpectedStatus(f"Unexpected status code: {response.status_code}")
+        raise errors.UnexpectedStatus(response.status_code, response.content)
     else:
         return None
 
@@ -128,7 +129,7 @@ def sync(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Action]
+        Action
     """
 
     return sync_detailed(
@@ -209,7 +210,7 @@ async def asyncio(
         httpx.TimeoutException: If the request takes longer than Client.timeout.
 
     Returns:
-        Response[Action]
+        Action
     """
 
     return (