csv_exporter.py 861 B

123456789101112131415161718192021
  1. from .models import Review
  2. import csv
  3. revs = Review.objects.all()
  4. def write_to_csv():
  5. with open('review_data_set.csv', 'w') as file:
  6. header_row = ['reviewer', 'location', 'review', 'star_rating', 'create_time', 'reply', 'reply_update_time']
  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. star_rating = rev.star_rating
  12. create_time = rev.create_time
  13. reviewer = rev.reviewer_name
  14. location = rev.location.care_name
  15. reply = rev.reply.replied_text if rev.reply else 'NULL'
  16. reply_update_time = rev.reply.create_time if rev.reply else 'NULL'
  17. row = [reviewer, location, review, star_rating, create_time, reply, reply_update_time]
  18. writer.writerow(row)