add charge endpoint

Author: surajit

app.py

@@ -7,11 +7,13 @@
 from routes.v1 import email
 from routes.v1 import contact
 from routes.v1 import company
+from routes.v1 import charge
 
 app = FastAPI()
 app.include_router(email.router)
 app.include_router(contact.router)
 app.include_router(company.router)
+app.include_router(charge.router)
 
 templates = Jinja2Templates(directory="templates")
 

crud/charge.py

@@ -24,9 +24,10 @@
 
 # add your model's MetaData object here
 # for 'autogenerate' support
-from models import payment
+from models.payment import Charge
 from models import contact
 from models import message
+
 target_metadata = Base.metadata
 
 

migrations/env.py

@@ -0,0 +1,30 @@
+"""add created_at and modified_at to charge table
+
+Revision ID: 573a2f55e475
+Revises: 3f4f812d4ca6
+Create Date: 2022-05-26 19:50:38.717366
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '573a2f55e475'
+down_revision = '3f4f812d4ca6'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('charge', sa.Column('created_at', sa.DateTime(), nullable=True))
+    op.add_column('charge', sa.Column('modified_at', sa.DateTime(), nullable=True))
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('charge', 'modified_at')
+    op.drop_column('charge', 'created_at')
+    # ### end Alembic commands ###

migrations/versions/573a2f55e475_add_created_at_and_modified_at_to_.py

@@ -0,0 +1,28 @@
+"""remove unique constraint
+
+Revision ID: 5bf098d393cd
+Revises: 9b9c2855cad1
+Create Date: 2022-05-26 20:04:41.897338
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '5bf098d393cd'
+down_revision = '9b9c2855cad1'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    pass
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    pass
+    # ### end Alembic commands ###

migrations/versions/5bf098d393cd_remove_unique_constraint.py

@@ -0,0 +1,28 @@
+"""add unique constraint for charge type and org_id for charge table
+
+Revision ID: 9b9c2855cad1
+Revises: 573a2f55e475
+Create Date: 2022-05-26 19:53:14.981004
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '9b9c2855cad1'
+down_revision = '573a2f55e475'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    pass
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    pass
+    # ### end Alembic commands ###

migrations/versions/9b9c2855cad1_add_unique_constraint_for_charge_type_.py

@@ -0,0 +1,28 @@
+"""modify unique constraint
+
+Revision ID: a9e00646e1d7
+Revises: ad9ebc89187d
+Create Date: 2022-05-26 23:23:53.573992
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = 'a9e00646e1d7'
+down_revision = 'ad9ebc89187d'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.create_unique_constraint(None, 'charge', ['org_id', 'charge_type'])
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_constraint(None, 'charge', type_='unique')
+    # ### end Alembic commands ###

migrations/versions/a9e00646e1d7_modify_unique_constraint.py

@@ -0,0 +1,28 @@
+"""add unique constraint to charge
+
+Revision ID: ad9ebc89187d
+Revises: 5bf098d393cd
+Create Date: 2022-05-26 20:05:35.891406
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = 'ad9ebc89187d'
+down_revision = '5bf098d393cd'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    pass
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    pass
+    # ### end Alembic commands ###

migrations/versions/ad9ebc89187d_add_unique_constraint_to_charge.py

@@ -1,10 +1,13 @@
 import enum
+import datetime
 
 from sqlalchemy import Column
 from sqlalchemy import Integer
 from sqlalchemy import String
 from sqlalchemy import Enum
 from sqlalchemy import ForeignKey
+from sqlalchemy import DateTime
+from sqlalchemy import UniqueConstraint
 from sqlalchemy.orm import relationship
 
 from database import Base
@@ -32,13 +35,15 @@ class Organization(Base):
 
 class Charge(Base):
     __tablename__ = "charge"
-
+    __table_args__ = (UniqueConstraint("org_id", "charge_type"),)
     id = Column(Integer, primary_key=True, index=True)
     currency = Column(Enum(Currency), nullable=False, default=Currency.USD)
     amount = Column(Integer, nullable=False)
     charge_type = Column(Enum(ChargeType), nullable=False)
     org_id = Column(Integer, ForeignKey("organization.id"))
     organization = relationship("Organization", backref="charges")
+    created_at = Column(DateTime, default=datetime.datetime.now)
+    modified_at = Column(DateTime, default=datetime.datetime.now)
 
 
 class Payment(Base):

models/payment.py

@@ -0,0 +1,39 @@
+import datetime
+
+from fastapi import APIRouter
+from fastapi import Depends
+from fastapi import HTTPException
+from fastapi import Request
+from sqlalchemy.orm import Session
+
+from crud.charge import ChargeExistException
+from crud.company import filter_company_by_name
+from logger.log import save_log
+from dependencies.dependencies import get_db
+from crud import charge as _charge
+
+router = APIRouter(
+    prefix="/api/v1",
+    tags=["charge"],
+    responses={404: {"description": "Not found"}},
+)
+
+
+@router.post("/charges/")
+@save_log
+async def create_charge(request: Request, db: Session = Depends(get_db)):
+    try:
+        data = await request.json()
+        company_name = data["company_name"]
+        company = filter_company_by_name(db, company_name).all()[0]
+        company_data = dict(
+            currency=data["currency"],
+            amount=data["amount"],
+            charge_type=data["charge_type"],
+            org_id=company.id,
+            created_at=datetime.datetime.now(),
+        )
+        response = await _charge.add_charge(db, company_data)
+    except ChargeExistException as exc:
+        raise HTTPException(status_code=200, detail=str(exc))
+    return response