Backend api and models are added

app/__init__.py

@@ -0,0 +1,25 @@
+from flask import Flask
+from flask_sqlalchemy import SQLAlchemy
+from flask_jwt_extended import JWTManager
+from flask_bcrypt import Bcrypt
+
+db = SQLAlchemy()
+jwt = JWTManager()
+bcrypt = Bcrypt()
+
+def create_app():
+    app = Flask(__name__)
+    app.config.from_object('app.config.Config')
+
+    db.init_app(app)
+    jwt.init_app(app)
+    bcrypt.init_app(app)
+
+    with app.app_context():
+        from app.routes import auth, backtest, index
+        app.register_blueprint(auth.bp)
+        app.register_blueprint(backtest.bp)
+        app.register_blueprint(index.bp)
+        db.create_all()
+
+    return app

app/config.py

@@ -0,0 +1,4 @@
+class Config:
+    SQLALCHEMY_DATABASE_URI = 'db_utl'
+    SQLALCHEMY_TRACK_MODIFICATIONS = False
+    JWT_SECRET_KEY = 'your_secret_key'

app/database.py

@@ -0,0 +1,2 @@
+from app import db
+from app.models import backtest, user

app/main.py

@@ -0,0 +1,6 @@
+from app import create_app
+
+app = create_app()
+
+if __name__ == "__main__":
+    app.run(debug=True)

app/models/backtest.py

@@ -0,0 +1,41 @@
+from app import db
+
+class Backtest(db.Model):
+    __tablename__ = 'backtests'
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String(255))
+    start_date = db.Column(db.Date)
+    end_date = db.Column(db.Date)
+    status = db.Column(db.String(50))
+    created_at = db.Column(db.DateTime, default=db.func.current_timestamp())
+
+class Indicator(db.Model):
+    __tablename__ = 'indicators'
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String(255))
+    description = db.Column(db.Text)
+
+class Parameter(db.Model):
+    __tablename__ = 'parameters'
+    id = db.Column(db.Integer, primary_key=True)
+    backtest_id = db.Column(db.Integer, db.ForeignKey('backtests.id'), nullable=False)
+    indicator_id = db.Column(db.Integer, db.ForeignKey('indicators.id'), nullable=False)
+    value = db.Column(db.String(255))
+
+class Result(db.Model):
+    __tablename__ = 'results'
+    id = db.Column(db.Integer, primary_key=True)
+    backtest_id = db.Column(db.Integer, db.ForeignKey('backtests.id'), nullable=False)
+    total_return = db.Column(db.Numeric(10, 2))
+    number_of_trades = db.Column(db.Integer)
+    winning_trades = db.Column(db.Integer)
+    losing_trades = db.Column(db.Integer)
+    max_drawdown = db.Column(db.Numeric(10, 2))
+    sharpe_ratio = db.Column(db.Numeric(10, 2))
+
+class Metric(db.Model):
+    __tablename__ = 'metrics'
+    id = db.Column(db.Integer, primary_key=True)
+    backtest_id = db.Column(db.Integer, db.ForeignKey('backtests.id'), nullable=False)
+    metric_name = db.Column(db.String(255))
+    metric_value = db.Column(db.Numeric(10, 2))

app/models/user.py

@@ -0,0 +1,8 @@
+from app import db
+
+class User(db.Model):
+    __tablename__ = 'users'
+    id = db.Column(db.Integer, primary_key=True)
+    username = db.Column(db.String(255), unique=True, nullable=False)
+    password = db.Column(db.String(255), nullable=False)
+    created_at = db.Column(db.DateTime, default=db.func.current_timestamp())

app/routes/auth.py

@@ -0,0 +1,38 @@
+from flask import Blueprint, request, jsonify
+from app.models.user import User
+from app import db, bcrypt, jwt
+from flask_jwt_extended import create_access_token
+
+bp = Blueprint('auth', __name__)
+
+
+@bp.route('/register', methods=['POST'])
+def register():
+    data = request.get_json()
+    username = data.get('username')
+    password = data.get('password')
+
+    if User.query.filter_by(username=username).first():
+        return jsonify({"msg": "Username already exists"}), 409
+
+    hashed_password = bcrypt.generate_password_hash(password).decode('utf-8')
+    new_user = User(username=username, password=hashed_password)
+    db.session.add(new_user)
+    db.session.commit()
+
+    return jsonify({"msg": "User registered successfully"}), 201
+
+
+@bp.route('/login', methods=['POST'])
+def login():
+    data = request.get_json()
+    username = data.get('username')
+    password = data.get('password')
+
+    user = User.query.filter_by(username=username).first()
+
+    if user and bcrypt.check_password_hash(user.password, password):
+        access_token = create_access_token(identity={'username': user.username})
+        return jsonify(access_token=access_token), 200
+
+    return jsonify({"msg": "Bad username or password"}), 401

app/routes/backtest.py

@@ -0,0 +1,56 @@
+from flask import Blueprint, request, jsonify
+from app.models.backtest import Backtest, Parameter, Result
+from app import db
+from flask_jwt_extended import jwt_required
+
+bp = Blueprint('backtest', __name__)
+
+
+@bp.route('/backtest', methods=['POST'])
+@jwt_required()
+def run_backtest():
+    data = request.get_json()
+    name = data.get('name')
+    start_date = data.get('start_date')
+    end_date = data.get('end_date')
+    parameters = data.get('parameters')
+
+    # Check if backtest with same parameters exists
+    existing_backtest = Backtest.query.filter_by(name=name, start_date=start_date, end_date=end_date).first()
+    if existing_backtest:
+        return jsonify(
+            {"msg": "Backtest with same parameters already exists", "backtest_id": existing_backtest.id}), 200
+
+    # Create new backtest
+    new_backtest = Backtest(name=name, start_date=start_date, end_date=end_date, status="pending")
+    db.session.add(new_backtest)
+    db.session.commit()
+
+    # Add parameters
+    for param in parameters:
+        new_param = Parameter(backtest_id=new_backtest.id, indicator_id=param['indicator_id'], value=param['value'])
+        db.session.add(new_param)
+
+    db.session.commit()
+
+    # Run the backtest (dummy processing here)
+    result = run_backtest_logic(new_backtest.id)
+
+    return jsonify(result), 201
+
+
+def run_backtest_logic(backtest_id):
+    # Dummy backtest logic
+    result = Result(
+        backtest_id=backtest_id,
+        total_return=10.5,
+        number_of_trades=20,
+        winning_trades=15,
+        losing_trades=5,
+        max_drawdown=3.5,
+        sharpe_ratio=1.8
+    )
+    db.session.add(result)
+    db.session.commit()
+
+    return {"msg": "Backtest completed", "backtest_id": backtest_id, "result": result.id}

app/routes/index.py

@@ -0,0 +1,7 @@
+from flask import Blueprint
+
+bp = Blueprint('index', __name__)
+
+@bp.route('/')
+def index():
+    return "Welcome to the Backtest API!"

requirements.txt

@@ -0,0 +1,5 @@
+flask
+Flask-SQLAlchemy
+Flask-JWT-Extended
+Flask-Bcrypt
+psycopg2-binary

run.py

@@ -0,0 +1,6 @@
+from app import create_app
+
+app = create_app()
+
+if __name__ == "__main__":
+    app.run(debug=True)