JuliusFreudenberger/tripadvisor-parser
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add check to ensure export directory exists

Browse source
This commit is contained in:
Julius Freudenberger 2023-05-01 12:03:38 +02:00
parent d5081f6cd8
commit ce7c2102a1
2 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
.gitignore vendored
View file

@ -2,6 +2,7 @@
# Edit at https://www.toptal.com/developers/gitignore?templates=python,pycharm
urls.txt
export/
### PyCharm ###
# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio, WebStorm and Rider

7
csv_exporter.py
View file

@ -1,6 +1,9 @@
import os.path
from tripadvisor_attraction import TripadvisorAttraction, TripadvisorReview
TABLE_HEADING = "username^review_title^review_text^posting_date^count_stars^count_likes^translated_by"
EXPORT_DIRECTORY = 'export'
def export_attraction(attraction: TripadvisorAttraction):
@ -8,7 +11,9 @@ def export_attraction(attraction: TripadvisorAttraction):
for review in attraction.reviews:
export_string += line_for_review(review) + '\n'
with open(f'export/{attraction.title} ({attraction.count_of_reviews}).csv', 'w') as export_file:
if not os.path.isdir(EXPORT_DIRECTORY):
os.makedirs(EXPORT_DIRECTORY)
with open(f'{EXPORT_DIRECTORY}/{attraction.title} ({attraction.count_of_reviews}).csv', 'w') as export_file:
export_file.writelines(export_string)