Link trading times to strategies

core/models.py

@@ -221,11 +221,25 @@ class Callback(models.Model):
     symbol = models.CharField(max_length=255)
 
 
+class TradingTime(models.Model):
+    user = models.ForeignKey(User, on_delete=models.CASCADE)
+    name = models.CharField(max_length=255)
+    description = models.TextField(null=True, blank=True)
+    start_day = models.CharField(choices=DAY_CHOICES, max_length=255)
+    end_day = models.CharField(choices=DAY_CHOICES, max_length=255)
+    start_time = models.TimeField()
+    end_time = models.TimeField()
+
+    def __str__(self):
+        return f"{self.name} ({self.start_day} at {self.start_time} - {self.end_day} at {self.end_time})"
+
+
 class Strategy(models.Model):
     user = models.ForeignKey(User, on_delete=models.CASCADE)
     name = models.CharField(max_length=255)
     description = models.TextField(null=True, blank=True)
     account = models.ForeignKey(Account, on_delete=models.CASCADE)
+    trading_times = models.ManyToManyField(TradingTime, blank=True)
     order_type = models.CharField(
         choices=TYPE_CHOICES, max_length=255, default="market"
     )
@@ -246,16 +260,6 @@ class Strategy(models.Model):
         return self.name
 
 
-class TradingTime(models.Model):
-    user = models.ForeignKey(User, on_delete=models.CASCADE)
-    name = models.CharField(max_length=255)
-    description = models.TextField(null=True, blank=True)
-    start_day = models.CharField(choices=DAY_CHOICES, max_length=255)
-    end_day = models.CharField(choices=DAY_CHOICES, max_length=255)
-    start_time = models.TimeField()
-    end_time = models.TimeField()
-
-
 # class Perms(models.Model):
 #     class Meta:
 #         permissions = (