Finish working on CLI Implementation

server/api.yml

@@ -564,6 +564,34 @@ paths:
           description: Invalid input
         '404':
           description: No accounts found
+  /Admin/Balance:
+    put:
+      tags:
+        - admin
+      summary: Modify account balance
+      description: Modify account balance
+      operationId: manager.modify_balance
+      parameters:
+        - name: account_id
+          in: query
+          description: ID of account to modify
+          required: true
+          schema:
+            type: string
+        - name: balance
+          in: query
+          description: Amount to modify balance by
+          required: true
+          schema:
+            type: number
+            format: float
+      responses:
+        '200':
+          description: Successful operation
+        '400':
+          description: Invalid input
+        '404':
+          description: Account not found
   /Admin/Transactions:
     get:
       tags:
@@ -625,7 +653,7 @@ paths:
             description: Invalid input
           '401':
             description: Unauthorised
-  /Admin/Balance:
+  /Admin/TestBalance:
     get:
       tags:
         - admin

server/manager.py

@@ -159,7 +159,6 @@ def generate_otp(client_id: str):
             send_email(email, "Luxbank One Time Password", f"Your one-time password is: {password}")
             otps[client_id] = (password, time.time())  # Store the OTP and the current time
             event_logger(f"OTP Code {password} emailed to {email}")
-            print(f"OTP Code {password} emailed to {email}")
             return format_response(True, "OTP generated and sent successfully."), 200
         except EmailSendingError as e:
             event_logger(f"Error sending email: {str(e)}")
@@ -477,7 +476,7 @@ def delete_transaction(transaction_id:int):
     return format_response(False, "Transaction not found."), 404
 
 @admin_required
-def modify_balance(transaction_id:int, amount:int):
+def modify_transaction_amount(transaction_id:str, amount:float):
     """Modifies the amount of a transaction in the database. If the transaction is not found, returns an error message."""
     for transaction in session.query(Transaction).all():
         if transaction.transaction_id == transaction_id:
@@ -487,6 +486,17 @@ def modify_balance(transaction_id:int, amount:int):
             return format_response(True, f"Transaction ID: {transaction_id} has been modified."), 200
     return format_response(False, "Transaction not found."), 404
 
+@admin_required
+def modify_balance(account_id:str, balance:float):
+    """Modifies the balance of an account in the database. If the account is not found, returns an error message."""
+    for account in session.query(Account).all():
+        if account.account_id == account_id:
+            account.balance = balance
+            session.commit()
+            event_logger(f"Account ID: {account_id} has been modified by {flask_session['client_id']}.")
+            return format_response(True, f"Account ID: {account_id} has been modified."), 200
+    return format_response(False, "Account not found."), 404
+
 @admin_required
 def test_account_balances():
     """Checks all account balances in the database and returns a list of discrepancies."""