import argparse import datetime import logging import subprocess import sys import re import requests import yaml from cbpi.utils.utils import load_config from zipfile import ZipFile from cbpi.craftbeerpi import CraftBeerPi import os import pathlib import shutil import yaml import click from jinja2 import Template MAIN_DIR = os.sep.join(os.path.abspath(__file__).split(os.sep)[:-1]) def create_config_file(): if os.path.exists(os.path.join(MAIN_DIR, 'config', "config.yaml")) is False: srcfile = os.path.join(MAIN_DIR, "config", "config.yaml") destfile = os.path.join(MAIN_DIR, 'config') shutil.copy(srcfile, destfile) print("Config Folder created") if os.path.exists(os.path.join(MAIN_DIR, 'config', "actor.json")) is False: srcfile = os.path.join(MAIN_DIR, "config", "actor.json") destfile = os.path.join(MAIN_DIR, 'config') shutil.copy(srcfile, destfile) if os.path.exists(os.path.join(MAIN_DIR, 'config', "sensor.json")) is False: srcfile = os.path.join(MAIN_DIR, "config", "sensor.json") destfile = os.path.join(MAIN_DIR, 'config') shutil.copy(srcfile, destfile) if os.path.exists(os.path.join(MAIN_DIR, 'config', "kettle.json")) is False: srcfile = os.path.join(MAIN_DIR, "config", "kettle.json") destfile = os.path.join(MAIN_DIR, 'config') shutil.copy(srcfile, destfile) if os.path.exists(os.path.join(MAIN_DIR, 'config', "step_data.json")) is False: srcfile = os.path.join(MAIN_DIR, "config", "step_data.json") destfile = os.path.join(MAIN_DIR, 'config') shutil.copy(srcfile, destfile) if os.path.exists(os.path.join(MAIN_DIR, 'config', "dashboard", "cbpi_dashboard_1.json")) is False: srcfile = os.path.join(MAIN_DIR, "config", "dashboard", "cbpi_dashboard_1.json") destfile = os.path.join(MAIN_DIR, "config", "dashboard") shutil.copy(srcfile, destfile) def create_home_folder_structure(): pathlib.Path(os.path.join(MAIN_DIR, 'logs/sensors')).mkdir(parents=True, exist_ok=True) pathlib.Path(os.path.join(MAIN_DIR, 'config')).mkdir(parents=True, exist_ok=True) pathlib.Path(os.path.join(MAIN_DIR, 'config/dashboard')).mkdir(parents=True, exist_ok=True) pathlib.Path(os.path.join(MAIN_DIR, 'config/dashboard/widgets')).mkdir(parents=True, exist_ok=True) print("Folder created") def copy_splash(): srcfile = os.path.join(MAIN_DIR, "config", "splash.png") destfile = os.path.join(MAIN_DIR, 'config') shutil.copy(srcfile, destfile) print("Splash Srceen created") def clear_db(): import os.path if os.path.exists(os.path.join(MAIN_DIR, "craftbeerpi.db")) is True: os.remove(os.path.join(MAIN_DIR, "craftbeerpi.db")) print("database Cleared") def check_for_setup(): if os.path.exists(os.path.join(MAIN_DIR, "config", "config.yaml")) is False: print("***************************************************") print("CraftBeerPi Config File not found: %s" % os.path.join(".", "config", "config.yaml")) print("Please run 'cbpi setup' before starting the server ") print("***************************************************") return False else: return True def plugins_add(package_name): if package_name is None: print("Pleaes provide a plugin Name") return try: with open(os.path.join(MAIN_DIR, 'config', "config.yaml"), 'rt') as f: data = yaml.load(f, Loader=yaml.FullLoader) if package_name in data["plugins"]: print("") print("Plugin {} already active".format(package_name)) print("") return data["plugins"].append(package_name) with open(os.path.join(MAIN_DIR, 'config', "config.yaml"), 'w') as outfile: yaml.dump(data, outfile, default_flow_style=False) print("") print("Plugin {} activated".format(package_name)) print("") except Exception as e: print(e) pass def plugin_remove(package_name): if package_name is None: print("Pleaes provide a plugin Name") return try: with open(os.path.join(MAIN_DIR, 'config', "config.yaml"), 'rt') as f: data = yaml.load(f, Loader=yaml.FullLoader) data["plugins"] = list(filter(lambda k: package_name not in k, data["plugins"])) with open(os.path.join(MAIN_DIR, 'config', "config.yaml"), 'w') as outfile: yaml.dump(data, outfile, default_flow_style=False) print("") print("Plugin {} deactivated".format(package_name)) print("") except Exception as e: print(e) pass def plugins_list(): print("--------------------------------------") print("List of active pluigins") try: with open(os.path.join(MAIN_DIR, 'config', "config.yaml"), 'rt') as f: data = yaml.load(f, Loader=yaml.FullLoader) for p in data["plugins"]: print("- {}".format(p)) except Exception as e: print(e) pass print("--------------------------------------") def plugin_create(name): if os.path.exists(os.path.join(MAIN_DIR, name)) is True: print("Cant create Plugin. Folder {} already exists ".format(name)) return url = 'https://github.com/Manuel83/craftbeerpi4-plugin-template/archive/main.zip' r = requests.get(url) with open('temp.zip', 'wb') as f: f.write(r.content) with ZipFile('temp.zip', 'r') as repo_zip: repo_zip.extractall() os.rename(MAIN_DIR + "/craftbeerpi4-plugin-template-main", os.path.join(MAIN_DIR, name)) os.rename(os.path.join(MAIN_DIR, name, "src"), os.path.join(MAIN_DIR, name, name)) import jinja2 templateLoader = jinja2.FileSystemLoader(searchpath=os.path.join(MAIN_DIR, name)) templateEnv = jinja2.Environment(loader=templateLoader) TEMPLATE_FILE = "setup.py" template = templateEnv.get_template(TEMPLATE_FILE) outputText = template.render(name=name) with open(os.path.join(MAIN_DIR, name, "setup.py"), "w") as fh: fh.write(outputText) TEMPLATE_FILE = "MANIFEST.in" template = templateEnv.get_template(TEMPLATE_FILE) outputText = template.render(name=name) with open(os.path.join(MAIN_DIR, name, "MANIFEST.in"), "w") as fh: fh.write(outputText) TEMPLATE_FILE = os.path.join("/", name, "config.yaml") template = templateEnv.get_template(TEMPLATE_FILE) outputText = template.render(name=name) with open(os.path.join(MAIN_DIR, name, name, "config.yaml"), "w") as fh: fh.write(outputText) print("") print("") print( "Plugin {} created! See https://craftbeerpi.gitbook.io/craftbeerpi4/development how to run your plugin ".format( name)) print("") print("Happy Development! Cheers") print("") print("") @click.group() def main(): level = logging.INFO logging.basicConfig(level=level, format='%(asctime)s - %(levelname)s - %(name)s - %(message)s') pass @click.command() def setup(): '''Create Config folder''' print("Setting up CraftBeerPi") create_home_folder_structure() create_config_file() @click.command() def start(): if check_for_setup() is False: return print("START") cbpi = CraftBeerPi() cbpi.start() @click.command() def plugins(): '''List active plugins''' plugins_list() return @click.command() @click.argument('name') def add(name): '''Activate Plugin''' plugins_add(name) @click.command() @click.argument('name') def remove(name): '''Deactivate Plugin''' plugin_remove(name) @click.command() @click.argument('name') def create(name): '''Deactivate Plugin''' plugin_create(name) main.add_command(setup) main.add_command(start) main.add_command(plugins) main.add_command(add) main.add_command(remove) main.add_command(create)