Change print to log for lines in twitter stream class (#1784)

This commit is contained in:
Jonathan Zernik 2021-11-07 02:47:09 -08:00 committed by GitHub
parent 2f2f2059b4
commit 581ef4f226
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -20,6 +20,7 @@
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.
import json
import logging
from typing import List
import requests
@ -27,6 +28,9 @@ import requests
from squeaknode.core.tweet_stream import TweetStream
logger = logging.getLogger(__name__)
class TwitterStream:
twitter_stream_url = "https://api.twitter.com/2/tweets/search/stream"
@ -64,7 +68,7 @@ class TwitterStream:
"Cannot get rules (HTTP {}): {}".format(
response.status_code, response.text)
)
print(json.dumps(response.json()))
logger.info(json.dumps(response.json()))
return response.json()
def delete_all_rules(self, rules):
@ -84,7 +88,7 @@ class TwitterStream:
response.status_code, response.text
)
)
print(json.dumps(response.json()))
logger.info(json.dumps(response.json()))
def set_rules(self, delete):
sample_rules = [
@ -102,7 +106,7 @@ class TwitterStream:
"Cannot add rules (HTTP {}): {}".format(
response.status_code, response.text)
)
print(json.dumps(response.json()))
logger.info(json.dumps(response.json()))
def get_stream(self, set) -> TweetStream:
response = requests.get(
@ -110,7 +114,7 @@ class TwitterStream:
auth=self.bearer_oauth_fn,
stream=True,
)
print(response.status_code)
logger.info(response.status_code)
if response.status_code != 200:
raise Exception(
"Cannot get stream (HTTP {}): {}".format(