StormFalcon32

InputOutput.py

Feb 11th, 2020
152
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.89 KB | None | 0 0
  1. import csv
  2. from datetime import datetime
  3.  
  4. num_tweets = 1
  5. prev_num = 1
  6. file_num = -1
  7.  
  8.  
  9. def debug_out(value):
  10.     print(value + ' NumTweets: ' + str(num_tweets))
  11.  
  12.  
  13. def output_file(columns, values, print_columns):
  14.     global num_tweets
  15.     global prev_num
  16.     global file_num
  17.     new_file = False
  18.     if file_num == -1:
  19.         new_file = True
  20.         file_num = 0
  21.     if num_tweets >= (prev_num + 100000):
  22.         file_num += 1
  23.         prev_num = num_tweets
  24.         new_file = True
  25.     with open('Data%s.csv' % file_num, 'a', encoding='utf-8', newline='') as csv_out:
  26.         writer = csv.writer(csv_out, delimiter=',',
  27.                             quotechar='"', quoting=csv.QUOTE_MINIMAL)
  28.         if new_file and print_columns:
  29.             writer.writerow(columns)
  30.         writer.writerow(values)
  31.         debug_out(str(datetime.now()))
  32.         num_tweets += 1
  33.         new_file = False
Advertisement
Add Comment
Please, Sign In to add comment