mirror of
https://github.com/eitchtee/WYGIWYH.git
synced 2026-03-24 02:11:18 +01:00
12 lines
379 B
Python
12 lines
379 B
Python
from import_export.widgets import ForeignKeyWidget
|
|
|
|
|
|
class AutoCreateForeignKeyWidget(ForeignKeyWidget):
|
|
def clean(self, value, row=None, *args, **kwargs):
|
|
if value:
|
|
try:
|
|
return super().clean(value, row, **kwargs)
|
|
except self.model.DoesNotExist:
|
|
return self.model.objects.create(name=value)
|
|
return None
|