mirror of
https://github.com/kjanat/livegraphs-django.git
synced 2026-01-16 13:02:08 +01:00
50 lines
1.3 KiB
Python
50 lines
1.3 KiB
Python
# dashboard/forms.py
|
|
|
|
from django import forms
|
|
|
|
from .models import Dashboard, DataSource
|
|
|
|
|
|
class DataSourceUploadForm(forms.ModelForm):
|
|
"""Form for uploading CSV files"""
|
|
|
|
class Meta:
|
|
model = DataSource
|
|
fields = ["name", "description", "file"]
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.company = kwargs.pop("company", None)
|
|
super().__init__(*args, **kwargs)
|
|
|
|
def save(self, commit=True):
|
|
instance = super().save(commit=False)
|
|
if self.company:
|
|
instance.company = self.company
|
|
if commit:
|
|
instance.save()
|
|
return instance
|
|
|
|
|
|
class DashboardForm(forms.ModelForm):
|
|
"""Form for creating and editing dashboards"""
|
|
|
|
class Meta:
|
|
model = Dashboard
|
|
fields = ["name", "description", "data_sources"]
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.company = kwargs.pop("company", None)
|
|
super().__init__(*args, **kwargs)
|
|
|
|
if self.company:
|
|
self.fields["data_sources"].queryset = DataSource.objects.filter(company=self.company)
|
|
|
|
def save(self, commit=True):
|
|
instance = super().save(commit=False)
|
|
if self.company:
|
|
instance.company = self.company
|
|
if commit:
|
|
instance.save()
|
|
self.save_m2m()
|
|
return instance
|