B >?[@s\dZddlmZddlZddlZddlmZdddZGdd d eZ dd d Z d d Z dS)z@ Authentication utilities to accompany :module:`twitterclient`. )print_functionN)TwythonFcCstj|||dS)z1 Convenience function for authentication ) creds_filesubdirverbose) Authenticate load_creds)rrrr 0lib/python3.7/site-packages/nltk/twitter/util.py credsfromfilesr c@s,eZdZdZddZd ddZd dd ZdS) rz2 Methods for authenticating with Twitter. cCsPd|_d|_i|_ytjd|_|j|_Wn tk rJd|_d|_YnXdS)Nzcredentials.txtZTWITTER)rcreds_fullpathoauthosenvironZ twitter_dir creds_subdirKeyError)selfr r r __init__#s  zAuthenticate.__init__NFc Cs|dk r||_|dkr.|jdkr4d}t|n||_tjtj|j|j|_tj|jsnt d |jt |jR}|rt d |jx6|D].}d|kr| dd\}}||j|<qWWdQRX|j|d|jS)ag 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'` NzQSupply a value to the 'subdir' parameter or set the TWITTER environment variable.zCannot find file {}zReading credentials file {}=)r)rr ValueErrorrpathnormpathjoinr isfileOSErrorformatopenprintsplitstripr _validate_creds_file) rrrrmsginfilelinenamevaluer r r r/s(      zAuthenticate.load_credscsd}ddddg}d}dddg}tfdd|Dr:d }ntfd d|DrTd }|s|sd j}|tj7}t|n|rtd jd S)z%Check validity of a credentials file.Fapp_key app_secretZ oauth_tokenZoauth_token_secret access_tokenc3s|]}|jkVqdS)N)r ).0k)rr r nsz4Authenticate._validate_creds_file..Tc3s|]}|jkVqdS)N)r )r*r+)rr r r,psz#Missing or incorrect entries in {} z Credentials file "{}" looks goodN)allrrpprintZpformatr rr)rrZoauth1Z oauth1_keysoauth2Z oauth2_keysr"r )rr r!hs    z!Authenticate._validate_creds_file)NNF)F)__name__ __module__ __qualname____doc__rrr!r r r r rs 9rc Cs|dkr"tjt}tj|d}t|d}|d}|d}t||dd}|}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_versionzaccess_token={} a)file) rrdirname__file__rr rZobtain_access_tokenrrr) rrr/r'r(Ztwitterr)tokr#r r r add_access_token{s    r:cCs*tj|r|Stjtjd|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)rrisabs expanduserr)Zpthr r r guess_paths r>)NNF)N) r3Z __future__rrr.Ztwythonrr objectrr:r>r r r r  s   ]