reader = csv.reader(file)
	for row in reader:
		if row[3] == '?':
			row[3] = None
		new_user = User(
			screen_name = row[0], #screen_name
			user_id = row[1], #user_id
			contributors_enabled = row[2], #contributors_enabled
			hours_since_last_tweet = row[3], #hours_since_last_tweet
			declared_blogger = row[4], #declared_blogger
			declared_company = row[5], #declared_company
			num_entities = row[6], #num_profile_entities
			tweets_favorited  = row[7], #tweets_favorited
			num_followers = row[8], #num_followers
			num_friends = row[9], #num_friendships
			geo_enabled = row[10], #geo_enabled
			is_translator = row[11], #is_translator
			listed_count = row[12], #listed_count
			protected = row[13], #protected
			num_tweets = row[14], #num_tweets
			has_profile_url = row[15], #has_profile_url
			verified = row[16]) #verifed
		new_user.save()

		new_classified_user = TrainingData(
			user = new_user,
			classification = row[17])
		
		new_classified_user.save()

示例#2
0
def main():
    if 'HEROKU_CONSUMER_KEY' in os.environ:
        CONSUMER_KEY = os.environ['HEROKU_CONSUMER_KEY']
        CONSUMER_SECRET = os.environ['HEROKU_CONSUMER_SECRET']
        ACCESS_TOKEN = os.environ['HEROKU_ACCESS_TOKEN']
        ACCESS_SECRET = os.environ['HEROKU_ACCESS_SECRET']

        twitter_api = API(consumer_key=CONSUMER_KEY, consumer_secret=CONSUMER_SECRET, access_token=ACCESS_TOKEN, access_token_secret=ACCESS_SECRET)
    else:
        twitter_api = API()
    TWITTER_BOT = twitter_api.authenticate()
    
    did_ff = False
    ind_adv = False
    while True:
        ####create the needed API Objects
        searcher = Search(TWITTER_BOT)
        followers = Follower(TWITTER_BOT)
        classifier = tree.DecisionTreeClassifier()
        advertiser = TeaAdvertiser(TWITTER_BOT)
        #wda = WeekDayAware()

        #### check to see if theres a new follower to follow
        #if new follower does not already exist
        #i guess i could try/except IntegrityError instead:

        
        if not User.objects.filter(user_id=followers.most_recent_id).exists():


            newUser = followers.arrangeUserData()
            try:
                newUser.save()
                print(newUser)
            except Exception as e:
                print(newUser, e)


            ### if adding was successful (new user)
            #print(user_id)

            if newUser.user_id:
                ##retrive him from user database
                most_recent_user_object = User.objects.get(user_id=newUser.user_id)
    
                ## set up the classifier with all the training data
                u = User
                td = TrainingData
                target = TrainingData.get_targets(td)
                data = TrainingData.get_data(td)
                classifier = classifier.fit(data, target)
                
                ## predict the class of the new user
                new_data_to_predict_on = User.get_data(u, user_object=most_recent_user_object)
                prediction = classifier.predict(new_data_to_predict_on)
               
                ### save the new prediction in the TrainingData database
                try:
                    #prediction is a numpy array type, whose first index is the (string) number i want.
                    newTD = TrainingData.saveClassified(td, newUser, prediction[0])
                    print(newTD)
                    print('updated training data and user table')
                except Exception as e:
                    print(newUser, prediction, e)
                

            #get the trianing data we just saved and tweet at them if they are an individual
            if TrainingData.objects.filter(user=most_recent_user_object).exists():
            ## if it was a blogger or individual, tweet at them!
                newFollower = TrainingData.objects.get(user=followers.most_recent_id)
                if newFollower.classification == '0' or newFollower.classification == '2':
                    followers.mention_new_follower()
               ## if it was a blogger or individual, tweet at them!
                elif newFollower.classification == '1':
                    #steal followers
                    followers.poach_followers(newFollower.user.screen_name, 20)
        ##if the user was a dupe, we've already done this.
        else: 
            print('already following most recent')

        #### done with following/mentioning

        #begin favoriting! Favorite no matter what.    
        statuses = searcher.get_statuses(query="#genmaicha", count=5)
        for status in statuses:
            try:
                TWITTER_BOT.create_favorite(status.id)
                print('favorited ' + str(status.id))
            except tweepy.TweepError as e:
                print(e)
                print("favorite failed on status: " + str(status.id))
                print(datetime.datetime.now())
                continue


        statuses = searcher.get_statuses(query="#teasontheloose", count=5)
        
        for status in statuses:
            try:
                TWITTER_BOT.create_favorite(status.id)
                print('favorited ' + str(status.id))
            except tweepy.TweepError as e:
                print(e)
                print("favorite failed on status: " + str(status.id))
                print(datetime.datetime.now())
                continue

        #advertise

        advertiser.advertise_this_months()
        advertiser.advertise_last_months()

        ##follow friday shout-outs,
        # if its friday,
        # and we havent already done it:

        # if (wda.today_int == 4) and (not did_ff):
        #     followers.ff_tweet(wda=wda)
        #     did_ff = True
        # #reset if we've done ff on a non-friday
        # #so it will be ready next week
        # elif wda.today_int != 4:
        #     did_ff = False
        # ## done, wait and go again.
        print('done')
        time.sleep(1800)