Доступ к методам класса в шаблонах Jinja2

Я использую приложение Flask в бэкэнд, которое должно отображать список кодов SKU (Stock Keeping Unit) в интерфейсе, используя цикл внутри шаблона Jinja2. Класс SKU выглядит следующим образом:

class SKU:
"""Class to hold SKU data returned from the database."""

def __init__(self, sku, scanned_at, actual_count, expected_count):
    """Initialize the SKU class with relevant attributes."""
    self.sku = str(sku)
    self.scanned_at = str(scanned_at)
    self.actual_count = int(actual_count)
    self.expected_count = int(expected_count)

def get_progress(self):
    """Get the SKU production progress as a percentage."""
    return ((self.actual_count / self.expected_count) *
            100 if self.expected_count != 0 else 0)

У меня есть метод get_all_skus_today() который возвращает все строки в базе данных для сегодняшней даты, как список объектов SKU . Я хочу сделать это, когда кто-то /skus со следующим маршрутом:

@app.route("/skus")
def skus():
    """Get all SKUs for the day and render the skus.html template."""
    skus = get_all_skus_today()
    return render_template("skus.html", skus=skus)

Проблема в том, что я хочу показать значение прогресса, то есть возвращение функции get_progress() , которая не является атрибутом Class, а метод. Я хочу сделать что-то вроде этого:

{% for sku_row in skus %}
    {{ sku_row.sku }}
    {{ sku_row.get_progress }}
{% endfor %}

Но это не работает. Я хочу, чтобы вам не пришлось перебирать список объектов SKU и преобразовывать их в кортеж, чтобы затем перейти к функции render_template (это то, что я делал раньше).

Любая помощь очень ценится - дайте мне знать, если вам нужно какое-либо уточнение.

python python-3.x flask jinja2

Всего 1 ответ

Вы можете создать дополнительный класс для загрузки и обработки информации из базы данных и создания списка объектов Sku :

import sqlite3
class _Sku:
   def __init__(self, row):
     self.__dict__ = dict(zip(row, ['_sku', 'scanned_at', 'actual_count', 'expected_count']))
   @property
   def sku(self):
      return str(self._sku)
   @property
   def get_progress(self):
     return ((int(self.actual_count) / int(self.expected_count)) *
        100 if int(self.expected_count) != 0 else 0)

class Sku:
  def __init__(self, _listing):
    self.all_vals = [_Sku(i) for i in _listing]
  def __iter__(self):
    yield from self.all_vals
  @classmethod
  def create_skus(cls, _filename='somefilename.db'):
    #if so desired, you can replace query below with your own
    return cls(sqlite3.connect(_filename).cursor().execute("SELECT scanned, actual, expected FROM skus"))

Затем в приложении:

@app.route("/skus")
def skus():
  """Get all SKUs for the day and render the skus.html template."""
   return render_template("skus.html", skus=Sku.create_skus())

Теперь приведенный выше код позволит использовать ваши оригинальные шаблоны:

{% for sku_row in skus %}
   {{ sku_row.sku }}
   {{ sku_row.get_progress }}
{% endfor %}