parser args added

Author: abhijitbonik

website/management/commands/fetch_questions.py

@@ -7,6 +7,8 @@
 import uuid
 
 class Command(BaseCommand):
+    def add_arguments(self, parser):
+        parser.add_argument('foss', nargs='+',type=str, help='Indicates the foss list.')
 
     @tx.atomic
     def handle(self, *args, **options):
@@ -15,13 +17,15 @@ def handle(self, *args, **options):
         with open(settings.MEDIA_ROOT + meta_file_name, "w+", newline='') as metafile:
             metawriter = csv.writer(metafile, dialect='excel', delimiter='|')
             metawriter.writerow(["FOSS","Tutorial","Video Minute Range","Video Second Range","Question Title","Question Body","Question_Date","Question Posted By","Answer No.","Answer Body","Answer Date", "Answer Posted By"])
-            questions = Question.objects.filter(category="Advance-C", status=1)
-            for q in questions:
-                metawriter.writerow([q.category,q.tutorial,q.minute_range,q.second_range,q.title,strip_tags(q.body).strip(" "),q.date_created,q.user()])
-                answers = Answer.objects.filter(question=q)
-                for i, a in enumerate(answers):
-                    metawriter.writerow(["","","","","","","","",i+1, strip_tags(a.body).strip(" "),a.date_created, a.user()])
-                metawriter.writerow([])
+            foss = options['foss']
+            for f in foss:
+                questions = Question.objects.filter(category=f, status=1)
+                for q in questions:
+                    metawriter.writerow([q.category,q.tutorial,q.minute_range,q.second_range,q.title,strip_tags(q.body).strip(" "),q.date_created,q.user()])
+                    answers = Answer.objects.filter(question=q)
+                    for i, a in enumerate(answers):
+                        metawriter.writerow(["","","","","","","","",i+1, strip_tags(a.body).strip(" "),a.date_created, a.user()])
+                    metawriter.writerow([])
             print("Metadata File Generated. Please find the file at location given below.")
             print(metafile.name)