mirror of
https://github.com/Manuel83/craftbeerpi3
synced 2026-08-10 02:55:27 +02:00
First Refactoring
- Major API Changes in Core Module - Will be changed further. Still work in progress
This commit is contained in:
Manuel83
+152
-1
@@ -1 +1,152 @@
|
||||
import endpoints
|
||||
import yaml
|
||||
from flask import json, url_for, Response
|
||||
from flask_classy import FlaskView, route
|
||||
from git import Repo, Git
|
||||
from modules.core.core 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):
|
||||
"""
|
||||
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")
|
||||
|
||||
@route('/reboot', methods=['POST'])
|
||||
def reboot(self):
|
||||
"""
|
||||
System Reboot
|
||||
---
|
||||
tags:
|
||||
- system
|
||||
responses:
|
||||
200:
|
||||
description: Reboot triggered
|
||||
"""
|
||||
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):
|
||||
"""
|
||||
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}})
|
||||
|
||||
@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)
|
||||
|
||||
@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('/dump', methods=['GET'])
|
||||
def dump(self):
|
||||
"""
|
||||
Dump Cache
|
||||
---
|
||||
tags:
|
||||
- system
|
||||
responses:
|
||||
200:
|
||||
description: CraftBeerPi System Cache
|
||||
"""
|
||||
return json.dumps(cbpi.cache)
|
||||
|
||||
|
||||
@cbpi.addon.core.initializer()
|
||||
def init(cbpi):
|
||||
|
||||
SystemView.api = cbpi
|
||||
SystemView.register(cbpi._app, route_base='/api/system')
|
||||
|
||||
@@ -1,145 +0,0 @@
|
||||
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')
|
||||
Reference in New Issue
Block a user