cookie api works, but need to select profile

getcookie.py

@@ -1,29 +1,42 @@
+from flask import Flask, jsonify, request
 import os
 import pyotp
+import time
 from urllib.parse import urlencode
 from bs4 import BeautifulSoup
 import requests
 from dotenv import load_dotenv
-import json
 
 class MIBLogin:
     def __init__(self):
-        self.session = requests.Session()
         self.base_url = "https://faisanet.mib.com.mv"
         self.headers = {
             "User-Agent": "Mozilla/5.0 (give-api) ismath-owes-me-real-api/give-real-api AppleWebKit/537.36 (KHTML, like Gecko) Safari/537.36",
             "Accept": "*/*"
         }
+        self.last_cookie = None
+        self.last_renewal_time = 0
+        self.create_new_session()
+
+    def create_new_session(self):
+        self.session = requests.Session()
         
     def get_login_page(self):
         response = self.session.get(f"{self.base_url}/auth", headers=self.headers)
         soup = BeautifulSoup(response.text, 'html.parser')
-        return soup.find('input', {'name': 'rTag'})['value']
+        rtag_input = soup.find('input', {'name': 'rTag'})
+        if not rtag_input:
+            raise ValueError("Failed to find rTag input field in login page.")
+        return rtag_input['value']
 
     def get_auth_type(self, rtag, username, retain=1):
         data = {'rTag': rtag, 'pgf01': username, 'retain': retain}
         headers = {**self.headers, 'Content-Type': 'application/x-www-form-urlencoded'}
-        response = self.session.post(f"{self.base_url}/aAuth/getAuthType", headers=headers, data=urlencode(data))
+        response = self.session.post(
+            f"{self.base_url}/aAuth/getAuthType",
+            headers=headers,
+            data=urlencode(data)
+        )
         return response.json()
 
     def login(self, username, password, retain=1):
@@ -32,7 +45,11 @@ class MIBLogin:
         
         data = {'rTag': rtag, 'pgf01': username, 'pgf02': password, 'retain': retain}
         headers = {**self.headers, 'Content-Type': 'application/x-www-form-urlencoded'}
-        self.session.post(f"{self.base_url}/aAuth", headers=headers, data=urlencode(data))
+        self.session.post(
+            f"{self.base_url}/aAuth",
+            headers=headers,
+            data=urlencode(data)
+        )
 
     def auth_2fa(self, totp_seed):
         self.session.get(f"{self.base_url}/auth2FA", headers=self.headers)
@@ -40,18 +57,70 @@ class MIBLogin:
         
         data = {'otpType': 3, 'otp': totp.now()}
         headers = {**self.headers, 'Content-Type': 'application/x-www-form-urlencoded'}
-        self.session.post(f"{self.base_url}/aAuth2FA/verifyOTP", headers=headers, data=urlencode(data))
+        self.session.post(
+            f"{self.base_url}/aAuth2FA/verifyOTP",
+            headers=headers,
+            data=urlencode(data)
+        )
         
-        return {cookie.name: cookie.value for cookie in self.session.cookies if cookie.name in ['IBSID', 'ql_0']}
+        cookies = {
+            cookie.name: cookie.value 
+            for cookie in self.session.cookies 
+            if cookie.name in ['IBSID', 'ql_0']
+        }
+        self.last_cookie = cookies
+        self.last_renewal_time = time.time()
+        return cookies
 
-def main():
+    def clear_cookie(self):
-    load_dotenv()
+        self.last_cookie = None
-    mib = MIBLogin()
+        self.create_new_session()
-    mib.login(os.getenv('USERNAME'), os.getenv('PASSWORD'))
-    cookies = mib.auth_2fa(os.getenv('TOTP_SEED'))
 
-    # Print the cookies in JSON format
+# Initialize Flask app
-    print(json.dumps(cookies, indent=4))
+app = Flask(__name__)
+mib = MIBLogin()
+load_dotenv()
 
-if __name__ == "__main__":
+def renew_cookie_internal():
-    main()
+    try:
+        print(f"Starting cookie renewal process...")
+        mib.clear_cookie()
+        mib.login(os.getenv('USERNAME'), os.getenv('PASSWORD'))
+        cookies = mib.auth_2fa(os.getenv('TOTP_SEED'))
+        print(f"Cookie renewal completed successfully")
+        return cookies
+    except Exception as e:
+        print(f"Error during cookie renewal: {str(e)}")
+        return {"error": str(e)}
+
+@app.route('/getcookie', methods=['GET'])
+def get_cookie():
+    client_ip = request.remote_addr
+    print(f"Providing cookie to: {client_ip}")
+    
+    if mib.last_cookie:
+        return jsonify(mib.last_cookie)
+    else:
+        cookies = renew_cookie_internal()
+        if "error" in cookies:
+            return jsonify(cookies), 500
+        return jsonify(cookies)
+
+@app.route('/newcookie', methods=['GET'])
+def new_cookie():
+    client_ip = request.remote_addr
+    current_time = time.time()
+    
+    # Check if less than 60 seconds since last renewal
+    if current_time - mib.last_renewal_time < 60:
+        print(f"Providing existing cookie to {client_ip} (last renewal was {int(current_time - mib.last_renewal_time)} seconds ago)")
+        return jsonify(mib.last_cookie)
+    
+    print(f"Renewing cookie as requested by: {client_ip}")
+    cookies = renew_cookie_internal()
+    if "error" in cookies:
+        return jsonify(cookies), 500
+    return jsonify(cookies)
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=5000)