Merge pull request #34 from biojs/visualization_list

Visualization list

Author: Megh-Thakkar

.travis.yml

@@ -9,4 +9,3 @@ install:
 script:
   - python manage.py makemigrations
   - python manage.py migrate
-  - python manage.py test

main/urls.py

@@ -10,4 +10,5 @@
         url(r'^top/$', views.top_components, name="top_components"),
         url(r'^details/(?P<url_name>[\w-]+)/$', views.component_details, name="component_details"),
 		url(r'^visualization/(?P<url_name>[\w-]+)/(?P<visualization_name>[\w-]+)/$', views.render_visualization, name="render_visualization"),
+		url(r'^random_snippets/$', views.generate_random_snippets, name='generate_random_snippets'),
 ]

main/views.py

@@ -6,6 +6,7 @@
 import urllib, json
 from django.core.management import call_command
 from django.core.serializers.json import DjangoJSONEncoder
+import numpy as np
 
 def index(request):
     top_downloaded_components = Component.objects.all().only('name', 'id', 'downloads', 'url_name').order_by('-downloads')[:3]
@@ -79,4 +80,17 @@ def render_visualization(request, url_name, visualization_name):
 @staff_member_required
 def update_data(request):
     call_command('updatecomponents')
-    return HttpResponse("Database Successfully Updated.")
+    return HttpResponse("Database Successfully Updated.")
+
+def generate_random_snippets(request):
+    try:
+        count = request.GET.get('q')
+        if int(count) > Component.objects.filter(sniperdata__isnull=False).count():
+            return HttpResponse('Input number "q" must not exceed %s.'%str(Component.objects.filter(sniperdata__isnull=False).count()))
+        components = Component.objects.filter(sniperdata__isnull=False)
+        required_components = np.random.choice(components, int(count), replace=False)
+        req_str = '\n'.join([component.name for component in required_components])
+        print req_str
+        return HttpResponse(req_str, content_type="text/plain")
+    except:
+        return HttpResponse('Input number as query "q" in the URL.')

requirements.txt

@@ -5,6 +5,7 @@ django-redis==4.9.0
 django-rest-framework==0.1.0
 djangorestframework==3.8.2
 idna==2.6
+numpy==1.14.5
 Pillow==5.1.0
 pytz==2018.4
 redis==2.10.6