Update: model createdgit add .

Author: Shafaf

api/v1/plan.py

@@ -1,11 +1,39 @@
-from flask import jsonify
+from flask import jsonify, request
 from flask.views import MethodView
 
 from . import v1
+# from model.subscription import Plan
+from extension import db
 
 # test v1 api endpoint.
 class Plan(MethodView):
     def get(self):
         return jsonify({"test": "This is plan api"}), 200
+    
+    def post(self):
+        req :dict = request.get_json()
+        username = req.get("name")
+        if username is None:
+            return jsonify({"message": "name is required"}), 400
+        
+        email = req.get("email")
+        if email is None:
+            return jsonify({"message": "email is required"}), 400
+        # user = Plan(
+        #     username=username,
+        #     email=email
+        # )
+        
+        print("start")
+        print(username)
+        print(email)
+        print("end")
+        
+        # db.session.add(user)
+        # db.session.commit()
+        
+        print(user.id)
+        
+        return jsonify({"status": True})
 
 v1.add_url_rule("/plan", view_func=Plan.as_view("plan"))

app.py

@@ -1,23 +1,37 @@
 from flask import Flask
-from sqlalchemy import create_engine
+from flask_sqlalchemy import SQLAlchemy
 
 from api.v1 import v1
-
-# from connection import conn
-# from model.subscription import hello
+from connection import SQLALCHEMY_DATABASE_URI
+from extension import db , migrate
+from model import Plan
 
 def create_app():
     app = Flask(__name__)
 
-    print("Result fetched successfully")
+    app.config["SQLALCHEMY_DATABASE_URI"] = SQLALCHEMY_DATABASE_URI
+    
+    
+    extensions(app)
 
+    
+    # register the api points.
     app.register_blueprint(v1)
 
+    # checking is the application is running, remove me later.
     @app.route("/")
     def index():
         return "hello how are you!"
 
     return app
 
 
+def extensions(app):
+    db.init_app(app)
+    migrate.init_app(app, db)
+    
+    with app.app_context():
+        db.create_all()
+    print("Result fetched successfully")
+
 create_app()

connection.py

@@ -1,18 +1,14 @@
 import os
-import psycopg2;
 
 # Get the environment variables
-DB_USER = os.getenv('POSTGRES_USER', "postgres")
-DB_PASSWORD = os.getenv('POSTGRES_PASSWORD', "postgres")
-# DB_HOST = os.getenv('POSTGRES_HOST', "postgres")
-DB_PORT = os.getenv('POSTGRES_PORT', "5432")
-DB_NAME = os.getenv('POSTsGRES_DB', "demodb")
+pg_user = os.getenv('POSTGRES_USER', "muhammedshafaf")
+pg_pass = os.getenv('POSTGRES_PASSWORD', "postgres")
+pg_host = os.getenv('POSTGsRES_HOST', "localhost")
+pg_port = os.getenv('POSTGRES_PORT', "5432")
+pg_db = os.getenv('POSTsGRES_DB', "demodb")
 
-# database connection login
-conn = psycopg2.connect(
-    dbname=DB_NAME,
-    user=DB_USER,
-    password=DB_PASSWORD,
-    port=DB_PORT
-)
 
+SQLALCHEMY_DATABASE_URI = os.getenv(
+    "DATABASE_URL",
+    f"postgresql://{pg_user}:{pg_pass}@{pg_host}:{pg_port}/{pg_db}",
+)

extension.py

@@ -0,0 +1,5 @@
+from flask_sqlalchemy import SQLAlchemy
+from flask_migrate import Migrate
+
+db = SQLAlchemy()
+migrate = Migrate()

model/__init__.py

@@ -0,0 +1 @@
+from .subscription import Plan

model/subscription.py

@@ -1,6 +1,37 @@
+from sqlalchemy import Column, String, Integer, ForeignKey, Enum
+from sqlalchemy.orm import Mapped, mapped_column
+from extension import db
 
+class Plan(db.Model):
+    __tablename__ = "plan"
+    
+    id = Column(Integer, primary_key=True, nullable=False)
+    name = Column(String, nullable=False)
+    description = Column(String(300), nullable=True)
+    amount = Column(Integer, nullable=False)
+    intervals = Column(Integer, nullable=False)
+    plan_id = Column(String)
+    
+class Subscription(db.Model):
+    __tablename__ = "subscription"
+    
+    STATUS_CHOICES = ["active", "disabled"]
+    
+    id = Column(Integer, primary_key=True, nullable=False)
+    plan_id = Column(Integer, ForeignKey(Plan.id), nullable=False)
+    razorpay_user_id = Column(String)
+    subscription_id = Column(String)
+    subscription_status = Column(
+        Enum(*STATUS_CHOICES, name="status", native_enum=False),
+        nullable=False,
+        default="active",
+    )
+    
 
-# engine = create_engine('postgresql:///tutorial.db', echo=True)
-def hello():
-    print("Hello from database.py")
-    print("========== Database connection established successfully ==========")
+class User(db.Model):
+    __tablename__ = "user"
+    id = Column(Integer, primary_key=True, nullable=False)
+    name = Column(String, nullable=False)
+    email = Column(String, nullable=False)
+    phone_number = Column(String, nullable=False)
+    

requirements.txt

@@ -1,3 +1,5 @@
 Flask==3.0.2
 psycopg2==2.9.9
-SQLAlchemy==2.0.29
+SQLAlchemy==2.0.29
+Flask-SQLAlchemy
+Flask-Migrate