日期格式化问题：函数间传递日期数据

正如摘要中所述，本文旨在解决在Python函数间传递日期数据时，由于日期格式不匹配导致的 AttributeError 错误。

在使用Python进行数据处理时，经常需要在不同的函数之间传递日期数据。如果日期格式不一致，可能会导致程序出错。本教程将针对一个常见的错误场景，提供详细的解决方案。

问题描述

假设我们有两个函数：oi_data 和 fut_data。oi_data 函数从API获取数据，并提取出到期日信息，将其格式化为特定的字符串格式。fut_data 函数则需要使用这个到期日信息作为参数，从另一个数据源获取数据。

from jugaad_data.nse import derivatives_df
from datetime import timedelta, date
from datetime import datetime
import pandas as pd
import requests
import json

def oi_data(Symbols):
  baseurl = "https://www.nseindia.com/"
  url = f'https://www.nseindia.com/api/option-chain-equities'
  headers = {'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, ''like Gecko) ''Chrome/80.0.3987.149 Safari/537.36','accept-language': 'en,gu;q=0.9,hi;q=0.8', 'accept-encoding': 'gzip, deflate, br'}
  session = requests.Session()
  request = session.get(baseurl, headers=headers, timeout=30)
  cookies = dict(request.cookies)
  params=[('symbol', Symbols)]
  res = session.get(url, headers=headers, params=params, cookies=cookies, timeout=30)
  res_text = res.text
  data = json.loads(res_text)
  exp_dt1 = data["records"]["expiryDates"][0]
  datetime_obj = datetime.strptime(exp_dt1, '%d-%b-%Y')
  exp_dt = datetime_obj.strftime("%Y, %m, %d") #问题所在
  print(exp_dt)
  return exp_dt

def fut_data(Symbols):
    ticker_df = derivatives_df(symbol=Symbols, from_date = date.today() - timedelta(days = 30), to_date= date.today(),expiry_date=date(2023,11,30), instrument_type="FUTSTK")
    fut_df = ticker_df
    print(fut_df)
    return fut_df

Symbols = ['MARUTI']
for symbol in Symbols:
    exp_dt = oi_data(symbol)
    fut_df = fut_data(symbol)

上述代码中，oi_data 函数将日期格式化为字符串 "%Y, %m, %d"，然后将其传递给 fut_data 函数。fut_data 函数期望接收的是 datetime.date 对象，因此会报错：AttributeError: 'str' object has no attribute 'strftime'。

解决方案

问题的根源在于日期格式不匹配。 derivatives_df 函数期望 expiry_date 参数是一个 datetime.date 对象，而不是字符串。我们需要修改 oi_data 函数，使其返回 datetime.date 对象。

修改 oi_data 函数的代码如下：

def oi_data(Symbols):
  baseurl = "https://www.nseindia.com/"
  url = f'https://www.nseindia.com/api/option-chain-equities'
  headers = {'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, ''like Gecko) ''Chrome/80.0.3987.149 Safari/537.36','accept-language': 'en,gu;q=0.9,hi;q=0.8', 'accept-encoding': 'gzip, deflate, br'}
  session = requests.Session()
  request = session.get(baseurl, headers=headers, timeout=30)
  cookies = dict(request.cookies)
  params=[('symbol', Symbols)]
  res = session.get(url, headers=headers, params=params, cookies=cookies, timeout=30)
  res_text = res.text
  data = json.loads(res_text)
  exp_dt1 = data["records"]["expiryDates"][0]
  datetime_obj = datetime.strptime(exp_dt1, '%d-%b-%Y')
  exp_dt = datetime_obj.date() # 修改这里
  print(exp_dt)
  return exp_dt

通过将 exp_dt = datetime_obj.strftime("%Y, %m, %d") 替换为 exp_dt = datetime_obj.date()，我们将 datetime 对象转换为 date 对象。

同时，需要修改调用 fut_data 函数的方式，将 exp_dt 作为参数传递进去：

def fut_data(Symbols, exp_dt):
    ticker_df = derivatives_df(symbol=Symbols, from_date = date.today() - timedelta(days = 30), to_date= date.today(),expiry_date=exp_dt, instrument_type="FUTSTK")
    fut_df = ticker_df
    print(fut_df)
    return fut_df

Symbols = ['MARUTI']
for symbol in Symbols:
    exp_dt = oi_data(symbol)
    fut_df = fut_data(symbol, exp_dt) # 修改这里

完整代码

from jugaad_data.nse import derivatives_df
from datetime import timedelta, date
from datetime import datetime
import pandas as pd
import requests
import json

def oi_data(Symbols):
  baseurl = "https://www.nseindia.com/"
  url = f'https://www.nseindia.com/api/option-chain-equities'
  headers = {'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, ''like Gecko) ''Chrome/80.0.3987.149 Safari/537.36','accept-language': 'en,gu;q=0.9,hi;q=0.8', 'accept-encoding': 'gzip, deflate, br'}
  session = requests.Session()
  request = session.get(baseurl, headers=headers, timeout=30)
  cookies = dict(request.cookies)
  params=[('symbol', Symbols)]
  res = session.get(url, headers=headers, params=params, cookies=cookies, timeout=30)
  res_text = res.text
  data = json.loads(res_text)
  exp_dt1 = data["records"]["expiryDates"][0]
  datetime_obj = datetime.strptime(exp_dt1, '%d-%b-%Y')
  exp_dt = datetime_obj.date()
  print(exp_dt)
  return exp_dt

def fut_data(Symbols, exp_dt):
    ticker_df = derivatives_df(symbol=Symbols, from_date = date.today() - timedelta(days = 30), to_date= date.today(),expiry_date=exp_dt, instrument_type="FUTSTK")
    fut_df = ticker_df
    print(fut_df)
    return fut_df

Symbols = ['MARUTI']
for symbol in Symbols:
    exp_dt = oi_data(symbol)
    fut_df = fut_data(symbol, exp_dt)

总结与注意事项

• 在函数间传递日期数据时，务必确保日期格式一致。
• 如果函数需要 datetime.date 对象，则不要传递字符串格式的日期。
• 可以使用 datetime.strptime() 将字符串转换为 datetime 对象，然后使用 .date() 方法提取日期部分。
• 在函数定义中明确指定参数类型，可以帮助避免类型错误。

通过以上步骤，我们可以解决日期格式不匹配导致的 AttributeError 错误，使程序能够正确地处理日期数据。希望本教程能够帮助您更好地理解和应用Python中的日期处理。