38 lines
931 B
Python
38 lines
931 B
Python
from markdown import markdown
|
|
import frontmatter
|
|
|
|
import os
|
|
|
|
class Card(object):
|
|
def __init__(self, card_file: str):
|
|
metadata = frontmatter.load(card_file).metadata
|
|
self.service_type = metadata['type']
|
|
self.title = metadata['title']
|
|
self.images = self._get_image_list(
|
|
metadata['image-dir'], metadata['images'])
|
|
self.price = metadata['price']
|
|
self.included = metadata['included']
|
|
|
|
def _get_image_list(self, base_dir: str, images: list):
|
|
image_list = list()
|
|
for img in images:
|
|
image_list.append(os.path.join(base_dir, img))
|
|
|
|
return image_list
|
|
|
|
def get_title(self):
|
|
return self.title
|
|
|
|
def get_price(self):
|
|
return self.price
|
|
|
|
def get_included(self):
|
|
return self.included
|
|
|
|
def get_images(self):
|
|
return self.images
|
|
|
|
def get_default_image(self):
|
|
return self.images[1]
|
|
|