rushilwiz/reinvest
1
0
Fork 0
mirror of https://github.com/Rushilwiz/reinvest.git synced 2025-04-04 20:40:19 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(backend): added buy func

Browse Source
This commit is contained in:
Rushil Umaretiya 2021-01-31 11:34:57 -05:00
parent 6e8494a516
commit 2e51f0904f
No known key found for this signature in database
GPG Key ID: 4E8FAF9C926AF959
3 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/api/serializers.py
View File

@ -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:

3
backend/api/urls.py
View File

@ -12,7 +12,8 @@ urlpatterns = [
path('charity/<int:pk>', views.CharityViewSet.as_view({'get': 'retrieve'})),
path('charity/', views.CharityViewSet.as_view({'get': 'list', 'post': 'create'})),
path('stock/<uuid:pk>', 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'),

9
backend/api/views.py
View File

@ -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()