from flask import Flask, jsonify, request from flask_sqlalchemy import SQLAlchemy from datetime import datetime app = Flask(__name__) app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///tokens.db' # Use SQLite for simplicity app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False db = SQLAlchemy(app) class Token(db.Model): id = db.Column(db.Integer, primary_key=True) access_token = db.Column(db.String(255), nullable=False) refresh_token = db.Column(db.String(255), nullable=False) expiration_seconds = db.Column(db.Integer, nullable=False) timestamp = db.Column(db.DateTime, default=datetime.utcnow) def __repr__(self): return f'' # Create an application context with app.app_context(): db.create_all() # API to create a new token entry @app.route('/token', methods=['POST']) def create_token(): data = request.get_json() access_token = data.get('access_token') refresh_token = data.get('refresh_token') expiration_seconds = data.get('expiration_seconds') if not access_token or not refresh_token or not expiration_seconds: return 'Missing required fields', 400 new_token = Token(access_token=access_token, refresh_token=refresh_token, expiration_seconds=expiration_seconds) db.session.add(new_token) db.session.commit() return ('', 204) # API to get the latest token entry @app.route('/token/latest', methods=['GET']) def get_latest_token(): latest_token = Token.query.order_by(Token.timestamp.desc()).first() if latest_token: token_info = { 'access_token': latest_token.access_token, 'refresh_token': latest_token.refresh_token, 'expiration_seconds': latest_token.expiration_seconds, 'timestamp': int(latest_token.timestamp.timestamp()) } return jsonify({'token': token_info}), 200 else: return '', 204 if __name__ == '__main__': app.run(host='0.0.0.0', port=5000, debug=True)