From 50d368cb76cbd023e50cc55f3c073f5fb6220d75 Mon Sep 17 00:00:00 2001
From: AGGIB <113549303+AGGIB@users.noreply.github.com>
Date: Mon, 28 Oct 2024 02:06:02 +0500
Subject: [PATCH] Update main.py

---
 main.py | 74 +++++++++++++++++++++++++++++++++++++++++++++++++++++----
 1 file changed, 69 insertions(+), 5 deletions(-)

diff --git a/main.py b/main.py
index f781854e..39a00ea3 100644
--- a/main.py
+++ b/main.py
@@ -1,13 +1,77 @@
-from flask import Flask, jsonify
+from flask import Flask, request, jsonify
+from flask_sqlalchemy import SQLAlchemy
+from datetime import datetime
 import os
 
 app = Flask(__name__)
+app.config['SQLALCHEMY_DATABASE_URI'] = os.getenv('postgresql://postgres:AmXtGqYBLoaySfqtBhGRzTkaiSojGJds@postgres.railway.internal:5432/railway')
+db = SQLAlchemy(app)
 
+class Flight(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    from_city = db.Column(db.String(100), nullable=False)
+    to_city = db.Column(db.String(100), nullable=False)
+    departure_time = db.Column(db.DateTime, nullable=False)
+    price = db.Column(db.Float, nullable=False)
+    available_seats = db.Column(db.Integer, nullable=False)
 
-@app.route('/')
-def index():
-    return jsonify({"Choo Choo": "Welcome to your Flask app 🚅"})
+class Booking(db.Model):
+    id = db.Column(db.Integer, primary_key=True)
+    flight_id = db.Column(db.Integer, db.ForeignKey('flight.id'), nullable=False)
+    passenger_name = db.Column(db.String(100), nullable=False)
+    booking_status = db.Column(db.String(20), nullable=False)
+    payment_status = db.Column(db.String(20), nullable=False)
 
+@app.route('/flights/search', methods=['GET'])
+def search_flights():
+    from_city = request.args.get('from')
+    to_city = request.args.get('to')
+    date = request.args.get('date')
+    
+    query = Flight.query.filter_by(
+        from_city=from_city,
+        to_city=to_city,
+    )
+    
+    if date:
+        date = datetime.fromisoformat(date)
+        query = query.filter(
+            Flight.departure_time >= date,
+            Flight.departure_time < date.replace(hour=23, minute=59)
+        )
+    
+    flights = query.all()
+    return jsonify([{
+        'id': f.id,
+        'from': f.from_city,
+        'to': f.to_city,
+        'departure': f.departure_time.isoformat(),
+        'price': f.price,
+        'available_seats': f.available_seats
+    } for f in flights])
+
+@app.route('/bookings', methods=['POST'])
+def create_booking():
+    data = request.json
+    flight_id = data.get('flight_id')
+    passenger_name = data.get('passenger_name')
+    
+    flight = Flight.query.get(flight_id)
+    if not flight or flight.available_seats < 1:
+        return jsonify({'error': 'No available seats'}), 400
+        
+    booking = Booking(
+        flight_id=flight_id,
+        passenger_name=passenger_name,
+        booking_status='pending',
+        payment_status='pending'
+    )
+    
+    db.session.add(booking)
+    flight.available_seats -= 1
+    db.session.commit()
+    
+    return jsonify({'booking_id': booking.id})
 
 if __name__ == '__main__':
-    app.run(debug=True, port=os.getenv("PORT", default=5000))
+    app.run(debug=True)