def save_to_database(data): try: user = session.query(User).filter_by(id=str(data['user']['id'])).one() except NoResultFound: user = create_user_helper(data['user']) session.add(user) hashtag_results = [] hashtags = data['entities']['hashtags'] for hastag in hashtags: hashtag = hashtag['text'].lower() try: hashtag_obj = session.query(Hashtag).filter_by(text=hashtag).one() except NoResultFound: hashtag_obj = Hashtag(text=hashtag) session.add(hashtag_obj) hashtag_results.append(hashtag_obj) tweet = create_tweet_helper(data, user) for hastag in hashtag_results: tweet.hashtags.append(hashtag) session.add(tweet) session.commit()
def save_to_database(data): try: user = session.query(User).filter_by(id=str(data['user']['id'])).one() except NoResultFound: user = create_user_helper(data['user']) session.add(user) hashtag_results = [] hashtags = data['entities']['hashtags'] for hashtag in hashtags: hashtag = hashtag['text'].lower() try: hashtag_obj = session.query(Hashtag).filter_by(text=hashtag).one() except NoResultFound: hashtag_obj = Hashtag(text=hashtag) session.add(hashtag_obj) hashtag_results.append(hashtag_obj) tweet = create_tweet_helper(data, user) for hashtag in hashtag_results: tweet.hashtags.append(hashtag) session.add(tweet) session.commit()