def _edit_service(method: str, server: ServiceSettings):
form = ServiceSettingsForm(obj=server)
if method == 'POST' and form.validate_on_submit():
server = form.update_entry(server)
server.save()
return jsonify(status='ok'), 200
return render_template('service/edit.html', form=form)
def add(self):
model = ServiceSettings()
form = ServiceSettingsForm(request.form, obj=model)
if request.method == 'POST' and form.validate_on_submit():
form.populate_obj(model)
model = model.save()
current_user.add_server(model)
return jsonify(status='ok'), 200
return render_template('service/add.html', form=form)
def update_entry(self, settings: ServiceSettings):
settings.name = self.name.data
settings.host = self.host.get_data()
settings.http_host = self.http_host.get_data()
settings.vods_host = self.vods_host.get_data()
settings.feedback_directory = self.feedback_directory.data
settings.timeshifts_directory = self.timeshifts_directory.data
settings.hls_directory = self.hls_directory.data
settings.playlists_directory = self.playlists_directory.data
settings.dvb_directory = self.dvb_directory.data
settings.capture_card_directory = self.capture_card_directory.data
return settings
def remove(self):
servers = ServiceSettings.objects()
for server in servers:
server.remove_user(current_user.id)
current_user.delete()
return redirect(url_for('HomeView:index'))
def remove(self):
sid = request.form['sid']
server = ServiceSettings.objects(id=sid).first()
if server:
return _remove_server(server)
return jsonify(status='failed'), 404
def remove(self):
sid = request.form['sid']
server = ServiceSettings.objects(id=sid).first()
if server:
server.delete()
return jsonify(status='ok'), 200
return jsonify(status='failed'), 404
def edit(self, sid):
server = ServiceSettings.objects(id=sid).first()
form = ServiceSettingsForm(request.form, obj=server)
if request.method == 'POST' and form.validate_on_submit():
form.save()
return jsonify(status='ok'), 200
return render_template('service/edit.html', form=form)
def add(self):
model = ServiceSettings()
form = ServiceSettingsForm(obj=model)
if request.method == 'POST' and form.validate_on_submit():
new_entry = form.make_entry()
admin = UserPair(current_user.id, constants.Roles.ADMIN)
new_entry.add_user(admin)
current_user.add_server(new_entry)
return jsonify(status='ok'), 200
return render_template('service/add.html', form=form)
def _add_service(method: str):
server = ServiceSettings()
form = ServiceSettingsForm(obj=server)
if method == 'POST' and form.validate_on_submit():
new_entry = form.make_entry()
new_entry.users.append(current_user.id)
new_entry.save()
current_user.add_server(new_entry)
return jsonify(status='ok'), 200
return render_template('service/add.html', form=form)
def user_add(self, sid):
form = UserServerForm()
if request.method == 'POST' and form.validate_on_submit():
user = User.objects(email=form.email.data).first()
server = ServiceSettings.objects(id=sid).first()
if server and user:
admin = UserPair(user.id, form.role.data)
server.add_user(admin)
user.add_server(server)
return jsonify(status='ok'), 200
return render_template('service/user/add.html', form=form)
def make_entry(self):
return self.update_entry(ServiceSettings())
def _remove_server(server: ServiceSettings):
server.delete()
return jsonify(status='ok'), 200
def edit(self, sid):
server = ServiceSettings.objects(id=sid).first()
if server:
return _edit_service(request.method, server)
return jsonify(status='failed'), 404