Как использовать функцию python в django rest api?

Я создаю функции анализа данных с помощью Python, и каким-то образом мне нужно сделать restAPI этих функций с помощью Django. Поскольку я новичок в мире API и rest API, я изучил Django за последнюю неделю, но все еще не знал, как преобразовать эти функции python в Django-подобный код. Вот пример кода функции, над которым я работал:

def filter_dataset(data, period):  # filter dataset for XYZ Classification Function
   # copy the data first then change Date into datetime format
   data_copy = data.copy()
   data_copy['Date'] = pd.to_datetime(data_copy['Date'], dayfirst=True)
   data_copy['month'] = data_copy['Date'].dt.month

   # get the last 3 months data
   months = data_copy.month.unique()
   last_3_months = months[-3:]

   # filtering the data based on period choice
   if period == 'Last 3 Months':
       data_period = data_copy.loc[data_copy['month'].apply(
           lambda x: x in last_3_months)]
   else:
       data_period = data_copy

   data_period.drop(columns='month', inplace=True)

   return data_period

Приведенная выше функция используется для фильтрации данных на основе выбранного периода месяцев. Я хочу создать restAPI с использованием django, который будет работать подобным образом: Допустим, мы получаем данные в формате json:

[
  {
   'id': 1,
   'product_code': 'Product_001',
   'date' : '2017-05-29,
   'demand': 200
  },
  {
   'id': 2,
   'product_code': 'Product_002',
   'date' : '2017-05-29,
   'demand': 150
  },
]

И он вернет данные следующим образом:

[
  {
   'id': 1,
   'product_code': 'Product_001',
   'date' : '2017-05-29,
   'demand': 200,
   'class': 'A'
  },
  {
   'id': 2,
   'product_code': 'Product_002',
   'date' : '2017-05-29,
   'demand': 150,
   'class': 'C'
  },
]

Новый класс атрибутов приобретается через некоторые функции процессов в Python, которые я создал ранее. Есть ли какой-нибудь способ сделать это? Я так запутался, пожалуйста, помогите.

Вернуться на верх