2
0
mirror of https://github.com/inventree/InvenTree.git synced 2025-06-20 05:46:34 +00:00

Abstract the PartAttachment class

Now "Attachments" are much easier to implement for different models
This commit is contained in:
Oliver Walters
2020-03-22 17:59:23 +11:00
parent 3b6ed585ab
commit a661d7e1a6
3 changed files with 72 additions and 19 deletions

View File

@ -0,0 +1,19 @@
# Generated by Django 2.2.9 on 2020-03-22 04:53
import InvenTree.models
from django.db import migrations, models
class Migration(migrations.Migration):
dependencies = [
('part', '0031_auto_20200318_1044'),
]
operations = [
migrations.AlterField(
model_name='partattachment',
name='attachment',
field=models.FileField(help_text='Select file to attach', upload_to=InvenTree.models.rename_attachment),
),
]

View File

@ -34,7 +34,7 @@ import hashlib
from InvenTree import helpers
from InvenTree import validators
from InvenTree.models import InvenTreeTree
from InvenTree.models import InvenTreeTree, InvenTreeAttachment
from InvenTree.fields import InvenTreeURLField
from InvenTree.helpers import decimal2string
@ -941,28 +941,14 @@ def attach_file(instance, filename):
return os.path.join('part_files', str(instance.part.id), filename)
class PartAttachment(models.Model):
""" A PartAttachment links a file to a part
Parts can have multiple files such as datasheets, etc
Attributes:
part: Link to a Part object
attachment: File
comment: String descriptor for the attachment
"""
class PartAttachment(InvenTreeAttachment):
def getSubdir(self):
return os.path.join("part_files", str(self.part.id))
part = models.ForeignKey(Part, on_delete=models.CASCADE,
related_name='attachments')
attachment = models.FileField(upload_to=attach_file,
help_text=_('Select file to attach'))
comment = models.CharField(max_length=100, help_text=_('File comment'))
@property
def basename(self):
return os.path.basename(self.attachment.name)
class PartStar(models.Model):
""" A PartStar object creates a relationship between a User and a Part.