155 lines
6.3 KiB
Python
155 lines
6.3 KiB
Python
from django import forms
|
|
from .models import *
|
|
from base.forms import ModelForm
|
|
from django.template.loader import render_to_string
|
|
from django.core.exceptions import ValidationError
|
|
from django.utils.translation import gettext_lazy as _
|
|
from pathlib import Path
|
|
from .gdrive import authenticate
|
|
from django.core.files.storage import default_storage
|
|
from django.core.files.base import ContentFile
|
|
import os
|
|
|
|
|
|
class LocalBackupSetupForm(ModelForm):
|
|
verbose_name = "Server Backup"
|
|
backup_db = forms.BooleanField(required=False, help_text="Enable to backup database to server.")
|
|
backup_media = forms.BooleanField(required=False, help_text="Enable to backup all media files to server.")
|
|
interval = forms.BooleanField(required=False, help_text="Enable to automate the backup in a period of seconds.")
|
|
fixed = forms.BooleanField(required=False, help_text="Enable to automate the backup in a fixed time.")
|
|
|
|
class Meta:
|
|
model = LocalBackup
|
|
exclude = ['active']
|
|
|
|
|
|
def as_p(self):
|
|
"""
|
|
Render the form fields as HTML table rows with Bootstrap styling.
|
|
"""
|
|
context = {"form": self}
|
|
table_html = render_to_string("common_form.html", context)
|
|
return table_html
|
|
|
|
|
|
def clean(self):
|
|
cleaned_data = super().clean()
|
|
backup_db = cleaned_data.get('backup_db')
|
|
backup_media = cleaned_data.get('backup_media')
|
|
interval = cleaned_data.get('interval')
|
|
fixed = cleaned_data.get('fixed')
|
|
seconds = cleaned_data.get('seconds')
|
|
hour = cleaned_data.get('hour')
|
|
minute = cleaned_data.get('minute')
|
|
backup_path = cleaned_data.get('backup_path')
|
|
path = Path(backup_path)
|
|
if not path.exists():
|
|
raise ValidationError({
|
|
'backup_path': _('The directory does not exist.')
|
|
})
|
|
if backup_db == False and backup_media == False:
|
|
raise forms.ValidationError("Please select any backup option.")
|
|
if interval == False and fixed == False:
|
|
raise forms.ValidationError("Please select any backup automate option.")
|
|
if interval == True and seconds == None:
|
|
raise ValidationError({
|
|
'seconds': _('This field is required.')
|
|
})
|
|
if fixed == True and hour == None:
|
|
raise ValidationError({
|
|
'hour': _('This field is required.')
|
|
})
|
|
if seconds:
|
|
if seconds < 0:
|
|
raise ValidationError({
|
|
'seconds': _('Negative value is not accepatable.')
|
|
})
|
|
if hour:
|
|
if hour < 0 or hour > 24:
|
|
raise ValidationError({
|
|
'hour': _('Enter a hour between 0 to 24.')
|
|
})
|
|
if minute:
|
|
if minute < 0 or minute > 60:
|
|
raise ValidationError({
|
|
'minute': _('Enter a minute between 0 to 60.')
|
|
})
|
|
return cleaned_data
|
|
|
|
|
|
class GdriveBackupSetupForm(ModelForm):
|
|
verbose_name = "Gdrive Backup"
|
|
backup_db = forms.BooleanField(required=False, label="Backup DB", help_text="Enable to backup database to Gdrive")
|
|
backup_media = forms.BooleanField(required=False, label="Backup Media", help_text="Enable to backup all media files to Gdrive")
|
|
interval = forms.BooleanField(required=False, help_text="Enable to automate the backup in a period of seconds.")
|
|
fixed = forms.BooleanField(required=False, help_text="Enable to automate the backup in a fixed time.")
|
|
|
|
class Meta:
|
|
model = GoogleDriveBackup
|
|
exclude = ['active']
|
|
|
|
|
|
def as_p(self):
|
|
"""
|
|
Render the form fields as HTML table rows with Bootstrap styling.
|
|
"""
|
|
context = {"form": self}
|
|
table_html = render_to_string("common_form.html", context)
|
|
return table_html
|
|
|
|
def clean(self):
|
|
cleaned_data = super().clean()
|
|
backup_db = cleaned_data.get('backup_db')
|
|
backup_media = cleaned_data.get('backup_media')
|
|
interval = cleaned_data.get('interval')
|
|
fixed = cleaned_data.get('fixed')
|
|
seconds = cleaned_data.get('seconds')
|
|
hour = cleaned_data.get('hour')
|
|
minute = cleaned_data.get('minute')
|
|
service_account_file = cleaned_data.get('service_account_file')
|
|
|
|
try:
|
|
if GoogleDriveBackup.objects.exists():
|
|
authenticate(service_account_file.path)
|
|
else:
|
|
file_data = service_account_file.read()
|
|
# Save the processed file to the desired location
|
|
file_name = service_account_file.name
|
|
new_file_name = file_name
|
|
# Save using Django's default storage system
|
|
relative_path = default_storage.save(new_file_name, ContentFile(file_data))
|
|
# Get the full absolute path
|
|
full_path = default_storage.path(relative_path)
|
|
authenticate(full_path)
|
|
os.remove(full_path)
|
|
|
|
except Exception as e:
|
|
raise forms.ValidationError("Please provide a valid service account file.")
|
|
if backup_db == False and backup_media == False:
|
|
raise forms.ValidationError("Please select any backup option.")
|
|
if interval == False and fixed == False:
|
|
raise forms.ValidationError("Please select any backup automate option.")
|
|
if interval == True and seconds == None:
|
|
raise ValidationError({
|
|
'seconds': _('This field is required.')
|
|
})
|
|
if fixed == True and hour == None:
|
|
raise ValidationError({
|
|
'hour': _('This field is required.')
|
|
})
|
|
if seconds:
|
|
if seconds < 0:
|
|
raise ValidationError({
|
|
'seconds': _('Negative value is not accepatable.')
|
|
})
|
|
if hour:
|
|
if hour < 0 or hour > 24:
|
|
raise ValidationError({
|
|
'hour': _('Enter a hour between 0 to 24.')
|
|
})
|
|
if minute:
|
|
if minute < 0 or minute > 60:
|
|
raise ValidationError({
|
|
'minute': _('Enter a minute between 0 to 60.')
|
|
})
|
|
return cleaned_data |