fix requests warnings and add nfts api schemas

Signed-off-by: wiseaidev <[email protected]>

Author: wiseaidev

app/nfts/router.py

@@ -20,8 +20,12 @@
 from typing import (
     Any,
     Dict,
+    Union,
 )
 
+from app.auth import (
+    schemas as auth_schemas,
+)
 from app.config import (
     settings,
 )
@@ -44,41 +48,46 @@
 @router.post(
     "/nft/mint-offer",
     name="nft:mint-nft-with-offer",
-    # response_model=Any,
-    # responses={
-    #     201: {
-    #         "model": Any,
-    #         "description": "A response object that contains info about"
-    #         " a wallet.",
-    #     },
-    # },
+    response_model=auth_schemas.ResponseSchema,
+    responses={
+        200: {
+            "model": auth_schemas.ResponseSchema,
+            "description": "A response object that indicates an nft"
+            " has been minted successfully!",
+        },
+    },
 )
 async def mint_nft_token_with_offer(
     nft_info: nfts_schemas.NFTObjectSchema,
     current_wallet: Any = Depends(jwt.get_current_active_wallet),
     session: AIOSession = Depends(dependencies.get_db_transactional_session),
 ) -> Dict[str, Any]:
     """
-    Generate a faucet wallet.
+    mint an nft token and create a sell offer.
     """
     meta_data = f"{nft_info.picture},{nft_info.title},{nft_info.price}"
-    results = await nfts_crud.mint_nft_token(
+    await nfts_crud.mint_nft_token(
         current_wallet.classic_address, meta_data, session, True
     )
-    return results
+    return {"status_code": 200, "message": "NFT minted successfully!"}
 
 
 @router.post(
     "/nft/upload-image",
     name="nft:upload-image",
-    # response_model=Any,
-    # responses={
-    #     201: {
-    #         "model": Any,
-    #         "description": "A response object that contains info about"
-    #         " a wallet.",
-    #     },
-    # },
+    response_model=Union[
+        auth_schemas.ResponseSchema, nfts_schemas.UploadImageResponseSchema
+    ],
+    responses={
+        400: {
+            "model": auth_schemas.ResponseSchema,
+            "description": "A response object that indicates something went wrong!",
+        },
+        200: {
+            "model": nfts_schemas.UploadImageResponseSchema,
+            "description": "A response object that contains an ipfs url of the image.",
+        },
+    },
 )
 async def upload_nft_image(
     file: UploadFile = File(...),
@@ -108,14 +117,14 @@ async def upload_nft_image(
 @router.post(
     "/nft/upload-mint-nft",
     name="nft:upload-mint-nft",
-    # response_model=Any,
-    # responses={
-    #     201: {
-    #         "model": Any,
-    #         "description": "A response object that contains info about"
-    #         " a wallet.",
-    #     },
-    # },
+    response_model=auth_schemas.ResponseSchema,
+    responses={
+        200: {
+            "model": auth_schemas.ResponseSchema,
+            "description": "A response object that indicates an nft"
+            " has been minted successfully!",
+        },
+    },
 )
 async def upload_nft_image_and_mint_nft(
     nft_info: nfts_schemas.NFTBase64ObjectSchema,
@@ -144,14 +153,14 @@ async def upload_nft_image_and_mint_nft(
 @router.get(
     "/nft/get-wallet-nfts",
     name="nft:get-wallet-nfts",
-    # response_model=Any,
-    # responses={
-    #     201: {
-    #         "model": Any,
-    #         "description": "A response object that contains info about"
-    #         " a wallet.",
-    #     },
-    # },
+    response_model=nfts_schemas.ResponseSchema,
+    responses={
+        200: {
+            "model": Any,
+            "description": "A response object that contains info about"
+            " a given wallet nfts.",
+        },
+    },
 )
 async def fetch_all_nfts(
     current_wallet: Any = Depends(jwt.get_current_active_wallet),
@@ -166,14 +175,14 @@ async def fetch_all_nfts(
 @router.get(
     "/nft/get-all",
     name="nft:get-wallets-nfts",
-    # response_model=Any,
-    # responses={
-    #     201: {
-    #         "model": Any,
-    #         "description": "A response object that contains info about"
-    #         " a wallet.",
-    #     },
-    # },
+    response_model=nfts_schemas.ResponseSchema,
+    responses={
+        200: {
+            "model": nfts_schemas.ResponseSchema,
+            "description": "A response object that contains info about"
+            " a all nfts available on the marketplace.",
+        },
+    },
 )
 async def fetch_all_wallets_nfts(
     session: AIOSession = Depends(dependencies.get_db_transactional_session),

app/nfts/schemas.py

@@ -4,6 +4,12 @@
     BaseModel,
     Field,
 )
+from typing import (
+    Any,
+    Dict,
+    List,
+    Optional,
+)
 
 
 class NFTObjectSchema(BaseModel):
@@ -25,3 +31,23 @@ class NFTBase64ObjectSchema(BaseModel):
     author_avatar: str = Field(..., example="Owner profile picture.")
     title: str = Field(..., example="Your NFT title.")
     price: str = Field(..., example="Your NFT price in XRP.")
+
+
+class ResponseSchema(BaseModel):
+    """
+    A Pydantic class that defines the wallet schema for fetching wallet info.
+    """
+
+    status_code: int = Field(..., example=200)
+    results: List[Optional[Dict[str, Any]]] = Field(
+        ..., example="Owner profile picture."
+    )
+
+
+class UploadImageResponseSchema(BaseModel):
+    """
+    A Pydantic class that defines the wallet schema for fetching wallet info.
+    """
+
+    status_code: int = Field(..., example=200)
+    url: str = Field(..., example="IPFS URL of the image.")

app/wallets/crud.py

@@ -120,7 +120,7 @@ async def get_wallet_info(classic_address: str, session: AIOSession) -> Dict[str
     for nft_token in account_nfts:
         meta_data_url = hex_to_str(nft_token["URI"])
         if "png" not in meta_data_url:
-            meta_data = requests.get(url=meta_data_url, verify=False, timeout=30).text
+            meta_data = requests.get(url=meta_data_url, timeout=30).text
             meta_data_array = meta_data.split(",")
             if len(meta_data_array) == 2:
                 first_name, bio = meta_data_array
@@ -166,7 +166,7 @@ async def update_wallet_info(
         meta_data_url = hex_to_str(nft_token["URI"])
         # burn it, then mint a new one.
         if "png" not in meta_data_url:
-            meta_data = requests.get(url=meta_data_url, verify=False, timeout=30).text
+            meta_data = requests.get(url=meta_data_url, timeout=30).text
             meta_data_array = meta_data.split(",")
             if len(meta_data_array) == 2:
                 await nfts_crud.burn_nft_token(
@@ -246,9 +246,7 @@ async def get_all_wallet_info(session: AIOSession) -> List[Any]:
         for nft_token in account_nfts:
             meta_data_url = hex_to_str(nft_token["URI"])
             if "png" not in meta_data_url:
-                meta_data = requests.get(
-                    url=meta_data_url, verify=False, timeout=30
-                ).text
+                meta_data = requests.get(url=meta_data_url, timeout=30).text
                 meta_data_array = meta_data.split(",")
                 if len(meta_data_array) == 2:
                     first_name, bio = meta_data_array