Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import sys
- import json
- from pprint import pprint
- import mysql.connector
- print sys.argv[1]
- count=0
- 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)"
- cnx = mysql.connector.connect(user='root', password='password123',
- host='localhost',
- database='db')
- cursor = cnx.cursor()
- with open(sys.argv[1]) as f:
- for line in f:
- data = json.loads(line)
- 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'])
- insertCommand = insertTemplate % data_review
- print insertCommand
- result = cursor.execute(insertTemplate, data_review)
- print result
- break
- cnx.commit()
- cursor.close()
- cnx.close()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement