diff --git a/backend/api/serializers.py b/backend/api/serializers.py index a59e9e4..443d458 100644 --- a/backend/api/serializers.py +++ b/backend/api/serializers.py @@ -39,7 +39,7 @@ class StockSerializer(serializers.ModelSerializer): class StockCreateSerializer(serializers.ModelSerializer): class Meta: model = models.Stock - fields = ('user', 'ticker', 'buy_price', 'quantity', 'uuid') + fields = ('user', 'ticker', 'quantity',) class CharitySerializer(serializers.ModelSerializer): class Meta: diff --git a/backend/api/urls.py b/backend/api/urls.py index a9f0dce..307edd4 100644 --- a/backend/api/urls.py +++ b/backend/api/urls.py @@ -12,7 +12,8 @@ urlpatterns = [ path('charity/', views.CharityViewSet.as_view({'get': 'retrieve'})), path('charity/', views.CharityViewSet.as_view({'get': 'list', 'post': 'create'})), path('stock/', views.StockViewSet.as_view({'get': 'retrieve'})), - path('stock/', views.StockViewSet.as_view({'get': 'list', 'post': 'create'})), + path('stock/create', views.StockViewSet.as_view({'post': 'create'})), + path('stock/', views.StockViewSet.as_view({'get': 'list'})), path('profile/create', views.UserProfileCreate.as_view()), path('profile/', views.UserProfileDetail.as_view()), path('token/', TokenObtainPairView.as_view(), name='token_obtain_pair'), diff --git a/backend/api/views.py b/backend/api/views.py index 7b5b2aa..af1639f 100644 --- a/backend/api/views.py +++ b/backend/api/views.py @@ -42,7 +42,13 @@ class StockViewSet(ModelViewSet): data.update({'user': request.user}) serializer = StockCreateSerializer(data=request.data) serializer.is_valid(raise_exception=True) - serializer.save() + login = r.login(os.getenv("ROBINHOOD_USER"), os.getenv("ROBINHOOD_PASS")) + order = r.order_buy_market(symbol=serializer.data['ticker'], quantity=serializer.data['quantity']) + stonks = r.build_holdings() + for key, value in stonks.items(): + if Stock.objects.filter(uuid=value['id']).count() != 1: + stock = Stock.objects.create(user=request.user.profile, quantity=ceil(float(value['quantity'])), ticker=key, buy_price=value['price'], uuid=value['id']) + return Response(serializer.data, status=status.HTTP_200_OK) class UserProfileDetail(APIView): @@ -57,7 +63,6 @@ class UserProfileCreate(CreateAPIView): serializer_class = UserCreateSerializer class FetchRobinhoodStocks(APIView): - permission_classes = [permissions.AllowAny] def get(self, request, format=None): login = r.login(os.getenv("ROBINHOOD_USER"), os.getenv("ROBINHOOD_PASS")) stonks = r.build_holdings()