update the message of upload placeholders

This commit is contained in:
Cayo Puigdefabregas 2022-09-12 17:38:29 +02:00
parent ce666692b3
commit 9969e6594d
2 changed files with 30 additions and 2 deletions

View File

@ -1490,6 +1490,11 @@ class UploadPlaceholderForm(FlaskForm):
'Select a Placeholder File', [validators.DataRequired()] 'Select a Placeholder File', [validators.DataRequired()]
) )
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.dev_new = 0
self.dev_update = 0
def get_data_file(self): def get_data_file(self):
files = request.files.getlist(self.placeholder_file.name) files = request.files.getlist(self.placeholder_file.name)
@ -1555,6 +1560,7 @@ class UploadPlaceholderForm(FlaskForm):
# update one # update one
if placeholder: if placeholder:
self.dev_update += 1
device = placeholder.device device = placeholder.device
device.model = "{}".format(data['Model'][i]).lower() device.model = "{}".format(data['Model'][i]).lower()
device.manufacturer = "{}".format(data['Manufacturer'][i]).lower() device.manufacturer = "{}".format(data['Manufacturer'][i]).lower()
@ -1594,9 +1600,11 @@ class UploadPlaceholderForm(FlaskForm):
snapshot_json = schema.load(json_snapshot) snapshot_json = schema.load(json_snapshot)
device = snapshot_json['device'] device = snapshot_json['device']
device.placeholder = Placeholder(**json_placeholder) device.placeholder = Placeholder(**json_placeholder)
self.dev_new += 1
typ = 'New device'
placeholder_log = PlaceholdersLog( placeholder_log = PlaceholdersLog(
type="New device", source=self.source, placeholder=device.placeholder type=typ, source=self.source, placeholder=device.placeholder
) )
self.placeholders.append((device, placeholder_log)) self.placeholders.append((device, placeholder_log))

View File

@ -1169,7 +1169,27 @@ class UploadPlaceholderView(GenericMixin):
lot.devices.add(device) lot.devices.add(device)
db.session.add(lot) db.session.add(lot)
db.session.commit() db.session.commit()
messages.success('Placeholders uploaded successfully!') dev_new = form.dev_new
dev_update = form.dev_update
total = dev_new + dev_update
txt = 'Placeholders uploaded successfully!'
if dev_update == 0:
txt = f'A total of {total} Placeholders have been successfully'
txt += ' uploaded. All of them have been new registrations in'
txt += ' the system.'
if dev_new == 0:
txt = f'A total of {total} Placeholders have been successfully'
txt += ' uploaded. All of them are updates.'
if dev_new and dev_update:
txt = f'A total of {total} Placeholders have been successfully'
txt += f' uploaded. Among these {dev_new} are registered for '
txt += ' the first time in the system and another'
txt += f' {dev_update} have been updated.'
messages.success(txt)
return flask.render_template(self.template_name, **self.context) return flask.render_template(self.template_name, **self.context)