Added command to export reviewer's votes

junction/proposals/management/commands/export_conf_proposals.py

@@ -0,0 +1,64 @@
+
+from os.path import join
+from settings.common import ROOT_DIR
+import pandas as pd
+
+from django.shortcuts import get_object_or_404
+from django.core.exceptions import PermissionDenied
+from django.core.management.base import BaseCommand
+from django.contrib.auth.models import User
+
+from junction.conferences.models import Conference
+from junction.proposals.models import (
+    Proposal,
+    ProposalComment,
+    ProposalSectionReviewer,
+    ProposalSectionReviewerVoteValue,
+    )
+from junction.base.constants import (
+    ProposalReviewVote,
+    ProposalStatus,
+    ProposalReviewStatus,
+    )
+
+class Command(BaseCommand):
+
+    def add_arguments(self, parser):
+        parser.add_argument('--conference_slug', default=None, help='Enter the conference slug where to export reviewer votes from')
+
+        parser.add_argument('--user_id', default=None, help='Enter your user id')
+
+    def handle(self, *args, **options):
+
+        conference = get_object_or_404(Conference, slug=self.options.get('conference_slug'))
+        user = User.objects.get(id=self.options.get('user_id'))
+
+        if not conference_is_moderator(user=user, conference=conference):
+            raise PermissionDenied
+
+        proposal_sections = conference.proposal_sections.all()
+        proposals_qs = Proposal.objects.select_related('proposal_type', 'proposal_section', 'conference', 'author').filter(conference=conference, status=ProposalStatus.PUBLIC)
+        proposals_qs = sorted(proposals_qs, key=lambda x: x.get_reviewer_votes_sum(), reverse=True)
+        vote_values_list = ProposalSectionReviewerVoteValue.objects.order_by('-vote_value')
+        vote_values_list = [v.vote_value for v in vote_values_list]
+        vote_values_desc = tuple(i.description for i in ProposalSectionReviewerVoteValue.objects.order_by('-vote_value'))
+        header = ('Proposal Type', 'Title', 'Sum of reviewer votes', 'No. of reviewer votes') + tuple(vote_values_desc) + ('Public votes count', 'Vote comments')
+        df = pd.DataFrame(columns=header)
+        i = 0
+
+        for section in proposal_sections:
+            section_proposals = [p for p in proposal_qs if p.proposal_section == section]
+
+            for index, p in enumerate(section_proposals, 1):
+                vote_details = tuple(p.get_reviewer_votes_count_by_value(v) for v in vote_values_list)
+                vote_comment = '\n'.join([comment.comment for comment in p.proposalcomment_set.filter(vote=True, deleted=False)])
+                row = (p.proposal_type.name, p.title, p.get_reviewer_votes_sum(), p.get_reviewer_votes_count(),) + vote_details + (p.get_voutes_count(), vote_comment,)
+                df.loc[i] = row
+                i+=1
+
+        csv_file_name = "%s-%s" % (user.name, conference.name)
+        csv_file_location = join(ROOT_DIR, 'csvs', excel_file_name)
+
+        df.to_csv(csv_file_location, sep='\t')
+
+        self.stdout.write("Successfully created the csv file")

requirements.txt

@@ -2,6 +2,7 @@ Django==1.9
 celery==3.1.20
 djangorestframework==3.4
 django-filter==1.0.1
+pandas==0.24.2
 
 # Persistent Store
 # -------------------------------------------------