def save_tweet(status=None, status_json=None, tw_id=None):
if status is not None:
status_dict = json.loads(status.json)
else:
status_dict = status_json
tweet_obj = TWTweet()
tweet_obj.tweet_id = status_dict["id_str"]
tweet_obj.tw_id = tw_id
tweet_obj.user = status_dict["user"]
tweet_obj.text = status_dict["text"]
tweet_obj.fields = status_dict
tweet_obj.created_at = datetime.datetime.utcnow()
if TWTweet.collection().find_one({"tweet_id": tweet_obj.tweet_id, "tw_id": tweet_obj.tw_id}) is None:
tweet_obj._id = TWTweet.collection().insert(tweet_obj.serialize())
else:
TWTweet.collection().update({"tweet_id": tweet_obj.tweet_id, "tw_id": tweet_obj.tw_id}, tweet_obj.serialize())
tweet_obj = TWTweet.unserialize(TWTweet.collection().find_one({"tweet_id": tweet_obj.tweet_id, "tw_id": tweet_obj.tw_id}))
return tweet_obj
def put_tweet(text, tw_id, key, secret, state, media_file):
datetime_now = datetime.datetime.utcnow().isoformat('T')
tw = TWTweet()
tw.tw_id = tw_id
tw.text = text
tw.created_at = datetime_now
if state == CampaignState.PUBLISHED:
if media_file is None:
api = get_api(key, secret)[0]
data = api.update_status(text)
tw = save_tweet(status=data, tw_id=tw_id)
else:
api = Twython(TW_CONSUMER_KEY, TW_CONSUMER_SECRET, key, secret)
data = api.update_status_with_media(media=open(media_file, 'rb'), status=text)
tw = save_tweet(status_json=data, tw_id=tw_id)
else:
tw._id = TWTweet.collection().insert(tw.serialize())
return tw
