Improve status tracking for upgrades; add package version diff

This commit is contained in:
Lance Edgar 2017-08-07 22:23:07 -05:00
parent 430a1416c6
commit e14b5a89c3
7 changed files with 209 additions and 11 deletions

View file

@ -47,7 +47,7 @@ from pyramid.renderers import get_renderer, render_to_response, render
from pyramid.response import FileResponse
from webhelpers2.html import HTML, tags
from tailbone import forms, grids
from tailbone import forms, grids, diffs
from tailbone.views import View
from tailbone.progress import SessionProgress
@ -1685,6 +1685,9 @@ class MasterView(View):
# TODO: make this smarter?
return {'uuid': row.uuid}
def make_diff(self, old_data, new_data, **kwargs):
return diffs.Diff(old_data, new_data, **kwargs)
##############################
# Config Stuff
##############################

View file

@ -27,7 +27,11 @@ Views for app upgrades
from __future__ import unicode_literals, absolute_import
import os
import re
import six
from pip.download import PipSession
from pip.req import parse_requirements
from sqlalchemy import orm
from rattail.db import model, Session as RattailSession
@ -49,13 +53,19 @@ class UpgradeView(MasterView):
model_class = model.Upgrade
executable = True
downloadable = True
labels = {
'executed_by': "Executed by",
'status_code': "Status",
'stdout_file': "STDOUT",
'stderr_file': "STDERR",
}
grid_columns = [
'created',
'description',
# 'not_until',
'enabled',
'executing',
'status_code',
'executed',
'executed_by',
]
@ -68,11 +78,12 @@ class UpgradeView(MasterView):
'created',
'created_by',
'enabled',
'executing',
'executed',
'executed_by',
'status_code',
'stdout_file',
'stderr_file',
'package_diff',
]
def __init__(self, request):
@ -95,11 +106,11 @@ class UpgradeView(MasterView):
super(UpgradeView, self).configure_grid(g)
g.set_joiner('executed_by', lambda q: q.join(model.User).outerjoin(model.Person))
g.set_sorter('executed_by', model.Person.display_name)
g.set_enum('status_code', self.enum.UPGRADE_STATUS)
g.set_type('created', 'datetime')
g.set_type('executed', 'datetime')
g.default_sortkey = 'created'
g.default_sortdir = 'desc'
g.set_label('executed_by', "Executed by")
g.set_link('created')
g.set_link('description')
# g.set_link('not_until')
@ -107,29 +118,28 @@ class UpgradeView(MasterView):
def configure_form(self, f):
super(UpgradeView, self).configure_form(f)
f.set_enum('status_code', self.enum.UPGRADE_STATUS)
f.set_type('created', 'datetime')
f.set_type('enabled', 'boolean')
f.set_type('executing', 'boolean')
f.set_type('executed', 'datetime')
# f.set_widget('not_until', dfwidget.DateInputWidget())
f.set_widget('notes', dfwidget.TextAreaWidget(cols=80, rows=8))
f.set_renderer('stdout_file', self.render_stdout_file)
f.set_renderer('stderr_file', self.render_stdout_file)
f.set_renderer('package_diff', self.render_package_diff)
# f.set_readonly('created')
# f.set_readonly('created_by')
f.set_readonly('executing')
f.set_readonly('executed')
f.set_readonly('executed_by')
f.set_label('stdout_file', "STDOUT")
f.set_label('stderr_file', "STDERR")
upgrade = f.model_instance
if self.creating or self.editing:
f.remove_field('created')
f.remove_field('created_by')
f.remove_field('stdout_file')
f.remove_field('stderr_file')
if self.creating:
f.remove_field('status_code')
if self.creating or not upgrade.executed:
f.remove_field('executing')
f.remove_field('executed')
f.remove_field('executed_by')
if self.editing and upgrade.executed:
@ -137,7 +147,6 @@ class UpgradeView(MasterView):
elif f.model_instance.executed:
f.remove_field('enabled')
f.remove_field('executing')
else:
f.remove_field('executed')
@ -145,6 +154,9 @@ class UpgradeView(MasterView):
f.remove_field('stdout_file')
f.remove_field('stderr_file')
if not self.viewing or not upgrade.executed:
f.remove_field('package_diff')
def render_stdout_file(self, upgrade, fieldname):
if fieldname.startswith('stderr'):
filename = 'stderr.log'
@ -157,6 +169,45 @@ class UpgradeView(MasterView):
return tags.link_to(content, url)
return filename
def render_package_diff(self, upgrade, fieldname):
try:
before = self.parse_requirements(upgrade, 'before')
after = self.parse_requirements(upgrade, 'after')
diff = self.make_diff(before, after,
columns=["package", "old version", "new version"],
render_value=self.render_diff_value,
)
return diff.render_html()
except:
return "(not available for this upgrade)"
def render_diff_value(self, field, value):
if value.startswith("u'") and value.endswith("'"):
return value[2:1]
return value
def parse_requirements(self, upgrade, type_):
packages = {}
path = self.rattail_config.upgrade_filepath(upgrade.uuid, filename='requirements.{}.txt'.format(type_))
session = PipSession()
for req in parse_requirements(path, session=session):
version = self.version_from_requirement(req)
packages[req.name] = version
return packages
def version_from_requirement(self, req):
if req.specifier:
match = re.match(r'^==(.*)$', six.text_type(req.specifier))
if match:
return match.group(1)
return six.text_type(req.specifier)
elif req.link:
match = re.match(r'^.*@(.*)#egg=.*$', six.text_type(req.link))
if match:
return match.group(1)
return six.text_type(req.link)
return ""
def get_size(self, path):
try:
return os.path.getsize(path)
@ -182,6 +233,7 @@ class UpgradeView(MasterView):
def before_create_flush(self, upgrade):
upgrade.created_by = self.request.user
upgrade.status_code = self.enum.UPGRADE_STATUS_PENDING
# TODO: this was an attempt to make the progress bar survive Apache restart,
# but it didn't work... need to "fork" instead of waiting for execution?
@ -192,9 +244,11 @@ class UpgradeView(MasterView):
def execute_instance(self, upgrade, user, **kwargs):
session = orm.object_session(upgrade)
upgrade.executing = True
upgrade.status_code = self.enum.UPGRADE_STATUS_EXECUTING
session.commit()
self.handler.execute(upgrade, user, **kwargs)
upgrade.executing = False
upgrade.status_code = self.enum.UPGRADE_STATUS_SUCCEEDED
upgrade.executed = make_utc()
upgrade.executed_by = user