mirror of
https://github.com/Blockstream/satellite-api.git
synced 2025-03-15 12:19:47 +01:00
2cfc39804e
This change removes the infinite transmission loop with sleep periods between each transmission. Now, a new transmission can start by a call to "tx_start" on the following two conditions: 1) As soon as the application starts (if there is a previously paid order waiting already). 2) As soon as the current transmission ends. The server will immediately look for the next order ready for transmission and start it if available. Meanwhile, the condition for ending the current transmission (i.e., for calling "tx_end(order)") is when all Tx confirmations are received from all regions.
59 lines
2 KiB
Python
59 lines
2 KiB
Python
import os
|
|
import shutil
|
|
import logging
|
|
|
|
from flask import Flask
|
|
from flask_restful import Api
|
|
import redis
|
|
|
|
import constants
|
|
from database import db
|
|
from info import InfoResource
|
|
from invoices import InvoiceResource
|
|
from orders import BumpOrderResource, GetMessageResource,\
|
|
GetMessageBySeqNumResource, OrderResource, OrdersResource,\
|
|
OrderUploadResource, RxConfirmationResource, TxConfirmationResource
|
|
from queues import QueueResource
|
|
|
|
|
|
def create_app(from_test=False):
|
|
if not os.path.isdir(constants.MSG_STORE_PATH):
|
|
os.makedirs(constants.MSG_STORE_PATH)
|
|
|
|
app = Flask(__name__)
|
|
app.config['SQLALCHEMY_DATABASE_URI'] = f'sqlite:///{constants.DB_FILE}'
|
|
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
|
|
app.config['TESTING'] = from_test
|
|
app.config["REDIS_INSTANCE"] = redis.from_url(constants.REDIS_URI)
|
|
|
|
db.init_app(app)
|
|
with app.app_context():
|
|
db.create_all()
|
|
api = Api(app)
|
|
api.add_resource(OrderUploadResource, '/order')
|
|
api.add_resource(OrdersResource, '/orders/<state>')
|
|
api.add_resource(OrderResource, '/order/<uuid>')
|
|
api.add_resource(BumpOrderResource, '/order/<uuid>/bump')
|
|
api.add_resource(TxConfirmationResource, '/order/tx/<tx_seq_num>')
|
|
api.add_resource(RxConfirmationResource, '/order/rx/<tx_seq_num>')
|
|
api.add_resource(InfoResource, '/info')
|
|
api.add_resource(InvoiceResource, '/callback/<lid>/<charged_auth_token>')
|
|
api.add_resource(GetMessageBySeqNumResource, '/message/<tx_seq_num>')
|
|
api.add_resource(QueueResource, '/queue.html')
|
|
|
|
if constants.env == 'development' or constants.env == 'test':
|
|
api.add_resource(GetMessageResource, '/order/<uuid>/sent_message')
|
|
|
|
logging.basicConfig(level=logging.DEBUG, format=constants.LOGGING_FORMAT)
|
|
return app
|
|
|
|
|
|
def teardown_app(app):
|
|
if (app.config['TESTING']):
|
|
shutil.rmtree(constants.MSG_STORE_PATH)
|
|
os.remove(constants.DB_FILE)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
app = create_app()
|
|
app.run(debug=True, host='0.0.0.0', port=constants.SERVER_PORT)
|