Browse Source

Merge branch 'python3-support'

pull/259/head
Juan Pablo Giménez 5 years ago
parent
commit
467b3feea0
28 changed files with 285 additions and 283 deletions
  1. +4
    -0
      .gitignore
  2. +2
    -2
      Dockerfile
  3. +71
    -71
      modules/__init__.py
  4. +1
    -1
      modules/addon/__init__.py
  5. +4
    -3
      modules/addon/endpoints.py
  6. +4
    -4
      modules/base_plugins/brew_steps/__init__.py
  7. +1
    -1
      modules/base_plugins/dummy_temp/__init__.py
  8. +6
    -7
      modules/base_plugins/gpio_actor/__init__.py
  9. +1
    -1
      modules/buzzer/__init__.py
  10. +1
    -1
      modules/config/__init__.py
  11. +1
    -1
      modules/core/baseview.py
  12. +1
    -1
      modules/core/controller.py
  13. +10
    -10
      modules/core/core.py
  14. +1
    -1
      modules/core/db.py
  15. +8
    -10
      modules/core/db_mirgrate.py
  16. +3
    -3
      modules/core/hardware.py
  17. +5
    -5
      modules/core/step.py
  18. +4
    -4
      modules/fermenter/__init__.py
  19. +1
    -1
      modules/kettle/__init__.py
  20. +1
    -1
      modules/logs/__init__.py
  21. +2
    -2
      modules/logs/endpoints.py
  22. +3
    -3
      modules/recipe_import/__init__.py
  23. +1
    -1
      modules/recipe_import/kbh.py
  24. +1
    -1
      modules/sensors/__init__.py
  25. +1
    -1
      modules/system/__init__.py
  26. +145
    -145
      modules/system/endpoints.py
  27. +1
    -1
      modules/ui/__init__.py
  28. +1
    -1
      requirements.txt

+ 4
- 0
.gitignore View File

@@ -17,3 +17,7 @@ yarn.lock




modules/ui/package-lock.json modules/ui/package-lock.json

.python-version
upload/*
*.bak

+ 2
- 2
Dockerfile View File

@@ -1,5 +1,5 @@
# Dockerfile for development on a pc/mac # Dockerfile for development on a pc/mac
FROM python:2
FROM python:3.5


EXPOSE 5000 EXPOSE 5000


@@ -11,4 +11,4 @@ RUN pip install --no-cache-dir -r requirements.txt


COPY . . COPY . .


CMD ["python", "run.py"]
CMD ["python", "run.py"]

+ 71
- 71
modules/__init__.py View File

@@ -1,72 +1,72 @@
import json
import pprint
import sys, os
from flask import Flask, render_template, redirect
from flask_socketio import SocketIO, emit
import logging
# Define the WSGI application object
from app_config import *
import pprint
from modules.core.db import get_db
@app.route('/')
def index():
return redirect('ui')
# Define the database object which is imported
# by modules and controllers
import modules.steps
import modules.config
import modules.logs
import modules.sensors
import modules.actor
import modules.notification
import modules.fermenter
from modules.addon.endpoints import initPlugins
import modules.ui
import modules.system
import modules.buzzer
import modules.stats
import modules.kettle
import modules.recipe_import
import modules.core.db_mirgrate
from app_config import cbpi
# Build the database:
# This will create the database file using SQLAlchemy
pp = pprint.PrettyPrinter(indent=6)
def init_db():
print "INIT DB"
with app.app_context():
db = get_db()
try:
with app.open_resource('../config/schema.sql', mode='r') as f:
db.cursor().executescript(f.read())
db.commit()
except Exception as e:
pass
init_db()
initPlugins()
cbpi.run_init()
cbpi.run_background_processes()
app.logger.info("##########################################")
app.logger.info("### STARTUP COMPLETE")
import json
import pprint
import sys, os
from flask import Flask, render_template, redirect
from flask_socketio import SocketIO, emit
import logging
# Define the WSGI application object
from modules.app_config import *
import pprint
from modules.core.db import get_db
@app.route('/')
def index():
return redirect('ui')
# Define the database object which is imported
# by modules and controllers
import modules.steps
import modules.config
import modules.logs
import modules.sensors
import modules.actor
import modules.notification
import modules.fermenter
from modules.addon.endpoints import initPlugins
import modules.ui
import modules.system
import modules.buzzer
import modules.stats
import modules.kettle
import modules.recipe_import
import modules.core.db_mirgrate
from .app_config import cbpi
# Build the database:
# This will create the database file using SQLAlchemy
pp = pprint.PrettyPrinter(indent=6)
def init_db():
print("INIT DB")
with app.app_context():
db = get_db()
try:
with app.open_resource('../config/schema.sql', mode='r') as f:
db.cursor().executescript(f.read())
db.commit()
except Exception as e:
pass
init_db()
initPlugins()
cbpi.run_init()
cbpi.run_background_processes()
app.logger.info("##########################################")
app.logger.info("### STARTUP COMPLETE")
app.logger.info("##########################################") app.logger.info("##########################################")

+ 1
- 1
modules/addon/__init__.py View File

@@ -1 +1 @@
import endpoints
import modules.addon.endpoints

+ 4
- 3
modules/addon/endpoints.py View File

@@ -11,6 +11,7 @@ import os
import requests import requests
import yaml import yaml
import shutil import shutil
import imp


blueprint = Blueprint('addon', __name__) blueprint = Blueprint('addon', __name__)


@@ -24,7 +25,7 @@ def merge(source, destination):
:param destination: :param destination:
:return: :return:
""" """
for key, value in source.items():
for key, value in list(source.items()):
if isinstance(value, dict): if isinstance(value, dict):
# get node or create one # get node or create one
node = destination.setdefault(key, {}) node = destination.setdefault(key, {})
@@ -115,7 +116,7 @@ def reload(name):
""" """
try: try:
if name in cache["modules"]: if name in cache["modules"]:
reload(cache["modules"][name])
imp.reload(cache["modules"][name])
cbpi.emit_message("REALOD OF PLUGIN %s SUCCESSFUL" % (name)) cbpi.emit_message("REALOD OF PLUGIN %s SUCCESSFUL" % (name))
return ('', 204) return ('', 204)
else: else:
@@ -134,7 +135,7 @@ def plugins():
""" """
response = requests.get("https://raw.githubusercontent.com/Manuel83/craftbeerpi-plugins/master/plugins.yaml") response = requests.get("https://raw.githubusercontent.com/Manuel83/craftbeerpi-plugins/master/plugins.yaml")
cbpi.cache["plugins"] = merge(yaml.load(response.text), cbpi.cache["plugins"]) cbpi.cache["plugins"] = merge(yaml.load(response.text), cbpi.cache["plugins"])
for key, value in cbpi.cache["plugins"].iteritems():
for key, value in cbpi.cache["plugins"].items():
value["installed"] = os.path.isdir("./modules/plugins/%s/" % (key)) value["installed"] = os.path.isdir("./modules/plugins/%s/" % (key))


return json.dumps(cbpi.cache["plugins"]) return json.dumps(cbpi.cache["plugins"])


+ 4
- 4
modules/base_plugins/brew_steps/__init__.py View File

@@ -58,7 +58,7 @@ class MashStep(StepBase):
# Check if timer finished and go to next step # Check if timer finished and go to next step
if self.is_timer_finished() == True: if self.is_timer_finished() == True:
self.notify("Mash Step Completed!", "Starting the next step", timeout=None) self.notify("Mash Step Completed!", "Starting the next step", timeout=None)
self.next()
next(self)




@cbpi.step @cbpi.step
@@ -121,7 +121,7 @@ class ChilStep(StepBase):
self.start_timer(int(self.timer) * 60) self.start_timer(int(self.timer) * 60)


if self.is_timer_finished() == True: if self.is_timer_finished() == True:
self.next()
next(self)


@cbpi.step @cbpi.step
class PumpStep(StepBase): class PumpStep(StepBase):
@@ -149,7 +149,7 @@ class PumpStep(StepBase):
self.start_timer(int(self.timer) * 60) self.start_timer(int(self.timer) * 60)


if self.is_timer_finished() == True: if self.is_timer_finished() == True:
self.next()
next(self)


@cbpi.step @cbpi.step
class BoilStep(StepBase): class BoilStep(StepBase):
@@ -226,4 +226,4 @@ class BoilStep(StepBase):
# Check if timer finished and go to next step # Check if timer finished and go to next step
if self.is_timer_finished() == True: if self.is_timer_finished() == True:
self.notify("Boil Step Completed!", "Starting the next step", timeout=None) self.notify("Boil Step Completed!", "Starting the next step", timeout=None)
self.next()
next(self)

+ 1
- 1
modules/base_plugins/dummy_temp/__init__.py View File

@@ -15,7 +15,7 @@ class DummyTempSensor(SensorActive):


@cbpi.action("My Custom Action") @cbpi.action("My Custom Action")
def my_action(self): def my_action(self):
print "HELLO WORLD"
print("HELLO WORLD")
pass pass


def get_unit(self): def get_unit(self):


+ 6
- 7
modules/base_plugins/gpio_actor/__init__.py View File

@@ -10,7 +10,7 @@ try:


GPIO.setmode(GPIO.BCM) GPIO.setmode(GPIO.BCM)
except Exception as e: except Exception as e:
print e
print(e)
pass pass




@@ -25,11 +25,11 @@ class GPIOSimple(ActorBase):
GPIO.output(int(self.gpio), 0) GPIO.output(int(self.gpio), 0)


def on(self, power=0): def on(self, power=0):
print "GPIO ON %s" % str(self.gpio)
print(("GPIO ON %s" % str(self.gpio)))
GPIO.output(int(self.gpio), 1) GPIO.output(int(self.gpio), 1)


def off(self): def off(self):
print "GPIO OFF"
print("GPIO OFF")
GPIO.output(int(self.gpio), 0) GPIO.output(int(self.gpio), 0)


@cbpi.actor @cbpi.actor
@@ -68,7 +68,7 @@ class GPIOPWM(ActorBase):
self.p.ChangeDutyCycle(self.power) self.p.ChangeDutyCycle(self.power)


def off(self): def off(self):
print "GPIO OFF"
print("GPIO OFF")
self.p.stop() self.p.stop()




@@ -99,11 +99,10 @@ class Dummy(ActorBase):
:param power: int value between 0 - 100 :param power: int value between 0 - 100
:return: :return:
''' '''
print "ON"
print("ON")


def off(self): def off(self):
print "OFF"

print("OFF")


@cbpi.actor @cbpi.actor
class DummyPWM(ActorBase): class DummyPWM(ActorBase):


+ 1
- 1
modules/buzzer/__init__.py View File

@@ -1,5 +1,5 @@
import time import time
from thread import start_new_thread
from _thread import start_new_thread
from modules import cbpi from modules import cbpi


try: try:


+ 1
- 1
modules/config/__init__.py View File

@@ -46,7 +46,7 @@ class ConfigView(BaseView):


with cls.api.app.app_context(): with cls.api.app.app_context():
cls.api.cache[cls.cache_key] = {} cls.api.cache[cls.cache_key] = {}
for key, value in cls.model.get_all().iteritems():
for key, value in cls.model.get_all().items():
cls.post_init_callback(value) cls.post_init_callback(value)
cls.api.cache[cls.cache_key][value.name] = value cls.api.cache[cls.cache_key][value.name] = value




+ 1
- 1
modules/core/baseview.py View File

@@ -104,6 +104,6 @@ class BaseView(FlaskView):
cls.api.cache[cls.cache_key].append(value) cls.api.cache[cls.cache_key].append(value)
else: else:
cls.api.cache[cls.cache_key] = {} cls.api.cache[cls.cache_key] = {}
for key, value in cls.model.get_all().iteritems():
for key, value in list(cls.model.get_all().items()):
cls.post_init_callback(value) cls.post_init_callback(value)
cls.api.cache[cls.cache_key][key] = value cls.api.cache[cls.cache_key][key] = value

+ 1
- 1
modules/core/controller.py View File

@@ -42,7 +42,7 @@ class ControllerBase(object):


@staticmethod @staticmethod
def init_global(): def init_global():
print "GLOBAL CONTROLLER INIT"
print("GLOBAL CONTROLLER INIT")


def notify(self, headline, message, type="success", timeout=5000): def notify(self, headline, message, type="success", timeout=5000):
self.api.notify(headline, message, type, timeout) self.api.notify(headline, message, type, timeout)


+ 10
- 10
modules/core/core.py View File

@@ -12,9 +12,9 @@ from time import localtime, strftime
from functools import wraps, update_wrapper from functools import wraps, update_wrapper




from props import *
from modules.core.props import *


from hardware import *
from modules.core.hardware import *


import time import time
import uuid import uuid
@@ -28,7 +28,7 @@ class ActorAPI(object):
def init_actors(self): def init_actors(self):
self.app.logger.info("Init Actors") self.app.logger.info("Init Actors")
t = self.cache.get("actor_types") t = self.cache.get("actor_types")
for key, value in t.iteritems():
for key, value in list(t.items()):
value.get("class").api = self value.get("class").api = self
value.get("class").init_global() value.get("class").init_global()


@@ -89,7 +89,7 @@ class SensorAPI(object):
self.app.logger.info("Init Sensors") self.app.logger.info("Init Sensors")


t = self.cache.get("sensor_types") t = self.cache.get("sensor_types")
for key, value in t.iteritems():
for key, value in list(t.items()):
value.get("class").init_global() value.get("class").init_global()


for key in self.cache.get("sensors"): for key in self.cache.get("sensors"):
@@ -292,7 +292,7 @@ class CraftBeerPi(ActorAPI, SensorAPI):
t = tmpObj.__getattribute__(m) t = tmpObj.__getattribute__(m)
self.cache[key][name]["properties"].append({"name": m, "label": t.label, "type": "kettle", "configurable": t.configurable, "description": t.description}) self.cache[key][name]["properties"].append({"name": m, "label": t.label, "type": "kettle", "configurable": t.configurable, "description": t.description})


for name, method in cls.__dict__.iteritems():
for name, method in list(cls.__dict__.items()):
if hasattr(method, "action"): if hasattr(method, "action"):
label = method.__getattribute__("label") label = method.__getattribute__("label")
self.cache[key][cls.__name__]["actions"].append({"method": name, "label": label}) self.cache[key][cls.__name__]["actions"].append({"method": name, "label": label})
@@ -309,10 +309,10 @@ class CraftBeerPi(ActorAPI, SensorAPI):
def actor2(self, description="", power=True, **options): def actor2(self, description="", power=True, **options):


def decorator(f): def decorator(f):
print f()
print f
print options
print description
print((f()))
print(f)
print(options)
print(description)
return f return f
return decorator return decorator


@@ -369,7 +369,7 @@ class CraftBeerPi(ActorAPI, SensorAPI):
t = tmpObj.__getattribute__(m) t = tmpObj.__getattribute__(m)
self.cache[key][name]["properties"].append({"name": m, "label": t.label, "type": "kettle", "configurable": t.configurable, "description": t.description}) self.cache[key][name]["properties"].append({"name": m, "label": t.label, "type": "kettle", "configurable": t.configurable, "description": t.description})


for name, method in cls.__dict__.iteritems():
for name, method in list(cls.__dict__.items()):
if hasattr(method, "action"): if hasattr(method, "action"):
label = method.__getattribute__("label") label = method.__getattribute__("label")
self.cache[key][cls.__name__]["actions"].append({"method": name, "label": label}) self.cache[key][cls.__name__]["actions"].append({"method": name, "label": label})


+ 1
- 1
modules/core/db.py View File

@@ -79,7 +79,7 @@ class DBModel(object):
cur = get_db().cursor() cur = get_db().cursor()




if cls.__priamry_key__ is not None and kwargs.has_key(cls.__priamry_key__):
if cls.__priamry_key__ is not None and cls.__priamry_key__ in kwargs:
query = "INSERT INTO %s (%s, %s) VALUES (?, %s)" % ( query = "INSERT INTO %s (%s, %s) VALUES (?, %s)" % (
cls.__table_name__, cls.__table_name__,
cls.__priamry_key__, cls.__priamry_key__,


+ 8
- 10
modules/core/db_mirgrate.py View File

@@ -1,10 +1,11 @@
import sqlite3 import sqlite3
import os import os
from modules import cbpi from modules import cbpi
from db import get_db
from .db import get_db


def execute_file(curernt_version, data):
if curernt_version >= data["version"]:

def execute_file(current_version, data):
if current_version >= data["version"]:
cbpi.app.logger.info("SKIP DB FILE: %s" % data["file"]) cbpi.app.logger.info("SKIP DB FILE: %s" % data["file"])
return return
try: try:
@@ -19,8 +20,9 @@ def execute_file(curernt_version, data):
conn.commit() conn.commit()


except sqlite3.OperationalError as err: except sqlite3.OperationalError as err:
print "EXCEPT"
print err
print("EXCEPT")
print(err)



@cbpi.initalizer(order=-9999) @cbpi.initalizer(order=-9999)
def init(app=None): def init(app=None):
@@ -28,7 +30,7 @@ def init(app=None):
with cbpi.app.app_context(): with cbpi.app.app_context():
conn = get_db() conn = get_db()
cur = conn.cursor() cur = conn.cursor()
current_version = None
current_version = 0
try: try:
cur.execute("SELECT max(version) as m FROM schema_info") cur.execute("SELECT max(version) as m FROM schema_info")
m = cur.fetchone() m = cur.fetchone()
@@ -41,7 +43,3 @@ def init(app=None):
d = {"version": int(filename[:filename.index('_')]), "file": filename} d = {"version": int(filename[:filename.index('_')]), "file": filename}
result.append(d) result.append(d)
execute_file(current_version, d) execute_file(current_version, d)





+ 3
- 3
modules/core/hardware.py View File

@@ -44,10 +44,10 @@ class SensorBase(Base):
last_value = 0 last_value = 0


def init(self): def init(self):
print "INIT Base SENSOR"
print("INIT Base SENSOR")


def stop(self): def stop(self):
print "STOP SENSOR"
print("STOP SENSOR")


def data_received(self, data): def data_received(self, data):


@@ -86,7 +86,7 @@ class SensorActive(SensorBase):


class SensorPassive(SensorBase): class SensorPassive(SensorBase):
def init(self): def init(self):
print "INIT PASSIV SENSOR"
print("INIT PASSIV SENSOR")
pass pass


def read(self): def read(self):


+ 5
- 5
modules/core/step.py View File

@@ -96,7 +96,7 @@ class StepBase(Timer, ActorAPI, SensorAPI, KettleAPI):
managed_fields = [] managed_fields = []
n = False n = False


def next(self):
def __next__(self):
self.n = True self.n = True


def init(self): def init(self):
@@ -109,10 +109,10 @@ class StepBase(Timer, ActorAPI, SensorAPI, KettleAPI):
pass pass


def execute(self): def execute(self):
print "-------------"
print "Step Info"
print "Kettle ID: %s" % self.kettle_id
print "ID: %s" % self.id
print("-------------")
print("Step Info")
print(("Kettle ID: %s" % self.kettle_id))
print(("ID: %s" % self.id))




def __init__(self, *args, **kwds): def __init__(self, *args, **kwds):


+ 4
- 4
modules/fermenter/__init__.py View File

@@ -210,7 +210,7 @@ class FermenterView(BaseView):
def toggle(self, id): def toggle(self, id):
fermenter = cbpi.cache.get(self.cache_key)[id] fermenter = cbpi.cache.get(self.cache_key)[id]
try: try:
print fermenter.state
print((fermenter.state))
if fermenter.state is False: if fermenter.state is False:
# Start controller # Start controller
if fermenter.logic is not None: if fermenter.logic is not None:
@@ -236,7 +236,7 @@ class FermenterView(BaseView):
cbpi.emit("UPDATE_FERMENTER", cbpi.cache.get(self.cache_key).get(id)) cbpi.emit("UPDATE_FERMENTER", cbpi.cache.get(self.cache_key).get(id))


except Exception as e: except Exception as e:
print e
print(e)
cbpi.notify("Toogle Fementer Controller failed", "Pleae check the %s configuration" % fermenter.name, cbpi.notify("Toogle Fementer Controller failed", "Pleae check the %s configuration" % fermenter.name,
type="danger", timeout=None) type="danger", timeout=None)
return ('', 500) return ('', 500)
@@ -261,7 +261,7 @@ class FermenterView(BaseView):
cbpi.emit("UPDATE_FERMENTER", cbpi.cache.get(self.cache_key)[id]) cbpi.emit("UPDATE_FERMENTER", cbpi.cache.get(self.cache_key)[id])


def check_step(self): def check_step(self):
for key, value in cbpi.cache["fermenter_task"].iteritems():
for key, value in cbpi.cache["fermenter_task"].items():


try: try:
fermenter = self.get_fermenter(key) fermenter = self.get_fermenter(key)
@@ -292,7 +292,7 @@ def read_target_temps(api):
:return: None :return: None
""" """
result = {} result = {}
for key, value in cbpi.cache.get("fermenter").iteritems():
for key, value in cbpi.cache.get("fermenter").items():
cbpi.save_to_file(key, value.target_temp, prefix="fermenter") cbpi.save_to_file(key, value.target_temp, prefix="fermenter")






+ 1
- 1
modules/kettle/__init__.py View File

@@ -86,7 +86,7 @@ def read_target_temps(api):
:return: None :return: None
""" """
result = {} result = {}
for key, value in cbpi.cache.get("kettle").iteritems():
for key, value in cbpi.cache.get("kettle").items():
cbpi.save_to_file(key, value.target_temp, prefix="kettle") cbpi.save_to_file(key, value.target_temp, prefix="kettle")


@cbpi.initalizer() @cbpi.initalizer()


+ 1
- 1
modules/logs/__init__.py View File

@@ -1 +1 @@
import endpoints
import modules.logs.endpoints

+ 2
- 2
modules/logs/endpoints.py View File

@@ -78,11 +78,11 @@ class LogView(FlaskView):


if t == "k": if t == "k":
kettle = cbpi.cache.get("kettle").get(id) kettle = cbpi.cache.get("kettle").get(id)
result = map(self.convert_chart_data_to_json, cbpi.get_controller(kettle.logic).get("class").chart(kettle))
result = list(map(self.convert_chart_data_to_json, cbpi.get_controller(kettle.logic).get("class").chart(kettle)))


if t == "f": if t == "f":
fermenter = cbpi.cache.get("fermenter").get(id) fermenter = cbpi.cache.get("fermenter").get(id)
result = map(self.convert_chart_data_to_json, cbpi.get_fermentation_controller(fermenter.logic).get("class").chart(fermenter))
result = list(map(self.convert_chart_data_to_json, cbpi.get_fermentation_controller(fermenter.logic).get("class").chart(fermenter)))


return json.dumps(result) return json.dumps(result)




+ 3
- 3
modules/recipe_import/__init__.py View File

@@ -1,3 +1,3 @@
import beerxml
import kbh
import restapi
from . import beerxml
from . import kbh
from . import restapi

+ 1
- 1
modules/recipe_import/kbh.py View File

@@ -29,7 +29,7 @@ class KBH(FlaskView):
result.append({"id": row[0], "name": row[1], "brewed": row[2]}) result.append({"id": row[0], "name": row[1], "brewed": row[2]})
return json.dumps(result) return json.dumps(result)
except Exception as e: except Exception as e:
print e
print(e)
self.api.notify(headline="Failed to load KHB database", message="ERROR", type="danger") self.api.notify(headline="Failed to load KHB database", message="ERROR", type="danger")
return ('', 500) return ('', 500)
finally: finally:


+ 1
- 1
modules/sensors/__init__.py View File

@@ -43,6 +43,6 @@ def read_passive_sensor(api):
:return: None :return: None


""" """
for key, value in cbpi.cache.get("sensors").iteritems():
for key, value in cbpi.cache.get("sensors").items():
if value.mode == "P": if value.mode == "P":
value.instance.read() value.instance.read()

+ 1
- 1
modules/system/__init__.py View File

@@ -1 +1 @@
import endpoints
import modules.system.endpoints

+ 145
- 145
modules/system/endpoints.py View File

@@ -1,145 +1,145 @@
import yaml
from flask import json, url_for, Response
from flask_classy import FlaskView, route
from git import Repo, Git
from modules.app_config import cbpi
import pprint
import time
class SystemView(FlaskView):
def doShutdown(self):
time.sleep(5)
from subprocess import call
call("halt")
@route('/shutdown', methods=['POST'])
def shutdown(self):
"""
Shutdown hook
:return: HTTP 204
"""
self.doShutdown()
return ('', 204)
def doReboot(self):
time.sleep(5)
from subprocess import call
call("reboot")
@route('/reboot', methods=['POST'])
def reboot(self):
"""
Reboot hook
:return: HTTP 204
"""
self.doReboot()
return ('', 204)
@route('/tags/<name>', methods=['GET'])
def checkout_tag(self,name):
repo = Repo('./')
repo.git.reset('--hard')
o = repo.remotes.origin
o.fetch()
g = Git('./')
g.checkout(name)
cbpi.notify("Checkout successful", "Please restart the system")
return ('', 204)
@route('/git/status', methods=['GET'])
def git_status(self):
repo = Repo('./')
o = repo.remotes.origin
o.fetch()
# Tags
tags = []
for t in repo.tags:
tags.append({"name": t.name, "commit": str(t.commit), "date": t.commit.committed_date,
"committer": t.commit.committer.name, "message": t.commit.message})
try:
branch_name = repo.active_branch.name
# test1
except:
branch_name = None
changes = []
commits_behind = repo.iter_commits('master..origin/master')
for c in list(commits_behind):
changes.append({"committer": c.committer.name, "message": c.message})
return json.dumps({"tags": tags, "headcommit": str(repo.head.commit), "branchname": branch_name,
"master": {"changes": changes}})
@route('/check_update', methods=['GET'])
def check_update(self):
repo = Repo('./')
o = repo.remotes.origin
o.fetch()
changes = []
commits_behind = repo.iter_commits('master..origin/master')
for c in list(commits_behind):
changes.append({"committer": c.committer.name, "message": c.message})
return json.dumps(changes)
@route('/git/pull', methods=['POST'])
def update(self):
repo = Repo('./')
o = repo.remotes.origin
info = o.pull()
cbpi.notify("Pull successful", "The lasted updated was downloaded. Please restart the system")
return ('', 204)
@route('/dump', methods=['GET'])
def dump(self):
return json.dumps(cbpi.cache)
@route('/endpoints', methods=['GET'])
def endpoints(self):
import urllib
output = []
vf = self.api.app.view_functions
for f in self.api.app.view_functions:
print f
endpoints = {}
re = {
"swagger": "2.0",
"host": "",
"info": {
"description":"",
"version": "",
"title": "CraftBeerPi"
},
"schemes": ["http"],
"paths": endpoints}
for rule in self.api.app.url_map.iter_rules():
r = rule
endpoints[rule.rule] = {}
if "HEAD" in r.methods: r.methods.remove("HEAD")
if "OPTIONS" in r.methods: r.methods.remove("OPTIONS")
for m in rule.methods:
endpoints[rule.rule][m] = dict(summary="", description="", consumes=["application/json"],produces=["application/json"])
with open("config/version.yaml", 'r') as stream:
y = yaml.load(stream)
pprint.pprint(y)
pprint.pprint(re)
return Response(yaml.dump(re), mimetype='text/yaml')
@cbpi.initalizer()
def init(cbpi):
SystemView.api = cbpi
SystemView.register(cbpi.app, route_base='/api/system')
import yaml
from flask import json, url_for, Response
from flask_classy import FlaskView, route
from git import Repo, Git
from modules.app_config import cbpi
import pprint
import time
class SystemView(FlaskView):
def doShutdown(self):
time.sleep(5)
from subprocess import call
call("halt")
@route('/shutdown', methods=['POST'])
def shutdown(self):
"""
Shutdown hook
:return: HTTP 204
"""
self.doShutdown()
return ('', 204)
def doReboot(self):
time.sleep(5)
from subprocess import call
call("reboot")
@route('/reboot', methods=['POST'])
def reboot(self):
"""
Reboot hook
:return: HTTP 204
"""
self.doReboot()
return ('', 204)
@route('/tags/<name>', methods=['GET'])
def checkout_tag(self,name):
repo = Repo('./')
repo.git.reset('--hard')
o = repo.remotes.origin
o.fetch()
g = Git('./')
g.checkout(name)
cbpi.notify("Checkout successful", "Please restart the system")
return ('', 204)
@route('/git/status', methods=['GET'])
def git_status(self):
repo = Repo('./')
o = repo.remotes.origin
o.fetch()
# Tags
tags = []
for t in repo.tags:
tags.append({"name": t.name, "commit": str(t.commit), "date": t.commit.committed_date,
"committer": t.commit.committer.name, "message": t.commit.message})
try:
branch_name = repo.active_branch.name
# test1
except:
branch_name = None
changes = []
commits_behind = repo.iter_commits('master..origin/master')
for c in list(commits_behind):
changes.append({"committer": c.committer.name, "message": c.message})
return json.dumps({"tags": tags, "headcommit": str(repo.head.commit), "branchname": branch_name,
"master": {"changes": changes}})
@route('/check_update', methods=['GET'])
def check_update(self):
repo = Repo('./')
o = repo.remotes.origin
o.fetch()
changes = []
commits_behind = repo.iter_commits('master..origin/master')
for c in list(commits_behind):
changes.append({"committer": c.committer.name, "message": c.message})
return json.dumps(changes)
@route('/git/pull', methods=['POST'])
def update(self):
repo = Repo('./')
o = repo.remotes.origin
info = o.pull()
cbpi.notify("Pull successful", "The lasted updated was downloaded. Please restart the system")
return ('', 204)
@route('/dump', methods=['GET'])
def dump(self):
return json.dumps(cbpi.cache)
@route('/endpoints', methods=['GET'])
def endpoints(self):
import urllib.request, urllib.parse, urllib.error
output = []
vf = self.api.app.view_functions
for f in self.api.app.view_functions:
print(f)
endpoints = {}
re = {
"swagger": "2.0",
"host": "",
"info": {
"description":"",
"version": "",
"title": "CraftBeerPi"
},
"schemes": ["http"],
"paths": endpoints}
for rule in self.api.app.url_map.iter_rules():
r = rule
endpoints[rule.rule] = {}
if "HEAD" in r.methods: r.methods.remove("HEAD")
if "OPTIONS" in r.methods: r.methods.remove("OPTIONS")
for m in rule.methods:
endpoints[rule.rule][m] = dict(summary="", description="", consumes=["application/json"],produces=["application/json"])
with open("config/version.yaml", 'r') as stream:
y = yaml.load(stream)
pprint.pprint(y)
pprint.pprint(re)
return Response(yaml.dump(re), mimetype='text/yaml')
@cbpi.initalizer()
def init(cbpi):
SystemView.api = cbpi
SystemView.register(cbpi.app, route_base='/api/system')

+ 1
- 1
modules/ui/__init__.py View File

@@ -1 +1 @@
import endpoints
import modules.ui.endpoints

+ 1
- 1
requirements.txt View File

@@ -11,4 +11,4 @@ requests==2.20.0
Werkzeug==0.11.10 Werkzeug==0.11.10
httplib2==0.9.2 httplib2==0.9.2
flask-classy==0.6.10 flask-classy==0.6.10
GitPython==2.1.3
GitPython

Loading…
Cancel
Save