Skip to content

Instantly share code, notes, and snippets.

@MilanRgm
Created September 15, 2018 15:44
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save MilanRgm/132eb6c0ba0cf66e48fa0ca4c17ef732 to your computer and use it in GitHub Desktop.
Save MilanRgm/132eb6c0ba0cf66e48fa0ca4c17ef732 to your computer and use it in GitHub Desktop.
class Company(models.Model):
owner = models.ForeignKey(User, on_delete=models.CASCADE, null=True)
name = models.CharField(max_length=150, blank=False, null=False)
phone_number = models.CharField(max_length=15, blank=False, null=False)
class Brand(models.Model):
company = models.ForeignKey(Company, related_name='company_brand', on_delete=models.CASCADE)
name = models.CharField(max_length=150, blank=False, null=False)
website = models.URLField()
description = models.TextField(blank=False)
class BusinessModel(models.Model):
company = models.ForeignKey(Company, related_name='company_business_model', on_delete=models.CASCADE)
industry = models.ForeignKey(Industry, null=True, related_name='industry', on_delete=models.SET_NULL)
segments = models.ForeignKey(Segment, on_delete=models.SET_NULL, null=True)
total_investment = models.CharField(max_length=50, choices=TOTAL_INVESTMENT, default=None)
def __str__(self):
return self.company.name
class BrandSerializer(serializers.ModelSerializer):
class Meta:
model = Brand
fields = '__all__'
class BusinessModelSerializer(serializers.ModelSerializer):
class Meta:
model = BusinessModel
fields = '__all__'
class CompanySerializer(serializers.ModelSerializer):
company_brand = BrandSerializer(many=True)
company_business_model = BusinessModelSerializer(many=True)
class Meta:
model = Company
fields = '__all__'
class CompanyView(views.APIView):
serializer_class = CompanySerializer
def get(self, request, pk=None, format=None):
reply = {}
try:
companies = Company.objects.all()
if pk:
company = companies.get(pk=pk)
reply['data'] = self.serializer_class(company).data
reply['data'] = self.serializer_class(companies, many=True).data
except Company.DoesNotExist:
reply['data'] = []
return Response(reply, status.HTTP_200_OK)
def post(self, request, pk=None, format=None):
reply = {}
company={}
if pk is not None:
try:
company = Company.object.get(pk=pk)
except Company.DoesNotExist:
return error.RequestedResourceNotFound().as_response()
serialized_data = self.serializer_class(instance=company, data=request.data, partial=True)
if serialized_data.is_valid():
serialized_data.save(owner=request.user)
else:
return error.ValidationError(serialized_data.errors).as_response()
reply['data'] = serialized_data.data
return Response(reply, status.HTTP_200_OK)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment