craftbeerpi4-pione/cbpi/controller/recipe_controller.py

import logging
import os.path
from os import listdir
from os.path import isfile, join
import json
import shortuuid
import yaml
from ..api.step import StepMove, StepResult, StepState

import re

class RecipeController:


    def __init__(self, cbpi):
        self.cbpi = cbpi
        self.logger = logging.getLogger(__name__)
    
    def urlify(self, s):

        # Remove all non-word characters (everything except numbers and letters)
        s = re.sub(r"[^\w\s]", '', s)

        # Replace all runs of whitespace with a single dash
        s = re.sub(r"\s+", '-', s)

        return s

    async def create(self, name):
        id = shortuuid.uuid()
        path = os.path.join(".", 'config', "recipes", "{}.yaml".format(id))
        data = dict(basic=dict(name=name, author=self.cbpi.config.get("AUTHOR", "John Doe")), steps=[])
        with open(path, "w") as file:
            yaml.dump(data, file)
        return id

    async def save(self, name, data):
        path = os.path.join(".", 'config', "recipes", "{}.yaml".format(name))
        with open(path, "w") as file:
            yaml.dump(data, file, indent=4, sort_keys=True)
        
        
    async def get_recipes(self):
        path = os.path.join(".", 'config', "recipes")
        onlyfiles = [os.path.splitext(f)[0] for f in listdir(path) if isfile(join(path, f)) and f.endswith(".yaml")]

        result = []
        for filename in onlyfiles:
            recipe_path = os.path.join(".", 'config', "recipes", "%s.yaml" % filename)
            with open(recipe_path) as file:
                data = yaml.load(file, Loader=yaml.FullLoader)
                dataset = data["basic"]
                dataset["file"] = filename
                result.append(dataset)
        return result
    
    async def get_by_name(self, name):
        
        recipe_path = os.path.join(".", 'config', "recipes", "%s.yaml" % name)
        with open(recipe_path) as file:
            return  yaml.load(file, Loader=yaml.FullLoader)

           
    async def remove(self, name):
        path = os.path.join(".", 'config', "recipes", "{}.yaml".format(name))
        os.remove(path)
        

    async def brew(self, name):

        recipe_path = os.path.join(".", 'config', "recipes", "%s.yaml" % name)
        with open(recipe_path) as file:
            data = yaml.load(file, Loader=yaml.FullLoader)
            await self.cbpi.step.load_recipe(data)

    async def clone(self, id, new_name):
        recipe_path = os.path.join(".", 'config', "recipes", "%s.yaml" % id)
        with open(recipe_path) as file:
            data = yaml.load(file, Loader=yaml.FullLoader)
            data["basic"]["name"] = new_name
            new_id = shortuuid.uuid()
            await self.save(new_id, data)

            return new_id