update block

api.saladeaula.digital/app/routes/orgs/subscription.py

@@ -31,6 +31,7 @@ def add(
     name: Annotated[str, Body(embed=True)],
     billing_day: Annotated[int, Body(embed=True, ge=1, le=31)],
     payment_method: Annotated[PaymentMethod, Body(embed=True)],
+    subscription_frozen: Annotated[bool, Body(embed=True)] = False,
 ):
     now_ = now()
 
@@ -71,6 +72,15 @@ def add(
             exc_cls=SubscriptionConflictError,
         )
 
+        if subscription_frozen:
+            transact.put(
+                item={
+                    'id': 'SUBSCRIPTION#FROZEN',
+                    'sk': f'ORG#{org_id}',
+                    'created_at': now_,
+                }
+            )
+
     return JSONResponse(status_code=HTTPStatus.CREATED)
 
 
@@ -100,19 +110,19 @@ def edit(
                 ':now': now_,
             },
             cond_expr='attribute_exists(sk)',
-            exc_cls=OrgNotFoundError,
+            exc_cls=SubscriptionRequiredError,
         )
 
         if subscription_frozen:
             transact.put(
                 item={
-                    'id': 'SUBSCRIPTION#FREEZE',
+                    'id': 'SUBSCRIPTION#FROZEN',
                     'sk': f'ORG#{org_id}',
                     'created_at': now_,
                 }
             )
         else:
-            transact.delete(key=KeyPair('SUBSCRIPTION#FREEZE', f'ORG#{org_id}'))
+            transact.delete(key=KeyPair('SUBSCRIPTION#FROZEN', f'ORG#{org_id}'))
 
     return JSONResponse(status_code=HTTPStatus.NO_CONTENT)
 
@@ -133,6 +143,6 @@ def remove(org_id: str):
         )
         transact.delete(key=KeyPair(org_id, 'METADATA#SUBSCRIPTION'))
         transact.delete(key=KeyPair('SUBSCRIPTION', f'ORG#{org_id}'))
-        transact.delete(key=KeyPair('SUBSCRIPTION#FREEZE', f'ORG#{org_id}'))
+        transact.delete(key=KeyPair('SUBSCRIPTION#FROZEN', f'ORG#{org_id}'))
 
     return JSONResponse(status_code=HTTPStatus.NO_CONTENT)