3 d@sNdZddlZddlZddlmZd ddZGdddZdd d Zd d ZdS)z8 Authentication utilities to accompany `twitterclient`. N)TwythonFcCstj|||dS)z1 Convenience function for authentication ) creds_filesubdirverbose) Authenticate load_creds)rrrr1/tmp/pip-build-v9q4h5k9/nltk/nltk/twitter/util.py credsfromfilesr c@s,eZdZdZddZd ddZd dd ZdS) rz2 Methods for authenticating with Twitter. c CsPd|_d|_i|_ytjd|_|j|_Wn tk rJd|_d|_YnXdS)Nzcredentials.txtZTWITTER)rcreds_fullpathoauthosenvironZ twitter_dir creds_subdirKeyError)selfrrr __init__!s  zAuthenticate.__init__NFc Cs|dk r||_|dkr.|jdkr4d }t|n||_tjjtjj|j|j|_tjj|jsnt d|jt |jR}|rt d|jx6|D].}d|kr|j dd\}}|j |j|j <qWWdQRX|j|d|jS) a Read OAuth credentials from a text file. File format for OAuth 1:: app_key=YOUR_APP_KEY app_secret=YOUR_APP_SECRET oauth_token=OAUTH_TOKEN oauth_token_secret=OAUTH_TOKEN_SECRET File format for OAuth 2:: app_key=YOUR_APP_KEY app_secret=YOUR_APP_SECRET access_token=ACCESS_TOKEN :param str file_name: File containing credentials. ``None`` (default) reads data from `TWITTER/'credentials.txt'` Nz+Supply a value to the 'subdir' parameter orz& set the TWITTER environment variable.zCannot find file zReading credentials file =)rzQSupply a value to the 'subdir' parameter or set the TWITTER environment variable.)rr ValueErrorr pathnormpathjoinr isfileOSErroropenprintsplitstripr _validate_creds_file) rrrrmsginfilelinenamevaluerrr r-s*      zAuthenticate.load_credscsd}ddddg}d}dddg}tfdd|Dr:d }ntfd d|DrTd }|pZ|sd jd }|tjj7}t|n|rtd jddS)z%Check validity of a credentials file.Fapp_key app_secretZ oauth_tokenZoauth_token_secret access_tokenc3s|]}|jkVqdS)N)r ).0k)rrr isz4Authenticate._validate_creds_file..Tc3s|]}|jkVqdS)N)r )r(r))rrr r*ksz Missing or incorrect entries in  zCredentials file "z " looks goodN)allrpprintpformatr rr)rrZoauth1Z oauth1_keysoauth2Z oauth2_keysr r)rr rcs   z!Authenticate._validate_creds_file)NNF)F)__name__ __module__ __qualname____doc__rrrrrrr rs 6rc Cs|dkr"tjjt}tjj|d}t|d}|d}|d}t||dd}|j}d|d }t|d }t ||d WdQRXdS) zc For OAuth 2, retrieve an access token for an app and append it to a credentials file. Nzcredentials2.txt)rr%r&)Z oauth_versionz access_token=r+a)file) r rdirname__file__rr rZobtain_access_tokenrr) rrr/r%r&Ztwitterr'tokr!rrr add_access_tokenvs    r:cCs*tjj|r|Stjjtjjd|SdS)z If the path is not absolute, guess that it is a subdirectory of the user's home directory. :param str pth: The pathname of the directory where files of tweets should be written ~N)r risabs expanduserr)pthrrr guess_paths r?)NNF)N) r3r r-Ztwythonrr rr:r?rrrr  s  Z