user.py

from database import CursorFromConnectionFromPool
import oauth2
import json
from twitter_util import consumer


class User:
    def __init__(self,screen_name,oauth_token,oauth_token_secret,id):
        self.screen_name=screen_name
        self.id=id
        self.oauth_token=oauth_token
        self.oauth_token_secret=oauth_token_secret

    def __repr__(self):
        return "<USER {}>".format(self.screen_name)


    def save_to_database(self):
        with CursorFromConnectionFromPool() as cursor:
                cursor.execute('INSERT into users (screen_name,oauth_token,oauth_token_secret) VALUES(%s,%s,%s)',
                                       (self.screen_name,self.oauth_token,self.oauth_token_secret))


    @classmethod
    def load_to_db_by_screen_name(cls,screen_name):
        with CursorFromConnectionFromPool() as cursor:
            cursor.execute('select * from users where screen_name=%s',(screen_name,))
            user_data= cursor.fetchone()
            if user_data is not None:
                return cls(user_data[1],user_data[2],user_data[3],user_data[0])

    def twitter_request(self,uri,verb='GET'):
        authorized_token = oauth2.Token(self.oauth_token, self.oauth_token_secret)
        authorized_client = oauth2.Client(consumer, authorized_token)

        response, content = authorized_client.request(uri, verb)
        if response.status != 200:
            print("Error occured while searching.")
        return json.loads(content.decode('utf-8'))