diff --git a/ktvmanager/lib/checker.py b/ktvmanager/lib/checker.py index 02b7301..4c2daab 100644 --- a/ktvmanager/lib/checker.py +++ b/ktvmanager/lib/checker.py @@ -138,6 +138,16 @@ def validate_account() -> Tuple[Response, int]: 200, ) + # Check if account is expired + exp_date_str = result["data"]["user_info"]["exp_date"] + if exp_date_str: + from datetime import datetime, timezone + exp_date = datetime.fromtimestamp(int(exp_date_str), tz=timezone.utc) + current_date = datetime.now(timezone.utc) + + if current_date > exp_date: + return jsonify({"message": "Account is expired", "data": result}), 401 + return jsonify({"message": "Account is valid", "data": result}), 200 else: return jsonify({"message": "Account is invalid"}), 401