mirror of
https://github.com/PiBrewing/craftbeerpi4.git
synced 2024-12-22 21:44:57 +01:00
389 lines
14 KiB
Python
389 lines
14 KiB
Python
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 subprocess import call
|
|
import zipfile
|
|
from importlib_metadata import version, metadata
|
|
|
|
from jinja2 import Template
|
|
|
|
|
|
def create_config_file():
|
|
if os.path.exists(os.path.join(".", 'config', "config.yaml")) is False:
|
|
srcfile = os.path.join(os.path.dirname(__file__), "config", "config.yaml")
|
|
destfile = os.path.join(".", 'config')
|
|
shutil.copy(srcfile, destfile)
|
|
|
|
if os.path.exists(os.path.join(".", 'config', "actor.json")) is False:
|
|
srcfile = os.path.join(os.path.dirname(__file__), "config", "actor.json")
|
|
destfile = os.path.join(".", 'config')
|
|
shutil.copy(srcfile, destfile)
|
|
|
|
if os.path.exists(os.path.join(".", 'config', "sensor.json")) is False:
|
|
srcfile = os.path.join(os.path.dirname(__file__), "config", "sensor.json")
|
|
destfile = os.path.join(".", 'config')
|
|
shutil.copy(srcfile, destfile)
|
|
|
|
if os.path.exists(os.path.join(".", 'config', "kettle.json")) is False:
|
|
srcfile = os.path.join(os.path.dirname(__file__), "config", "kettle.json")
|
|
destfile = os.path.join(".", 'config')
|
|
shutil.copy(srcfile, destfile)
|
|
|
|
if os.path.exists(os.path.join(".", 'config', "step_data.json")) is False:
|
|
srcfile = os.path.join(os.path.dirname(__file__), "config", "step_data.json")
|
|
destfile = os.path.join(".", 'config')
|
|
shutil.copy(srcfile, destfile)
|
|
|
|
if os.path.exists(os.path.join(".", 'config', "config.json")) is False:
|
|
srcfile = os.path.join(os.path.dirname(__file__), "config", "config.json")
|
|
destfile = os.path.join(".", 'config')
|
|
shutil.copy(srcfile, destfile)
|
|
|
|
if os.path.exists(os.path.join(".", 'config', "dashboard", "cbpi_dashboard_1.json")) is False:
|
|
srcfile = os.path.join(os.path.dirname(__file__), "config", "dashboard", "cbpi_dashboard_1.json")
|
|
destfile = os.path.join(".", "config", "dashboard")
|
|
shutil.copy(srcfile, destfile)
|
|
|
|
if os.path.exists(os.path.join(".", 'config', "carftbeerpi.service")) is False:
|
|
srcfile = os.path.join(os.path.dirname(__file__), "config", "craftbeerpi.service")
|
|
destfile = os.path.join(".", 'config')
|
|
shutil.copy(srcfile, destfile)
|
|
print("Config Folder created")
|
|
|
|
|
|
def create_home_folder_structure():
|
|
pathlib.Path(os.path.join(".", 'logs/sensors')).mkdir(parents=True, exist_ok=True)
|
|
pathlib.Path(os.path.join(".", 'config')).mkdir(parents=True, exist_ok=True)
|
|
pathlib.Path(os.path.join(".", 'config/dashboard')).mkdir(parents=True, exist_ok=True)
|
|
pathlib.Path(os.path.join(".", 'config/dashboard/widgets')).mkdir(parents=True, exist_ok=True)
|
|
pathlib.Path(os.path.join(".", 'config/recipes')).mkdir(parents=True, exist_ok=True)
|
|
pathlib.Path(os.path.join(".", 'config/upload')).mkdir(parents=True, exist_ok=True)
|
|
print("Folder created")
|
|
|
|
|
|
def setup_one_wire():
|
|
print("Setting up 1Wire")
|
|
with open('/boot/config.txt', 'w') as f:
|
|
f.write("dtoverlay=w1-gpio,gpiopin=4,pullup=on")
|
|
print("/boot/config.txt created")
|
|
|
|
def list_one_wire():
|
|
print("List 1Wire")
|
|
call(["modprobe", "w1-gpio"])
|
|
call(["modprobe", "w1-therm"])
|
|
try:
|
|
for dirname in os.listdir('/sys/bus/w1/devices'):
|
|
if (dirname.startswith("28") or dirname.startswith("10")):
|
|
print(dirname)
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
def copy_splash():
|
|
srcfile = os.path.join(".", "config", "splash.png")
|
|
destfile = os.path.join(".", 'config')
|
|
shutil.copy(srcfile, destfile)
|
|
print("Splash Srceen created")
|
|
|
|
|
|
def clear_db():
|
|
import os.path
|
|
if os.path.exists(os.path.join(".", "craftbeerpi.db")) is True:
|
|
os.remove(os.path.join(".", "craftbeerpi.db"))
|
|
print("database Cleared")
|
|
|
|
def recursive_chown(path, owner, group):
|
|
for dirpath, dirnames, filenames in os.walk(path):
|
|
shutil.chown(dirpath, owner, group)
|
|
for filename in filenames:
|
|
shutil.chown(os.path.join(dirpath, filename), owner, group)
|
|
|
|
def check_for_setup():
|
|
if os.path.exists(os.path.join(".", "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
|
|
if os.path.exists(os.path.join(".", "config", "upload")) is False:
|
|
print("***************************************************")
|
|
print("CraftBeerPi upload folder not found: %s" % os.path.join(".", "config/upload"))
|
|
print("Please run 'cbpi setup' before starting the server ")
|
|
print("***************************************************")
|
|
return False
|
|
backupfile = os.path.join(".", "restored_config.zip")
|
|
if os.path.exists(os.path.join(backupfile)) is True:
|
|
print("***************************************************")
|
|
print("Found backup of config. Starting restore")
|
|
required_content=['dashboard/', 'recipes/', 'upload/', 'config.json', 'config.yaml']
|
|
zip=zipfile.ZipFile(backupfile)
|
|
zip_content_list = zip.namelist()
|
|
zip_content = True
|
|
print("Checking content of zip file")
|
|
for content in required_content:
|
|
try:
|
|
check = zip_content_list.index(content)
|
|
except:
|
|
zip_content = False
|
|
|
|
if zip_content == True:
|
|
print("Found correct content. Starting Restore process")
|
|
output_path = pathlib.Path(os.path.join(".", 'config'))
|
|
print("Removing old config folder")
|
|
shutil.rmtree(output_path, ignore_errors=True)
|
|
print("Extracting zip file to config folder")
|
|
zip.extractall(output_path)
|
|
print("Changing owner and group of config folder recursively to pi:pi")
|
|
recursive_chown(output_path, "pi", "pi")
|
|
print("Removing backup file")
|
|
os.remove(backupfile)
|
|
else:
|
|
print("Wrong Content in zip file. No restore possible")
|
|
print("Removing zip file")
|
|
os.remove(backupfile)
|
|
print("***************************************************")
|
|
|
|
return True
|
|
else:
|
|
return True
|
|
|
|
|
|
def plugins_add(package_name):
|
|
if package_name is None:
|
|
print("Pleaes provide a plugin Name")
|
|
return
|
|
|
|
if package_name == 'autostart':
|
|
print("Add craftbeerpi.service to systemd")
|
|
try:
|
|
if os.path.exists(os.path.join("/etc/systemd/system","craftbeerpi.service")) is False:
|
|
srcfile = os.path.join(".", "config", "craftbeerpi.service")
|
|
destfile = os.path.join("/etc/systemd/system")
|
|
shutil.copy(srcfile, destfile)
|
|
print("Copied craftbeerpi.service to /etc/systemd/system")
|
|
os.system('systemctl enable craftbeerpi.service')
|
|
print('Enabled craftbeerpi service')
|
|
os.system('systemctl start craftbeerpi.service')
|
|
print('Started craftbeerpi.service')
|
|
else:
|
|
print("craftbeerpi.service is already located in /etc/systemd/system")
|
|
except Exception as e:
|
|
print(e)
|
|
return
|
|
return
|
|
|
|
try:
|
|
with open(os.path.join(".", '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(".", '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
|
|
|
|
if package_name == 'autostart':
|
|
print("Remove craftbeerpi.service from systemd")
|
|
try:
|
|
status = os.popen('systemctl list-units --type=service --state=running | grep craftbeerpi.service').read()
|
|
if status.find("craftbeerpi.service") != -1:
|
|
os.system('systemctl stop craftbeerpi.service')
|
|
print('Stopped craftbeerpi service')
|
|
os.system('systemctl disable craftbeerpi.service')
|
|
print('Removed craftbeerpi.service as service')
|
|
else:
|
|
print('craftbeerpi.service service is not running')
|
|
|
|
if os.path.exists(os.path.join("/etc/systemd/system","craftbeerpi.service")) is True:
|
|
os.remove(os.path.join("/etc/systemd/system","craftbeerpi.service"))
|
|
print("Deleted craftbeerpi.service from /etc/systemd/system")
|
|
else:
|
|
print("craftbeerpi.service is not located in /etc/systemd/system")
|
|
except Exception as e:
|
|
print(e)
|
|
return
|
|
return
|
|
|
|
|
|
|
|
try:
|
|
with open(os.path.join(".", '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(".", '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 plugins")
|
|
try:
|
|
with open(os.path.join(".", 'config', "config.yaml"), 'rt') as f:
|
|
data = yaml.load(f, Loader=yaml.FullLoader)
|
|
|
|
for p in data["plugins"]:
|
|
p_metadata= metadata(p)
|
|
p_Homepage= p_metadata['Home-page']
|
|
p_version = p_metadata['Version']
|
|
p_Author = p_metadata['Author']
|
|
print("- ({})\t{}".format(p_version,p))
|
|
except Exception as e:
|
|
print(e)
|
|
pass
|
|
print("--------------------------------------")
|
|
|
|
|
|
def plugin_create(name):
|
|
if os.path.exists(os.path.join(".", 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("./craftbeerpi4-plugin-template-main", os.path.join(".", name))
|
|
os.rename(os.path.join(".", name, "src"), os.path.join(".", name, name))
|
|
|
|
import jinja2
|
|
|
|
templateLoader = jinja2.FileSystemLoader(searchpath=os.path.join(".", 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(".", 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(".", 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(".", 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()
|
|
@click.option('--list', is_flag=True, help="List all 1Wire Devices")
|
|
@click.option('--setup', is_flag=True, help="Setup 1Wire on Raspberry Pi")
|
|
def onewire(list, setup):
|
|
'''Setup 1wire on Raspberry Pi'''
|
|
if setup is True:
|
|
setup_one_wire()
|
|
if list is True:
|
|
list_one_wire()
|
|
|
|
|
|
|
|
@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):
|
|
'''Create New Plugin'''
|
|
plugin_create(name)
|
|
|
|
|
|
main.add_command(setup)
|
|
main.add_command(start)
|
|
main.add_command(plugins)
|
|
main.add_command(onewire)
|
|
main.add_command(add)
|
|
main.add_command(remove)
|
|
main.add_command(create)
|