app.py

import os
import uuid
from flask import Flask, request
from flask_restful import Resource, Api
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///messages.db'
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
api = Api(app)
db = SQLAlchemy(app)

UPLOAD_FOLDER = 'messages'
ALLOWED_EXTENSIONS = {'txt'}
if not os.path.exists(UPLOAD_FOLDER):
    os.makedirs(UPLOAD_FOLDER)


class Message(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    file_path = db.Column(db.String(255), unique=True, nullable=False)
    content = db.Column(db.String(500), nullable=False)


class MessageResource(Resource):
    def post(self):
        message_content = request.json.get('message_content')
        if not message_content:
            return {'error': 'No message content provided'}, 400
        file_uuid = uuid.uuid4().hex
        filename = f"{file_uuid}.txt"
        file_path = os.path.join(UPLOAD_FOLDER, filename)
        with open(file_path, 'w') as f:
            f.write(message_content)
        message = Message(file_path=file_path, content=message_content)
        db.session.add(message)
        db.session.commit()
        return {'id': message.id}, 201


class NextMessageResource(Resource):
    def get(self):
        message = db.session.query(Message).order_by(Message.id.asc()).first()
        if not message:
            return {'error': 'No messages available'}, 404
        with open(message.file_path, 'r') as f:
            content = f.read()
        os.remove(message.file_path)
        db.session.delete(message)
        db.session.commit()
        return {'id': message.id, 'message': content}


api.add_resource(MessageResource, '/messages')
api.add_resource(NextMessageResource, '/next_message')

if __name__ == '__main__':
    db.create_all()
    app.run(debug=True)