59 lines
2.3 KiB
Python
59 lines
2.3 KiB
Python
from django.db import models
|
|
|
|
from Accounts .models import UserProfile
|
|
from Device.models import Devices
|
|
# Create your models here.
|
|
class Status(models.Model):
|
|
number = models.CharField(max_length=15)
|
|
def __str__(self):
|
|
return self.number
|
|
|
|
class Number(models.Model):
|
|
status = models.BooleanField(default=False) # Default value is False (0)
|
|
|
|
def __str__(self):
|
|
return f"Number (ID: {self.id}, Status: {self.status})"
|
|
|
|
|
|
class SqlStatus(models.Model):
|
|
value = models.CharField(max_length=15, default="0")
|
|
|
|
def __str__(self):
|
|
return f"Number (ID: {self.id}, Status: {self.value})"
|
|
|
|
class RestoreDatabase(models.Model):
|
|
value = models.CharField(max_length=15, default="0")
|
|
|
|
def __str__(self):
|
|
return f"Number (ID: {self.id}, Status: {self.value})"
|
|
|
|
|
|
|
|
class DdosPrediction(models.Model):
|
|
device = models.ForeignKey(Devices, on_delete=models.CASCADE)
|
|
user = models.ForeignKey(UserProfile, on_delete=models.CASCADE) # Add this field to reference the user
|
|
file_path = models.FileField(upload_to='ddos_predictions/')
|
|
uploaded_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
def __str__(self):
|
|
return f"Prediction for {self.device.device_name} by {self.user.user.username} at {self.uploaded_at}"
|
|
|
|
class Rensomware_TypePrediction(models.Model):
|
|
device = models.ForeignKey(Devices, on_delete=models.CASCADE)
|
|
user = models.ForeignKey(UserProfile, on_delete=models.CASCADE) # Add this field to reference the user
|
|
file_path = models.FileField(upload_to='ransomware_predictions/', max_length=555)
|
|
uploaded_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
def __str__(self):
|
|
return f"Prediction for {self.device.device_name} by {self.user.user.username} at {self.uploaded_at}"
|
|
|
|
class Rensomware_AuditPrediction(models.Model):
|
|
device = models.ForeignKey(Devices, on_delete=models.CASCADE)
|
|
user = models.ForeignKey(UserProfile, on_delete=models.CASCADE) # Add this field to reference the user
|
|
file_path = models.FileField(upload_to='ransomware_predictions/',max_length=555)
|
|
uploaded_at = models.DateTimeField(auto_now_add=True)
|
|
|
|
def __str__(self):
|
|
return f"Prediction for {self.device.device_name} by {self.user.user.username} at {self.uploaded_at}"
|
|
|
|
|