Merge pull request #234 from UnitapApp/refactor/tokentap-valid-chains

Get valid chains

tokenTap/constants.py

@@ -0,0 +1,5 @@
+CONTRACT_ADDRESSES = {
+    "100": "0xB67ec856346b22e4BDA2ab2B53d70D61a2014358",
+    "74": "0x3d1b1D14333D56472f7ADf044598d1605C67609f",
+    "10": "0x54a839FF128DC1891a03d7a81724bD5D51A5902b",
+}

tokenTap/urls.py

@@ -1,5 +1,14 @@
 from django.urls import path
-from tokenTap.views import *
+
+from tokenTap.views import (
+    GetTokenDistributionConstraintsView,
+    TokenDistributionClaimListView,
+    TokenDistributionClaimRetrieveView,
+    TokenDistributionClaimStatusUpdateView,
+    TokenDistributionClaimView,
+    TokenDistributionListView,
+    ValidChainsView,
+)
 
 urlpatterns = [
     path(
@@ -28,4 +37,9 @@
         GetTokenDistributionConstraintsView.as_view(),
         name="get-token-distribution-constraints",
     ),
+    path(
+        "get-valid-chains/",
+        ValidChainsView.as_view(),
+        name="get-valid-chains",
+    ),
 ]

tokenTap/views.py

@@ -13,7 +13,8 @@
 from rest_framework.views import APIView
 
 from core.constraints import ConstraintVerification, get_constraint
-from core.models import NetworkTypes
+from core.models import Chain, NetworkTypes
+from core.serializers import ChainSerializer
 from faucet.models import Chain as FaucetChain
 from faucet.models import ClaimReceipt
 from tokenTap.models import TokenDistribution, TokenDistributionClaim
@@ -25,6 +26,7 @@
     TokenDistributionSerializer,
 )
 
+from .constants import CONTRACT_ADDRESSES
 from .helpers import (
     create_uint32_random_nonce,
     has_weekly_credit_left,
@@ -253,3 +255,23 @@ def get_object(self):
         return TokenDistributionClaim.objects.get(
             pk=self.kwargs["pk"], user_profile=user_profile
         )
+
+
+class ValidChainsView(ListAPIView):
+    queryset = Chain.objects.filter(
+        chain_id__in=list(CONTRACT_ADDRESSES.keys())
+    ).order_by("pk")
+    serializer_class = ChainSerializer
+
+    def get(self, request):
+        queryset = self.get_queryset()
+        serializer = ChainSerializer(queryset, many=True)
+        response = []
+        for chain in serializer.data:
+            response.append(
+                {
+                    **chain,
+                    "tokentap_contract_address": CONTRACT_ADDRESSES[chain["chain_id"]],
+                }
+            )
+        return Response({"success": True, "data": response})