Advertisement
Guest User

Untitled

a guest
Jan 15th, 2016
77
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.93 KB | None | 0 0
  1. import sys
  2. import json
  3. from pprint import pprint
  4. import mysql.connector
  5.  
  6.  
  7. print sys.argv[1]
  8. count=0
  9. insertTemplate = "INSERT INTO yelp (funnyVote, usefulVote, coolVote, user_id, review_id, stars, date, text_review, business_id) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s)"
  10.  
  11. cnx = mysql.connector.connect(user='root', password='password123',
  12. host='localhost',
  13. database='db')
  14. cursor = cnx.cursor()
  15.  
  16. with open(sys.argv[1]) as f:
  17. for line in f:
  18. data = json.loads(line)
  19. data_review = (data['votes']['funny'], data['votes']['useful'], data['votes']['cool'], data['user_id'], data['review_id'], data['stars'], data['date'], data['text'], data['business_id'])
  20. insertCommand = insertTemplate % data_review
  21. print insertCommand
  22. result = cursor.execute(insertTemplate, data_review)
  23. print result
  24. break
  25.  
  26. cnx.commit()
  27. cursor.close()
  28. cnx.close()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement