Update scraper/views.py

scraper/views.py

@@ -4,7 +4,13 @@ from .utils.WallmartScraper import WallmartScraper
 from .utils.SephoraScraper import SephoraScraper
 from .utils.UltaScraper import UltaScraper
 from .utils.TargetScraper import TargetScraper
-
+from django_filters.views import FilterView
+from django_tables2.views import SingleTableMixin
+from .filters import ProductFilter
+from .tables import ProductTable
+from .models import Product
+import csv
+from django.http import HttpResponse
 # Create your views here.
 
 
@@ -40,4 +46,28 @@ def refreshTarget(request):
     return render(request, 'scraper/home.html', {'message': 'Target refreshed'})
 
 def index(request):
-    return render(request, 'scraper/home.html')
+    return render(request, 'scraper/home.html')
+
+class ProductListView(SingleTableMixin, FilterView):
+    table_class = ProductTable
+    model = Product
+    template_name = "scraper/home.html"
+    filterset_fields = {
+        'title': ['icontains'],
+        'ingredients': ['icontains'],
+        'product_id': ['exact'],
+        'url': ['exact'],
+        'store_name': ['exact'],
+    }
+    filtered_class = ProductFilter
+
+
+def download_all_product_csv(request):
+    parcels = Product.objects.all()
+    response = HttpResponse(content_type='text/csv')
+    response['Content-Disposition'] = 'attachment; filename="parcels.csv"'
+    writer = csv.writer(response)
+    writer.writerow(['Product Id', 'Title', 'Ingredients', 'Url', 'Store Name'])
+    for parcel in parcels:
+        writer.writerow([parcel.product_id, parcel.title, parcel.ingredients, parcel.url, parcel.store_name])
+    return response