added pdf and png downloads for abha card

ohcnetwork · Jul 5, 2023 · 5205e18 · 5205e18
1 parent 5af1584
commit 5205e18
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 4 deletions.
diff --git a/care/abdm/api/viewsets/healthid.py b/care/abdm/api/viewsets/healthid.py
@@ -300,10 +300,17 @@ def get_abha_card(self, request):
         if not patient.abha_number:
             raise ValidationError({"abha": "Patient hasn't linked thier abha"})
 
-        response = HealthIdGateway().get_abha_card_png(
-            {"refreshToken": patient.abha_number.refresh_token}
-        )
+        if data["type"] == "png":
+            response = HealthIdGateway().get_abha_card_png(
+                {"refreshToken": patient.abha_number.refresh_token}
+            )
+            return Response(response, status=status.HTTP_200_OK)
+
+        response = HealthIdGateway().get_abha_card_pdf(
+                {"refreshToken": patient.abha_number.refresh_token}
+            )
         return Response(response, status=status.HTTP_200_OK)
+
 
     @swagger_auto_schema(
         # /v1/registration/aadhaar/searchByHealthId

diff --git a/care/abdm/utils/api_call.py b/care/abdm/utils/api_call.py
@@ -291,7 +291,16 @@ def get_abha_card_png(self, data):
         path = "/v1/account/getPngCard"
         access_token = self.generate_access_token(data)
         response = self.api.get(path, {}, access_token)
-        return response
+
+        return b64encode(response.content)
+
+    def get_abha_card_pdf(self, data):
+        path = "/v1/account/getCard"
+        access_token = self.generate_access_token(data)
+        response = self.api.get(path, {}, access_token)
+
+        return b64encode(response.content)
+
 
     # /v1/account/qrCode
     def get_qr_code(self, data, auth):