mirror of
https://github.com/PiBrewing/craftbeerpi4.git
synced 2024-11-10 01:17:42 +01:00
85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
|
|
||
|
|
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)
|
||
|
|
self.cbpi.notify("{} saved".format(data["basic"].get("name")))
|
||
|
|
|
||
|
|
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)
|
||
|
|
self.cbpi.notify("{} delted".format(name))
|
||
|
|
|
||
|
|
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
|