script/stream/recorder_stream.py
changeset 884 07f1c6854df9
parent 883 8ae3d91ea4ae
child 886 1e110b03ae96
equal deleted inserted replaced
883:8ae3d91ea4ae 884:07f1c6854df9
    13 import inspect
    13 import inspect
    14 import iri_tweet.stream
    14 import iri_tweet.stream
    15 import logging
    15 import logging
    16 import os
    16 import os
    17 import re
    17 import re
    18 import requests.auth
    18 import requests_oauthlib
    19 import shutil
    19 import shutil
    20 import signal
    20 import signal
    21 import socket
    21 import socket
    22 import sqlalchemy.schema
    22 import sqlalchemy.schema
    23 import sys
    23 import sys
    58     if options.username and options.password:
    58     if options.username and options.password:
    59         auth = requests.auth.BasicAuthHandler(options.username, options.password)        
    59         auth = requests.auth.BasicAuthHandler(options.username, options.password)        
    60     else:
    60     else:
    61         consumer_key = models.CONSUMER_KEY
    61         consumer_key = models.CONSUMER_KEY
    62         consumer_secret = models.CONSUMER_SECRET
    62         consumer_secret = models.CONSUMER_SECRET
    63         auth = requests.auth.OAuth1(access_token[0], access_token[1], consumer_key, consumer_secret, signature_type='auth_header')
    63         auth = requests_oauthlib.OAuth1(access_token[0], access_token[1], consumer_key, consumer_secret, signature_type='auth_header')
    64     return auth
    64     return auth
    65 
    65 
    66 
    66 
    67 def add_process_event(type, args, session_maker):
    67 def add_process_event(type, args, session_maker):
    68     session = session_maker()
    68     session = session_maker()