def get(self, *args, **kwargs):
try:
Branch.stop_container(self.branch)
Branch.rm_container(self.branch)
except:
pass
self.branch.delete()
return super(BranchDeleteView, self).get(*args, **kwargs)
def _get_branch(self):
""" Get the model to save.
Could be new in case of create, or existing.
"""
if not self.branch:
self.branch = Branch()
return self.branch
def events():
try:
log_gen = Branch.get_log_stream(kwargs.get('container_id'))
for line in log_gen:
yield "data: {}\n".format(line)
except IOError:
log_gen.close()
return
def form_valid(self, form, **context):
branch = Branch.query.filter_by(
name=form.name.data).first() or self._get_branch()
form.populate_obj(branch)
branch.user = current_user
branch.save()
if not Branch.is_container_running(branch.container_id):
branch.run_container()
self.branch = branch
flash("Branch saved", "success")
return super(BranchEditView, self).form_valid(form, **context)
def get_value(self, row, **kwargs):
return Branch.check_status(row['container_id'])
def get(self, *args, **kwargs):
Branch.stop_container(self.branch)
Branch.rm_container(self.branch)
return super(BranchStopView, self).get(*args, **kwargs)