Server.py - master - olegk2001/CryptoApp

Server.py
76 строк2 KB
Oleg K
Initial sanitized commit
08 июн 2026, 10:08
08 июн 2026, 10:08104414c
from flask import Flask, request, jsonify
from flask_cors import CORS
import sqlite3
from datetime import datetime

app = Flask(__name__)
CORS(app)


# Инициализация базы данных
def init_db():
    conn = sqlite3.connect('messages.db')
    cursor = conn.cursor()
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS messages (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            user_id TEXT NOT NULL,
            message TEXT NOT NULL,
            timestamp TEXT NOT NULL
        )
    ''')
    conn.commit()
    conn.close()


# Сохранение сообщения в базу данных
@app.route('/save', methods=['POST'])
def save_message():
    data = request.json
    user_id = data.get('user_id')
    message = data.get('message')

    if not user_id or not message:
        return jsonify({'error': 'user_id and message are required'}), 400

    timestamp = datetime.now().isoformat()

    conn = sqlite3.connect('messages.db')
    cursor = conn.cursor()
    cursor.execute('INSERT INTO messages (user_id, message, timestamp) VALUES (?, ?, ?)', (user_id, message, timestamp))
    conn.commit()
    conn.close()

    return jsonify({'status': 'success', 'timestamp': timestamp}), 201


# Получение новых сообщений
@app.route('/messages', methods=['GET'])
def get_messages():
    user_id = request.args.get('user_id')
    last_timestamp = request.args.get('last_timestamp')

    if not user_id:
        return jsonify({'error': 'user_id is required'}), 400

    conn = sqlite3.connect('messages.db')
    cursor = conn.cursor()

    if last_timestamp:
        cursor.execute(
            'SELECT id, message, timestamp FROM messages WHERE user_id = ? AND timestamp > ? ORDER BY timestamp ASC',
            (user_id, last_timestamp))
    else:
        cursor.execute('SELECT id, message, timestamp FROM messages WHERE user_id = ? ORDER BY timestamp ASC',
                       (user_id,))

    messages = cursor.fetchall()
    conn.close()

    result = [{'id': msg[0], 'message': msg[1], 'timestamp': msg[2]} for msg in messages]
    return jsonify(result)


if __name__ == '__main__':
    init_db()
    app.run(debug=False, host='0.0.0.0', port=5000)