added status for the demon record usage call

Author: pratik-vii

metering/alembic/versions/9ecbd29ee907_status_added_in_calls_tables.py

@@ -0,0 +1,28 @@
+"""status added in calls tables
+
+Revision ID: 9ecbd29ee907
+Revises: eeec29a1af7b
+Create Date: 2019-08-12 15:13:51.507698
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '9ecbd29ee907'
+down_revision = 'eeec29a1af7b'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.add_column('usage_table', sa.Column('status', sa.VARCHAR(length=225), nullable=False))
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    op.drop_column('usage_table', 'status')
+    # ### end Alembic commands ###

metering/models.py

@@ -37,6 +37,7 @@ class UsageModel(Base):
     user_org_group_id = Column('user_org_group_id', Integer, ForeignKey(
         'user_org_group.id'), nullable=False)
     usage_type = Column('usage_type', VARCHAR(225), nullable=False)
+    status = Column('status', VARCHAR(225), nullable=False)
     usage_value = Column('usage_value', Integer, nullable=False)
     start_time = Column('start_time', TIMESTAMP(timezone=True))
     end_time = Column('end_time', TIMESTAMP(timezone=True))

metering/repository/usage_repository.py

@@ -10,7 +10,8 @@ def get_total_calls(self, user_name, org_id, service_id):
         query_data = session.query(func.count(UsageModel.id).label('total_calls')).join(UserOrgGroupModel) \
             .filter(UserOrgGroupModel.user_name == user_name)\
             .filter(UserOrgGroupModel.org_id == org_id) \
-            .filter(UserOrgGroupModel.service_id == service_id).all()
+            .filter(UserOrgGroupModel.service_id == service_id)\
+            .filter(UsageModel.status == 'success').all()
         session.commit()
         session.flush()
         return query_data[0].total_calls

metering/storage.py

@@ -50,6 +50,7 @@ def add_usage_data(self, usage_details):
         user_org_group_id = user_org_group_repo_data.id
         usage_record = UsageModel(
             user_org_group_id=user_org_group_id,
+            status=usage_details['status'],
             usage_type=usage_details['usage_type'],
             usage_value=usage_details['usage_value'],
             start_time=usage_details['start_time'],