Upload 3 files

Model.py

@@ -0,0 +1,144 @@
+from pydantic import BaseModel, ConfigDict, Field
+from typing import Any, List
+
+
+## Search Model
+class Stock(BaseModel):
+    ID: str
+    title: str
+    NSE_Symbol: str
+    contract_id: str
+
+search_response_example = {
+        "ID": "zomato-ltd",
+        "title": "Zomato Ltd.",
+        "NSE_Symbol": "ZOMATO",
+        "contract_id": "GSTK543320"
+}
+
+
+## LTP Model
+class LTP(BaseModel):
+    ltp: float
+
+ltp_response_example = {
+   "ltp" : 209.50
+}
+
+
+## Historical Data
+class HISTORICAL(BaseModel):
+    data: List[dict]
+
+historical_response_example = {
+  "data": [
+    {
+      "Date": "2024-06-24T09:15:00+05:30",
+      "Open": 193.9,
+      "High": 194.84,
+      "Low": 192.02,
+      "Close": 194.15,
+      "Volume": 2767062
+    },
+    {
+      "Date": "2024-06-24T09:30:00+05:30",
+      "Open": 194.14,
+      "High": 194.2,
+      "Low": 192.8,
+      "Close": 193.97,
+      "Volume": 4066133
+    }
+  ]
+}
+
+
+## Historical Data
+class CHAIN(BaseModel):
+    expiry: str
+    data: List[dict]
+
+chain_response_example = {
+  "expiry": "BANKNIFTY24703",
+  "data": [
+    {
+      "Symbol_CE": "BANKNIFTY2470351600CE",
+      "OI_CALL": 3347,
+      "CALL": 676.8,
+      "strikePrice": 51600,
+      "PUT": 43.3,
+      "OI_PUT": 61530,
+      "Symbol_PE": "BANKNIFTY2470351600PE"
+    },
+    {
+      "Symbol_CE": "BANKNIFTY2470351700CE",
+      "OI_CALL": 4870,
+      "CALL": 585.55,
+      "strikePrice": 51700,
+      "PUT": 56.45,
+      "OI_PUT": 66516,
+      "Symbol_PE": "BANKNIFTY2470351700PE"
+    },
+  ]
+}
+
+
+## News Data
+class NEWS(BaseModel):
+    data: List[dict]
+
+news_response_example = {
+  "data": [
+    {
+      "title": "Karnataka's new gig worker draft bill: Why Morgan Stanley is 'overweight' on Zomato",
+      "summary": "NSE Morgan Stanley has maintained an overweight rating on Zomato with a target price of ₹235 per share.",
+      "url": "https://www.cnbctv18.com/market/karnataka-new-draft-gig-worker-bill-zomato-shares-price-stock-morgan-stanley-target-overweight-19436697.htm",
+      "pubDate": "2024-07-02T09:38:15",
+      "source": "CNBC TV18",
+      "companyName": "Zomato",
+      "symbol": "ZOMATO",
+      "blogUrl": "https://www.cnbctv18.com/market/karnataka-new-draft-gig-worker-bill-zomato-shares-price-stock-morgan-stanley-target-overweight-19436697.htm",
+      "topics": "Regulatory and Legal"
+    },
+    {
+      "title": "Zomato Shares Touch An All-Time High After ESOP Plan Gets Shareholders’ Nod",
+      "summary": "Shares of Zomato reached an all-time high of INR 209.75 apiece during Tuesday’s intraday trading session The price surge in the stock came after Zomato said it has secured approval from its shareholders to grant",
+      "url": "https://inc42.com/buzz/zomato-shares-touch-an-all-time-high-after-esop-plan-gets-shareholders-nod/",
+      "pubDate": "2024-07-02T07:36:51",
+      "source": "Inc42",
+      "companyName": "Zomato",
+      "symbol": "ZOMATO",
+      "blogUrl": "https://inc42.com/buzz/zomato-shares-touch-an-all-time-high-after-esop-plan-gets-shareholders-nod/",
+      "topics": "Financial Results"
+    }
+  ]
+}
+
+
+## Signal Model
+class SIGNAL(BaseModel):
+    data: dict
+
+signal_response_example = {
+  "data": {
+    "Date": "2024-07-02T15:15:00+05:30",
+    "Open": 52164.5,
+    "High": 52200.7,
+    "Low": 52131.2,
+    "Close": 52148.1,
+    "RSI": 42.25281555502232,
+    "Prev_RSI": 43.331830665536984,
+    "Signal": 0,
+    "Option": 52200,
+    "direction": "",
+    "symbol": "BANKNIFTY2470352200"
+  }
+}
+
+## get all stocks
+class ALL(BaseModel):
+    data: str
+    
+    
+all_response_example = {
+  "data": '''[{\"isin\":\"INE002A01018\",\"growwContractId\":\"GSTK500325\"'''
+}

main.py

@@ -1,13 +1,14 @@
 from fastapi.responses import HTMLResponse
 from fastapi.templating import Jinja2Templates
-from fastapi import FastAPI, Request, HTTPException
+from fastapi import FastAPI, Request, HTTPException, Query, Body
 from fastapi.middleware.cors import CORSMiddleware
 from utils import StockDataFetcher
-
-app = FastAPI(swagger_ui_parameters={"syntaxHighlight.theme": "obsidian"})
+from Model import *
+import json
 
 fetcher = StockDataFetcher()
 
+app = FastAPI(swagger_ui_parameters={"syntaxHighlight.theme": "obsidian"})
 origins = ["*"]
 
 app.add_middleware(
@@ -20,65 +21,100 @@ app.add_middleware(
 
 templates = Jinja2Templates(directory="templates")
 
+# HTML Page
 @app.get("/", response_class=HTMLResponse)
 async def read_root(request: Request):
     return templates.TemplateResponse("hello.html", {"request": request})
 
 
-@app.get('/ltp')
-async def get_data(ticker: str):
+# Latest Price Fetch
+@app.get('/ltp', response_model=LTP, responses={200: {"description": "Successful response", "content": {"application/json": {"example": ltp_response_example}}}})
+async def get_data(ticker: str = Query(..., description="Enter your stock name.",  example='zomato')):
     try:
         response = fetcher.fetch_latest_price(ticker)
-        return {'ltp' :response}
-    except:
-        return {"Timeout" : "Error"}
+        if response:
+            return {'ltp' :response}
+        else:
+            raise HTTPException(status_code=404, detail="ltp price not found")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
 
-@app.get('/historical')
-async def get_stocks_data(ticker: str, intervals: int, days: int):
-    try:
-        response = fetcher.fetch_stock_data(ticker, 15, 10).to_dict(orient="records")
-        return {"data" : response}
-    except:
-        return {"Timeout" : "Error"}
 
-@app.get('/chain')
-async def get_chain(ticker: str):
+# Search Stock Ticker
+@app.get('/search', response_model=Stock, responses={200: {"description": "Successful response", "content": {"application/json": {"example": search_response_example}}}})
+async def stock_search(ticker: str = Query(..., description="Enter your stock name.",  example='zomato')):
+    try:
+        fetcher = StockDataFetcher()
+        response = fetcher.search_entity(ticker)
+        if response:
+            return response 
+        else:
+            raise HTTPException(status_code=404, detail="Stock not found")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+ 
+ 
+# Fetch Historical Data    
+@app.get('/historical', response_model=HISTORICAL, responses={200: {"description": "Successful response", "content": {"application/json": {"example": historical_response_example}}}})
+async def get_stocks_data(ticker: str = Query(..., description="Enter your stock name.",  example='zomato'), intervals: int = Query(..., description="Select your interval 1m, 5m, 15m",  example=5), days: int = Query(..., description="Enter you numbers of day",  example=10)):
+    try:
+        response = fetcher.fetch_stock_data(ticker, intervals, days).to_dict(orient="records")
+        if response:
+            return {"data" : response}
+        else:
+            raise HTTPException(status_code=404, detail="historical data not found.")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+    
+    
+# Fetch Stock Chain
+@app.get('/chain', response_model=CHAIN, responses={200: {"description": "Successful response", "content": {"application/json": {"example": chain_response_example}}}})
+async def get_chain(ticker: str = Query(..., description="Enter your index name from [nifty, nifty-bank]",  example='nifty-bank')):
     try:
         response, exp = fetcher.fetch_option_chain(ticker)
         response = response.to_dict(orient="records")
-        print(response, exp)
-        return {"data" : response}
-    except:
-        return {"Timeout" : "Error"}
+        if response:
+            return {"expiry": exp, "data" : response}
+        else:
+            raise HTTPException(status_code=404, detail="Unable to fetch chain.")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
 
-@app.get('/search')
-async def stock_search(ticker: str):
-    try:
-        response = fetcher.search_entity(ticker)
-        return {"data" : response}
-    except:
-        return {"Timeout" : "Error"}
-
-# @app.get('/news')
-# async def get_stocks_data(ticker: str, page: int, size: int):
-#     try:
-#         response = fetcher.fetch_stock_news(ticker, page=page, size=size).to_dict(orient="records")
-#         return {"data" : response}
-#     except:
-#         return {"Timeout" : "Error"}
-
-# @app.get('/all')
-# async def get_stocks_data():
-#     try:
-#         response = fetcher.fetch_all_stock().to_dict(orient="records")
-#         return {"data" : response}
-#     except:
-#         return {"Timeout" : "Error"}
 
-@app.get('/signal')
-async def get_stock_signal(index: str):
+# Stocks News   
+@app.get('/news', response_model=NEWS, responses={200: {"description": "Successful response", "content": {"application/json": {"example": news_response_example}}}})
+async def get_stocks_data(ticker: str = Query(..., description="Enter your stock name.",  example='zomato'), page: int = Query(..., description="Enter number of pages.",  example=3), size: int = Query(..., description="Enter number of news you want to extract",  example=4)):
+    try:
+        response = fetcher.fetch_stock_news(ticker, page=page, size=size).to_dict(orient="records")
+        if response:
+            return {"data" : response}
+        else:
+            raise HTTPException(status_code=404, detail="Unable to fetch news data.")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+    
+ 
+# Get signal data
+@app.get('/signal', response_model=SIGNAL, responses={200: {"description": "Successful response", "content": {"application/json": {"example": signal_response_example}}}})
+async def get_stock_signal(index: str = Query(..., description="Enter your index name from [nifty, nifty-bank]", example='nifty')):
     try:
         response = fetcher.realtime_signal(index).tail(1).to_dict(orient="records")[0]
-        return {"data" : response}
-    except:
-        return {"Timeout" : "Error"}
+        if response:
+            return {"data" : response}
+        else:
+            raise HTTPException(status_code=404, detail="Unable to fetch signals")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+    
+# Fetch all stocks info
+@app.get('/all', response_model=ALL, responses={200: {"description": "Successful response", "content": {"application/json": {"example": all_response_example}}}})
+async def get_stocks_data():
+    try:
+        response = fetcher.fetch_all_stock().to_json(orient="records")
+        if response:
+            return {"data" : response}
+        else:
+            raise HTTPException(status_code=404, detail="Unable to fetch all stocks")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))

run.sh

@@ -0,0 +1 @@
+uvicorn main:app --host 0.0.0.0 --port 8000 --reload