Add: start price field in position table

web_app/alembic/versions/a009512f5362_start_price_field_added_in_position.py

@@ -0,0 +1,50 @@
+"""start_price field added in position
+
+Revision ID: a009512f5362
+Revises: b705d1435b64
+Create Date: 2024-10-24 18:56:29.399344
+
+"""
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy import inspect
+
+
+# revision identifiers, used by Alembic.
+revision = 'a009512f5362'
+down_revision = 'b705d1435b64'
+branch_labels = None
+depends_on = None
+
+
+def column_exists(table_name, column_name):
+    """Utility function to check if a column exists in the table."""
+    bind = op.get_bind()
+    inspector = inspect(bind)
+    columns = [col['name'] for col in inspector.get_columns(table_name)]
+    return column_name in columns
+
+def upgrade() -> None:
+    """Upgrade the database."""
+
+    if column_exists('position', 'start_price'):
+        print("Column 'start_price' already exists, skipping creation.")
+    else:
+        op.add_column('position', sa.Column('start_price', sa.DECIMAL(), nullable=False))
+        print("Column 'start_price' added to the 'position' table.")
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.alter_column('position', 'start_price',
+               existing_type=sa.DOUBLE_PRECISION(precision=53),
+               type_=sa.DECIMAL(),
+               existing_nullable=False)
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    """Downgrade the database."""
+
+    if column_exists('position', 'start_price'):
+        print("Column 'start_price' exists, downgrading.")
+        op.drop_column('position', 'start_price')
+    else:
+        print("Column 'start_price' already removed, skipping.")

web_app/db/crud.py

@@ -9,14 +9,8 @@
 from sqlalchemy import create_engine
 from sqlalchemy.exc import SQLAlchemyError
 from sqlalchemy.orm import scoped_session, sessionmaker
-
 from web_app.db.database import SQLALCHEMY_DATABASE_URL
-from web_app.db.models import (
-    Base,
-    User,
-    Position,
-    Status,
-)
+from web_app.db.models import Base, Position, Status, User
 
 logger = logging.getLogger(__name__)
 ModelType = TypeVar("ModelType", bound=Base)
@@ -168,6 +162,8 @@ class PositionDBConnector(UserDBConnector):
     Provides database connection and operations management for the Position model.
     """
 
+    START_PRICE = 0.0
+
     @staticmethod
     def _position_to_dict(position: Position) -> dict:
         """
@@ -197,7 +193,7 @@ def _get_user_by_wallet_id(self, wallet_id: str) -> User | None:
 
     def get_positions_by_wallet_id(self, wallet_id: str) -> list:
         """
-        Retrieves all positions for a user by their wallet ID 
+        Retrieves all positions for a user by their wallet ID
         and returns them as a list of dictionaries.
         :param wallet_id: str
         :return: list of dict
@@ -259,6 +255,7 @@ def create_position(
                 existing_position.token_symbol = token_symbol
                 existing_position.amount = amount
                 existing_position.multiplier = multiplier
+                existing_position.start_price = PositionDBConnector.START_PRICE
                 session.commit()  # Commit the changes to the database
                 session.refresh(existing_position)  # Refresh to get updated values
                 return existing_position
@@ -270,6 +267,7 @@ def create_position(
                 amount=amount,
                 multiplier=multiplier,
                 status=Status.PENDING.value,  # Set status as 'pending' by default
+                start_price=PositionDBConnector.START_PRICE,
             )
 
             # Write the new position to the database
@@ -287,7 +285,6 @@ def get_position_id_by_wallet_id(self, wallet_id: str) -> str | None:
             return position[0]["id"]
         return None
 
-
     def update_position(self, position: Position, amount: str, multiplier: int) -> None:
         """
         Updates a position in the database.

web_app/db/database.py

@@ -36,4 +36,4 @@ def get_database() -> SessionLocal:
     try:
         yield database
     finally:
-        database.close()
+        database.close()

web_app/db/models.py

@@ -2,11 +2,21 @@
 This module contains the SQLAlchemy models for the database.
 """
 
+from enum import Enum as PyEnum
 from uuid import uuid4
+
+from sqlalchemy import (
+    DECIMAL,
+    Boolean,
+    Column,
+    DateTime,
+    Enum,
+    ForeignKey,
+    Integer,
+    String,
+)
 from sqlalchemy.dialects.postgresql import UUID
 from sqlalchemy.sql import func
-from sqlalchemy import Column, String, Boolean, Integer, ForeignKey, DateTime, Enum
-from enum import Enum as PyEnum
 from web_app.db.database import Base
 
 
@@ -62,3 +72,4 @@ class Position(Base):
         nullable=True,
         default="pending",
     )
+    start_price = Column(DECIMAL, nullable=False)