-
-
Notifications
You must be signed in to change notification settings - Fork 41
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
1 changed file
with
136 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,136 @@ | ||
# main.py | ||
|
||
import os | ||
import json | ||
import logging | ||
from flask import Flask, request, jsonify, g | ||
from flask_jwt_extended import JWTManager, jwt_required, create_access_token, get_jwt_identity | ||
from flask_sqlalchemy import SQLAlchemy | ||
from flask_marshmallow import Marshmallow | ||
from flask_cors import CORS | ||
from app.models import User, Transaction, Account, Node, Block, Blockchain | ||
from app.services import AuthService, PaymentService, AnalyticsService, BlockchainService | ||
from app.utils import constants, helpers | ||
from app.blockchain import BlockchainUtils | ||
|
||
app = Flask(__name__) | ||
app.config.from_object(constants.Config) | ||
|
||
db = SQLAlchemy(app) | ||
ma = Marshmallow(app) | ||
jwt = JWTManager(app) | ||
cors = CORS(app, resources={r"/api/*": {"origins": "*"}}) | ||
|
||
logging.basicConfig(level=logging.INFO) | ||
logger = logging.getLogger(__name__) | ||
|
||
@app.before_request | ||
def before_request(): | ||
g.user = None | ||
if 'Authorization' in request.headers: | ||
token = request.headers['Authorization'].split()[1] | ||
g.user = get_jwt_identity() | ||
|
||
@app.route('/api/login', methods=['POST']) | ||
def login(): | ||
username = request.json.get('username') | ||
password = request.json.get('password') | ||
auth_service = AuthService() | ||
response = auth_service.login(username, password) | ||
return jsonify(response), 200 if response.get('access_token') else 401 | ||
|
||
@app.route('/api/register', methods=['POST']) | ||
def register(): | ||
username = request.json.get('username') | ||
password = request.json.get('password') | ||
email = request.json.get('email') | ||
auth_service = AuthService() | ||
response = auth_service.register(username, password, email) | ||
return jsonify(response), 201 | ||
|
||
@app.route('/api/users', methods=['GET']) | ||
@jwt_required | ||
def get_users(): | ||
user_service = UserService() | ||
users = user_service.get_users() | ||
return jsonify([user.to_dict() for user in users]), 200 | ||
|
||
@app.route('/api/transactions', methods=['GET']) | ||
@jwt_required | ||
def get_transactions(): | ||
transaction_service = TransactionService() | ||
transactions = transaction_service.get_transactions() | ||
return jsonify([transaction.to_dict() for transaction in transactions]), 200 | ||
|
||
@app.route('/api/accounts', methods=['GET']) | ||
@jwt_required | ||
def get_accounts(): | ||
account_service = AccountService() | ||
accounts = account_service.get_accounts() | ||
return jsonify([account.to_dict() for account in accounts]), 200 | ||
|
||
@app.route('/api/make_payment', methods=['POST']) | ||
@jwt_required | ||
def make_payment(): | ||
user_id = g.user | ||
account_id = request.json.get('account_id') | ||
amount = request.json.get('amount') | ||
payment_service = PaymentService() | ||
response = payment_service.make_payment(user_id, account_id, amount) | ||
return jsonify(response), 200 if response.get('message') == 'Payment successful' else 400 | ||
|
||
@app.route('/api/get_payment_history', methods=['GET']) | ||
@jwt_required | ||
def get_payment_history(): | ||
user_id = g.user | ||
payment_service = PaymentService() | ||
transactions = payment_service.get_payment_history(user_id) | ||
return jsonify(transactions), 200 | ||
|
||
@app.route('/api/analytics', methods=['GET']) | ||
@jwt_required | ||
def get_analytics(): | ||
analytics_service = AnalyticsService() | ||
top_spenders = analytics_service.get_top_spenders() | ||
transaction_volume = analytics_service.get_transaction_volume() | ||
return jsonify({'top_spenders': top_spenders, 'transaction_volume': transaction_volume}), 200 | ||
|
||
@app.route('/api/blockchain', methods=['GET']) | ||
@jwt_required | ||
def get_blockchain(): | ||
blockchain_service = BlockchainService() | ||
blockchain = blockchain_service.get_blockchain() | ||
return jsonify(blockchain), 200 | ||
|
||
@app.route('/api/nodes', methods=['GET']) | ||
@jwt_required | ||
def get_nodes(): | ||
node_service = NodeService() | ||
nodes = node_service.get_nodes() | ||
return jsonify([node.to_dict() for node in nodes]), 200 | ||
|
||
@app.route('/api/blocks', methods=['GET']) | ||
@jwt_required | ||
def get_blocks(): | ||
block_service = BlockService() | ||
blocks = block_service.get_blocks() | ||
return jsonify([block.to_dict() for block in blocks]), 200 | ||
|
||
@app.route('/api/transactions/pending', methods=['GET']) | ||
@jwt_required | ||
def get_pending_transactions(): | ||
transaction_service = TransactionService() | ||
transactions = transaction_service.get_pending_transactions() | ||
return jsonify([transaction.to_dict() for transaction in transactions]), 200 | ||
|
||
@app.route('/api/transactions/confirmed', methods=['GET']) | ||
@jwt_required | ||
def get_confirmed_transactions(): | ||
transaction_service = TransactionService() | ||
transactions = transaction_service.get_confirmed_transactions() | ||
return jsonify([transaction.to_dict() for transaction in transactions]), 200 | ||
|
||
@app.route('/api/blockchain/sync', methods=['POST']) | ||
@jwt_required | ||
def sync_blockchain(): | ||
blockchain_service = BlockchainService |