Refactor Planventure API to integrate SQLAlchemy for database management, add user and trip models, and implement database initialization script.

Author: LadyKerr

.gitignore

@@ -1,3 +1,4 @@
 .vscode
 # Bruno adds a dir to your vscode workspace
-Planventure
+Planventure
+venv/

planventure-api/app.py

@@ -1,16 +1,37 @@
 from flask import Flask, jsonify
 from flask_cors import CORS
+from flask_sqlalchemy import SQLAlchemy
+from os import environ
+from dotenv import load_dotenv
 
-app = Flask(__name__)
-CORS(app)
+# Load environment variables
+load_dotenv()
 
-@app.route('/')
-def home():
-    return jsonify({"message": "Welcome to PlanVenture API"})
+# Initialize SQLAlchemy
+db = SQLAlchemy()
 
-@app.route('/health')
-def health_check():
-    return jsonify({"status": "healthy"})
+def create_app():
+    app = Flask(__name__)
+    CORS(app)
+
+    # Database configuration
+    app.config['SQLALCHEMY_DATABASE_URI'] = environ.get('DATABASE_URL', 'sqlite:///planventure.db')
+    app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+
+    # Initialize extensions
+    db.init_app(app)
+
+    # Register routes
+    @app.route('/')
+    def home():
+        return jsonify({"message": "Welcome to PlanVenture API"})
+
+    @app.route('/health')
+    def health_check():
+        return jsonify({"status": "healthy"})
+
+    return app
 
 if __name__ == '__main__':
+    app = create_app()
     app.run(debug=True)

planventure-api/init_db.py

@@ -0,0 +1,12 @@
+from app import create_app, db
+from models import User
+
+def init_db():
+    app = create_app()
+    with app.app_context():
+        # Create all database tables
+        db.create_all()
+        print("Database tables created successfully!")
+
+if __name__ == '__main__':
+    init_db()

planventure-api/instance/planventure.db

@@ -0,0 +1,4 @@
+from .user import User
+from .trip import Trip
+
+__all__ = ['User', 'Trip']

planventure-api/models/__init__.py

@@ -0,0 +1,22 @@
+from datetime import datetime, timezone
+from app import db
+
+class Trip(db.Model):
+    __tablename__ = 'trips'
+
+    id = db.Column(db.Integer, primary_key=True)
+    user_id = db.Column(db.Integer, db.ForeignKey('users.id'), nullable=False)
+    destination = db.Column(db.String(200), nullable=False)
+    start_date = db.Column(db.DateTime, nullable=False)
+    end_date = db.Column(db.DateTime, nullable=False)
+    latitude = db.Column(db.Float)
+    longitude = db.Column(db.Float)
+    itinerary = db.Column(db.JSON)
+    created_at = db.Column(db.DateTime, default=lambda: datetime.now(timezone.utc))
+    updated_at = db.Column(db.DateTime, default=lambda: datetime.now(timezone.utc), onupdate=lambda: datetime.now(timezone.utc))
+
+    # Relationship
+    user = db.relationship('User', back_populates='trips')
+
+    def __repr__(self):
+        return f'<Trip {self.destination} ({self.start_date} - {self.end_date})>'

planventure-api/models/trip.py

@@ -0,0 +1,17 @@
+from datetime import datetime, timezone
+from app import db
+
+class User(db.Model):
+    __tablename__ = 'users'
+
+    id = db.Column(db.Integer, primary_key=True)
+    email = db.Column(db.String(120), unique=True, nullable=False)
+    password_hash = db.Column(db.String(128), nullable=False)
+    created_at = db.Column(db.DateTime, default=lambda: datetime.now(timezone.utc))
+    updated_at = db.Column(db.DateTime, default=lambda: datetime.now(timezone.utc), onupdate=lambda: datetime.now(timezone.utc))
+
+    # Add relationship
+    trips = db.relationship('Trip', back_populates='user', cascade='all, delete-orphan')
+
+    def __repr__(self):
+        return f'<User {self.email}>'

planventure-api/models/user.py

@@ -2,4 +2,14 @@
 flask==2.3.3
 flask-sqlalchemy==3.1.1
 flask-cors==4.0.0
-python-dotenv==1.0.0
+python-dotenv==1.0.0
+
+# Authentication
+flask-jwt-extended==4.5.3
+werkzeug==2.3.7
+
+# Password hashing
+bcrypt==4.0.1
+
+# Database migrations
+Flask-Migrate==4.0.5