csv_exporter.py 469 B

123456789101112131415
  1. from .models import Review
  2. import csv
  3. revs = Review.objects.all()
  4. def write_to_csv():
  5. with open('review_dataset.csv', 'w') as file:
  6. header_row = ['review', 'reply']
  7. writer = csv.writer(file)
  8. writer.writerow(header_row)
  9. for rev in revs:
  10. review = rev.comment if rev.comment else 'NULL'
  11. reply = rev.reply.replied_text if rev.reply else 'NULL'
  12. row = [review, reply]
  13. writer.writerow(row)