123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354 |
- import sqlite3, time, uuid, json
- from flask import Flask, request, render_template, jsonify
- conn = sqlite3.connect('texts.db', check_same_thread=False)
- conn.row_factory = sqlite3.Row
- def load_config():
- with open('config/api.json') as config:
- api = json.load(config)
- config = {
- "api": api
- }
- return config
- config = load_config()
- app = Flask(__name__)
- @app.route('/', methods=['GET'])
- def index():
- numbers = conn.execute('SELECT (SELECT body FROM public_messages WHERE `to` = msisdn ORDER BY timestamp), (SELECT `from` FROM public_messages WHERE `to` = msisdn ORDER BY timestamp), * FROM public_numbers;').fetchall()
- return render_template('index.html', numbers=numbers)
- @app.route('/public/<uid>', methods=['GET'])
- def show_public(uid):
- num = conn.execute("SELECT msisdn, country FROM public_numbers WHERE uid = ?;", [uid]).fetchone()
- if num is None:
- return 'Not found', 404
- messages = conn.execute("SELECT * FROM public_messages WHERE `to` = ?;", [num['msisdn']]).fetchall()
- return render_template('messages.html', num=num, messages=messages);
- @app.route('/api/receive', methods=['PUT'])
- def receive():
- message = request.get_json()
- if 'key' not in message or message['key'] != config["api"]["key"]:
- return 'Forbidden', 403
-
- numbers = list()
- for n in conn.execute('SELECT msisdn FROM public_numbers;').fetchall():
- numbers.append(n[0])
- if message['msisdn'] not in numbers:
- conn.execute('INSERT INTO public_numbers (uid, msisdn, country) VALUES (?, ?, ?);', (str(uuid.uuid4()), message['msisdn'], 'it'))
- conn.commit()
- try:
- conn.execute('INSERT INTO public_messages (uid, `to`, `from`, body, `timestamp`) VALUES (?, ?, ?, ?, ?);', [str(uuid.uuid4()), message['msisdn'], message['from'], message['text'], int(time.time())])
- conn.commit()
- return jsonify('Ok'), 200
- if __name__ == "__main__":
- app.run()
|