Advertisement
Guest User

Untitled

a guest
Jun 20th, 2019
97
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.14 KB | None | 0 0
  1. class Machine(models.Model):
  2. machine_name = models.CharField(max_length=120)
  3.  
  4. def __str__(self):
  5. return self.machine_name
  6.  
  7. class Project(models.Model):
  8. project_name = models.CharField(max_length=120)
  9.  
  10. def __str__(self):
  11. return self.project_name
  12.  
  13. class Simulation(models.Model):
  14. user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE, default=1)
  15. machine = models.ForeignKey(Machine, on_delete=models.CASCADE, default=1)
  16. project = models.ForeignKey(Project, on_delete=models.CASCADE, default=1)
  17.  
  18. test_api_local(method="post", data={"machine":2, "project":1 })
  19.  
  20. class SimulationSerializer(serializers.ModelSerializer):
  21. machine = serializers.SerializerMethodField()
  22. project = serializers.SerializerMethodField()
  23.  
  24.  
  25. class Meta:
  26. model = Simulation
  27.  
  28. fields = ('project',
  29. 'machine',
  30. )
  31.  
  32. def get_machine(self, obj):
  33. print(obj.machine)
  34. project_name = obj.project.project_name
  35. return str(project_name)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement