Update app.py

app.py

@@ -33,43 +33,60 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
-
-import requests
-from bs4 import BeautifulSoup
-
+import baostock as bs
+import pandas as pd
 @tool
-def get_stock_data(stock_code: str) -> str:
-    """A tool that fetches the latest stock data for a given stock code from East Money website.
+def get_stock_data_baostock(stock_code: str) -> str:
+    """Fetch the latest stock data for a given stock code using Baostock.
     Args:
-        stock_code: A string representing the stock code (e.g., 'SH600000' for Shanghai Pudong Development Bank).
+        stock_code: A string representing the stock code (e.g., 'sh.600941' for China Mobile).
     """
     try:
-        # Construct the URL for the stock data
-        url = f"https://quote.eastmoney.com/{stock_code}.html"
-        
-        # Send a GET request to the URL
-        response = requests.get(url)
-        response.raise_for_status()  # Raise an error for bad status codes
-        
-        # Parse the HTML content using BeautifulSoup
-        soup = BeautifulSoup(response.text, 'html.parser')
-        
-        # Extract the stock name
-        stock_name = soup.find('div', class_='stock-name').text.strip()
-        
-        # Extract the current price
-        current_price = soup.find('div', class_='stock-current').text.strip()
-        
-        # Extract the change in price and percentage
-        price_change = soup.find('div', class_='stock-change').text.strip()
-        change_percentage = soup.find('div', class_='stock-change-percent').text.strip()
-        
+        # Log in to the Baostock system
+        lg = bs.login()
+        if lg.error_code != '0':
+            return f"Failed to log in to Baostock: {lg.error_msg}"
+
+        # Fetch daily K-line data for the stock (defaults to the most recent trading day)
+        rs = bs.query_history_k_data_plus(
+            code=stock_code,
+            fields="date,code,open,high,low,close,volume",  # Fields to retrieve
+            frequency="d",  # Daily K-line
+            adjustflag="2"  # Adjustment type: 2 for forward adjustment
+        )
+        if rs.error_code != '0':
+            return f"Failed to fetch stock data: {rs.error_msg}"
+
+        # Convert the data to a DataFrame
+        data_list = []
+        while (rs.error_code == '0') & rs.next():
+            data_list.append(rs.get_row_data())
+        df = pd.DataFrame(data_list, columns=rs.fields)
+
+        # Log out of the Baostock system
+        bs.logout()
+
+        # Check if the data is empty
+        if df.empty:
+            return f"No data found for stock code '{stock_code}'."
+
+        # Extract the latest trading day's data
+        latest_data = df.iloc[-1]  # Get the last row of data (most recent trading day)
+        date = latest_data['date']  # Date
+        stock_name = latest_data['code']  # Stock code
+        close_price = latest_data['close']  # Closing price
+        volume = latest_data['volume']  # Trading volume
+
         # Construct the result string
-        result = f"Stock: {stock_name}\nCurrent Price: {current_price}\nChange: {price_change} ({change_percentage})"
-        
+        result = f"Stock Code: {stock_name}\nDate: {date}\nClosing Price: {close_price}\nVolume: {volume}"
         return result
+
     except Exception as e:
-        return f"Error fetching stock data for code '{stock_code}': {str(e)}"
+        return f"An error occurred while fetching stock data: {str(e)}"
+
+# Example call
+print(get_stock_data_baostock('sh.600941'))
+
 
 
 final_answer = FinalAnswerTool()