Added database
This commit is contained in:
parent
ec1f855261
commit
6609f8a054
GPG Key ID:
81C942771BB69898
200
database/main.py
Normal file
200
database/main.py
Normal file
|
|
@ -0,0 +1,200 @@
|
|||
from flask import Flask, jsonify, request
|
||||
from flask_sqlalchemy import SQLAlchemy
|
||||
from datetime import datetime
|
||||
import uuid
|
||||
|
||||
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.String(36), primary_key=True, default=str(uuid.uuid4()))
|
||||
email = db.Column(db.String(255), nullable=False)
|
||||
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)
|
||||
readwise_api_key = db.Column(db.String(255))
|
||||
active = db.Column(db.Boolean, default=True)
|
||||
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
||||
updated_at = db.Column(db.DateTime, default=datetime.utcnow)
|
||||
|
||||
def __repr__(self):
|
||||
return f'<Token {self.id}>'
|
||||
|
||||
# This table stores email-wise last annotation timestamp
|
||||
# only one entry per email
|
||||
class AnnotationLastUpdate(db.Model):
|
||||
id = db.Column(db.String(36), primary_key=True, default=str(uuid.uuid4()))
|
||||
email = db.Column(db.String(255), nullable=False)
|
||||
last_update_time = db.Column(db.DateTime, nullable=False)
|
||||
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
||||
updated_at = db.Column(db.DateTime, default=datetime.utcnow)
|
||||
|
||||
def __repr__(self):
|
||||
return f'<AnnotationLastUpdate {self.id}>'
|
||||
|
||||
# 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()
|
||||
email = data.get('email')
|
||||
access_token = data.get('access_token')
|
||||
refresh_token = data.get('refresh_token')
|
||||
expiration_seconds = data.get('expiration_seconds')
|
||||
readwise_api_key = data.get('readwise_api_key')
|
||||
|
||||
required_fields = ['email', 'access_token', 'refresh_token', 'expiration_seconds']
|
||||
missing_fields = [field for field in required_fields if not data.get(field)]
|
||||
|
||||
if missing_fields:
|
||||
return jsonify({'error': f'Missing required fields: {", ".join(missing_fields)}'}), 400
|
||||
|
||||
# unique email when active is true
|
||||
existing_token = Token.query.filter_by(email=email, active=True).first()
|
||||
if existing_token:
|
||||
return jsonify({'error': 'An active token with this email already exists'}), 400
|
||||
|
||||
new_token = Token(
|
||||
email=email,
|
||||
access_token=access_token,
|
||||
refresh_token=refresh_token,
|
||||
expiration_seconds=expiration_seconds,
|
||||
readwise_api_key=readwise_api_key
|
||||
)
|
||||
db.session.add(new_token)
|
||||
db.session.commit()
|
||||
|
||||
return jsonify({'id': new_token.id}), 201
|
||||
|
||||
# API to get the token based on the id
|
||||
@app.route('/token/<id>', methods=['GET'])
|
||||
def get_token_by_id(id):
|
||||
token = Token.query.get(id)
|
||||
if not token:
|
||||
return jsonify({'error': 'Token not found'}), 404
|
||||
token_info = {
|
||||
'id': token.id,
|
||||
'email': token.email,
|
||||
'access_token': token.access_token,
|
||||
'refresh_token': token.refresh_token,
|
||||
'expiration_seconds': int(token.expiration_seconds),
|
||||
'readwise_api_key': token.readwise_api_key,
|
||||
'active': token.active,
|
||||
'created_at': int(token.created_at.timestamp()),
|
||||
'updated_at': int(token.updated_at.timestamp())
|
||||
}
|
||||
return jsonify({'token': token_info}), 200
|
||||
|
||||
# API to get the token based on the email
|
||||
@app.route('/token', methods=['GET'])
|
||||
def get_token_by_email():
|
||||
email = request.args.get('email')
|
||||
if not email:
|
||||
return jsonify({'error': 'Missing email query parameter'}), 400
|
||||
token = Token.query.filter_by(email=email, active=True).first()
|
||||
if not token:
|
||||
return '', 204
|
||||
token_info = {
|
||||
'id': token.id,
|
||||
'email': token.email,
|
||||
'access_token': token.access_token,
|
||||
'refresh_token': token.refresh_token,
|
||||
'expiration_seconds': int(token.expiration_seconds),
|
||||
'readwise_api_key': token.readwise_api_key,
|
||||
'active': token.active,
|
||||
'created_at': int(token.created_at.timestamp()),
|
||||
'updated_at': int(token.updated_at.timestamp())
|
||||
}
|
||||
return jsonify({'token': token_info}), 200
|
||||
|
||||
# API to update the token based on the id
|
||||
@app.route('/token/<id>', methods=['PUT'])
|
||||
def update_token_by_id(id):
|
||||
token = Token.query.get_or_404(id)
|
||||
data = request.get_json()
|
||||
token.access_token = data.get('access_token', token.access_token)
|
||||
token.refresh_token = data.get('refresh_token', token.refresh_token)
|
||||
token.expiration_seconds = data.get('expiration_seconds', token.expiration_seconds)
|
||||
token.readwise_api_key = data.get('readwise_api_key', token.readwise_api_key)
|
||||
token.updated_at = datetime.utcnow()
|
||||
db.session.commit()
|
||||
return '', 204
|
||||
|
||||
# deactivate token
|
||||
@app.route('/token/<id>/deactivate', methods=['POST'])
|
||||
def deactivate_token_by_id(id):
|
||||
token = Token.query.get_or_404(id)
|
||||
token.active = False
|
||||
db.session.commit()
|
||||
return '', 204
|
||||
|
||||
# get all tokens
|
||||
@app.route('/token/all', methods=['GET'])
|
||||
def get_all_tokens():
|
||||
only_active = request.args.get('only_active')
|
||||
tokens = Token.query.all() if not only_active else Token.query.filter_by(active=True).all()
|
||||
tokens_info = [{
|
||||
'id': token.id,
|
||||
'email': token.email,
|
||||
'access_token': token.access_token,
|
||||
'refresh_token': token.refresh_token,
|
||||
'expiration_seconds': int(token.expiration_seconds),
|
||||
'readwise_api_key': token.readwise_api_key,
|
||||
'active': token.active,
|
||||
'created_at': int(token.created_at.timestamp()),
|
||||
'updated_at': int(token.updated_at.timestamp())
|
||||
} for token in tokens]
|
||||
return jsonify({'tokens': tokens_info}), 200
|
||||
|
||||
# API to create or update the last annotation timestamp
|
||||
@app.route('/annotation_last_update', methods=['POST'])
|
||||
def create_or_update_annotation_last_update():
|
||||
data = request.get_json()
|
||||
email = data.get('email')
|
||||
last_update_time = data.get('last_update_time')
|
||||
|
||||
required_fields = ['email', 'last_update_time']
|
||||
missing_fields = [field for field in required_fields if not data.get(field)]
|
||||
|
||||
if missing_fields:
|
||||
return jsonify({'error': f'Missing required fields: {", ".join(missing_fields)}'}), 400
|
||||
|
||||
existing_annotation_last_update = AnnotationLastUpdate.query.filter_by(email=email).first()
|
||||
if existing_annotation_last_update:
|
||||
existing_annotation_last_update.last_update_time = last_update_time
|
||||
existing_annotation_last_update.updated_at = datetime.utcnow()
|
||||
db.session.commit()
|
||||
return '', 204
|
||||
else:
|
||||
new_annotation_last_update = AnnotationLastUpdate(
|
||||
email=email,
|
||||
last_update_time=last_update_time
|
||||
)
|
||||
db.session.add(new_annotation_last_update)
|
||||
db.session.commit()
|
||||
return '', 204
|
||||
|
||||
# API to get the last annotation timestamp based on the email
|
||||
@app.route('/annotation_last_update/<email>', methods=['GET'])
|
||||
def get_annotation_last_update_by_email(email):
|
||||
if not email:
|
||||
return jsonify({'error': 'Missing email query parameter'}), 400
|
||||
annotation_last_update = AnnotationLastUpdate.query.filter_by(email=email).first()
|
||||
if not annotation_last_update:
|
||||
return '', 204
|
||||
annotation_last_update_info = {
|
||||
'id': annotation_last_update.id,
|
||||
'email': annotation_last_update.email,
|
||||
'last_update_time': int(annotation_last_update.last_update_time.timestamp()),
|
||||
'created_at': int(annotation_last_update.created_at.timestamp()),
|
||||
'updated_at': int(annotation_last_update.updated_at.timestamp())
|
||||
}
|
||||
return jsonify(annotation_last_update_info), 200
|
||||
|
||||
if __name__ == '__main__':
|
||||
app.run(host='0.0.0.0', port=5000, debug=True)
|
||||
2
database/requirements.txt
Normal file
2
database/requirements.txt
Normal file
|
|
@ -0,0 +1,2 @@
|
|||
Flask==3.0.1
|
||||
Flask-SQLAlchemy==3.1.1
|
||||
Loading…
Reference in New Issue
Block a user