Commit 7086132c by Őry Máté

storage: fix filename generation code

parent 1cf58446
...@@ -166,10 +166,8 @@ class Disk(AclBase, TimeStampedModel): ...@@ -166,10 +166,8 @@ class Disk(AclBase, TimeStampedModel):
if self.type not in type_mapping.keys(): if self.type not in type_mapping.keys():
raise self.WrongDiskTypeError(self.type) raise self.WrongDiskTypeError(self.type)
if self.type == 'iso':
filename = self.filename filename = self.filename if self.type == 'iso' else None
else:
self.generate_filename()
new_type = type_mapping[self.type] new_type = type_mapping[self.type]
return Disk.objects.create(base=self, datastore=self.datastore, return Disk.objects.create(base=self, datastore=self.datastore,
...@@ -210,9 +208,12 @@ class Disk(AclBase, TimeStampedModel): ...@@ -210,9 +208,12 @@ class Disk(AclBase, TimeStampedModel):
def clean(self, *args, **kwargs): def clean(self, *args, **kwargs):
if self.size == "" and self.base: if self.size == "" and self.base:
self.size = self.base.size self.size = self.base.size
super(Disk, self).clean(*args, **kwargs)
def save(self, *args, **kwargs):
if self.filename is None: if self.filename is None:
self.generate_filename() self.generate_filename()
super(Disk, self).clean(*args, **kwargs) return super(Disk, self).save(*args, **kwargs)
def deploy(self, user=None, task_uuid=None): def deploy(self, user=None, task_uuid=None):
"""Reify the disk model on the associated data store. """Reify the disk model on the associated data store.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment