update block

api.saladeaula.digital/app/routes/enrollments/enroll.py

@@ -171,7 +171,7 @@ def enroll_now(enrollment: Enrollment, context: Context):
         )
         transact.condition(
             key=KeyPair(
-                pk='SUBSCRIPTION#FREEZE',
+                pk='SUBSCRIPTION#FROZEN',
                 sk=f'ORG#{org.id}',
             ),
             cond_expr='attribute_not_exists(sk)',
@@ -288,7 +288,7 @@ def enroll_later(enrollment: Enrollment, context: Context):
         )
         transact.condition(
             key=KeyPair(
-                pk='SUBSCRIPTION#FREEZE',
+                pk='SUBSCRIPTION#FROZEN',
                 sk=f'ORG#{org.id}',
             ),
             cond_expr='attribute_not_exists(sk)',

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

@@ -45,8 +45,8 @@ def get_org(org_id: str):
         + SortKey('METADATA#ADDRESS', rename_key='address')
         + SortKey('METADATA#SUBSCRIPTION', rename_key='subscription')
         + KeyPair(
-            pk='SUBSCRIPTION#FREEZE',
+            pk='SUBSCRIPTION#FROZEN',
             sk=SortKey(f'ORG#{org_id}'),
-            rename_key='subscription_freeze',
+            rename_key='subscription_frozen',
         )
     )

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)