Added Support to get User files (#22)

This commit is contained in:
Aman Tahiliani
2024-04-22 09:57:12 -04:00
committed by GitHub
parent 776216ed11
commit 330e34f0cd
2 changed files with 24 additions and 0 deletions

View File

@@ -68,4 +68,5 @@ urlpatterns = [
path("api/report_user/", views.ReportUserView.as_view(), name="report-user"),
path("api/register/", views.RegisterFile.as_view(), name="file-register"),
path("api/files/filter/", views.FileFilterView.as_view(), name="file-filter-view"),
path("api/get-peer-files/", views.UserFilesView.as_view(), name="user-files-view"),
]

View File

@@ -319,6 +319,29 @@ class FileFilterView(APIView):
)
class UserFilesView(APIView):
authentication_classes = [TokenAuthentication]
permission_classes = [IsAuthenticated]
def get(self, request):
user = request.user
owned_files = user.owned_files.all()
shared_files = user.shared_files.all()
all_user_files = owned_files.union(shared_files)
serializer = FileSerializer(all_user_files, many=True)
try:
return Response(serializer.data, status=status.HTTP_200_OK)
except Exception as e:
print(f"Error: {e}")
return Response(
{"error": f"An error occured {e}"}, status=status.HTTP_400_BAD_REQUEST
)
class ReportUserView(APIView):
authentication_classes = [TokenAuthentication]
permission_classes = [IsAuthenticated]