craftbeerpi3/modules/system/__init__.py

import flask_login
import yaml
from flask import json, url_for, Response
from flask_classy import FlaskView, route
from flask_login import login_required, current_user
from git import Repo, Git
from modules.core.core import cbpi
import pprint
import time

from modules.login import User


class SystemView(FlaskView):
    def doShutdown(self):
        time.sleep(5)
        from subprocess import call
        call("halt")

    @login_required
    @route('/shutdown', methods=['POST'])
    def shutdown(self):
        """
        System Shutdown
        ---
        tags:
          - system
        responses:
          200:
            description: Shutdown triggered
        """
        self.doShutdown()

        return ('', 204)

    def doReboot(self):
        time.sleep(5)
        from subprocess import call
        call("reboot")

    @login_required
    @route('/reboot', methods=['POST'])
    def reboot(self):
        """
        System Reboot  
        ---
        tags:
          - system
        responses:
          200:
            description: Reboot triggered
        """
        self.doReboot()
        return ('', 204)

    @login_required
    @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)

    @login_required
    @route('/git/status', methods=['GET'])
    def git_status(self):
        """
        Check for GIT status
        ---
        tags:
          - system
        responses:
          200:
            description: Git Status
        """
        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}})

    @login_required
    @route('/check_update', methods=['GET'])
    def check_update(self):
        """
        Check for GIT update
        ---
        tags:
          - system
        responses:
          200:
            description: Git Changes
        """
        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)

    @login_required
    @route('/git/pull', methods=['POST'])
    def update(self):
        """
        System Update
        ---
        tags:
          - system
        responses:
          200:
            description: Git Pull Triggered
        """
        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('/connect', methods=['GET'])
    def connect(self):
        """
        Connect
        ---
        tags:
          - system
        responses:
          200:
            description: CraftBeerPi System Cache
        """
        if cbpi.get_config_parameter("password_security", "NO") == "NO":
            user = User()
            user.id = "craftbeerpi"
            flask_login.login_user(user)

        if self.api.get_config_parameter("setup", "YES") == "YES":
            return json.dumps(dict(setup=True, loggedin= current_user.is_authenticated ))
        else:
            return json.dumps(dict(setup=False, loggedin= current_user.is_authenticated))

    @login_required
    @route('/dump', methods=['GET'])
    def dump(self):
        """
        Dump Cache
        ---
        tags:
          - system
        responses:
          200:
            description: CraftBeerPi System Cache
        """
        return Response(response=json.dumps(cbpi.cache, sort_keys=True, indent=4), status=200, mimetype='application/json')



@cbpi.addon.core.initializer()
def init(cbpi):

    SystemView.api = cbpi
    SystemView.register(cbpi._app, route_base='/api/system')