Ejemplo n.º 1
0
    def process_attachments(self, files, parent):
        for k, file in files.iteritems():
            filename = file.filename
            if file and \
                    '.' in filename and \
                    filename.split('.', 1)[1] in app.config['ALLOWED_EXTENSIONS']:
                filename = secure_filename(filename)
                filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
                file.save(filepath)

                file_id = self.upload_file(filepath)
                attachment = Attachment(file_id=file_id)
                attachment.project = self
                attachment.parent = parent
                attachment.save()

                self.attachments.append(attachment)
                self.save()

                parent.attachments.append(attachment)
                parent.save()